1 //===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
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 the Expr constant evaluator.
10 //
11 // Constant expression evaluation produces four main results:
12 //
13 //  * A success/failure flag indicating whether constant folding was successful.
14 //    This is the 'bool' return value used by most of the code in this file. A
15 //    'false' return value indicates that constant folding has failed, and any
16 //    appropriate diagnostic has already been produced.
17 //
18 //  * An evaluated result, valid only if constant folding has not failed.
19 //
20 //  * A flag indicating if evaluation encountered (unevaluated) side-effects.
21 //    These arise in cases such as (sideEffect(), 0) and (sideEffect() || 1),
22 //    where it is possible to determine the evaluated result regardless.
23 //
24 //  * A set of notes indicating why the evaluation was not a constant expression
25 //    (under the C++11 / C++1y rules only, at the moment), or, if folding failed
26 //    too, why the expression could not be folded.
27 //
28 // If we are checking for a potential constant expression, failure to constant
29 // fold a potential constant sub-expression will be indicated by a 'false'
30 // return value (the expression could not be folded) and no diagnostic (the
31 // expression is not necessarily non-constant).
32 //
33 //===----------------------------------------------------------------------===//
34 
35 #include "Interp/Context.h"
36 #include "Interp/Frame.h"
37 #include "Interp/State.h"
38 #include "clang/AST/APValue.h"
39 #include "clang/AST/ASTContext.h"
40 #include "clang/AST/ASTDiagnostic.h"
41 #include "clang/AST/ASTLambda.h"
42 #include "clang/AST/Attr.h"
43 #include "clang/AST/CXXInheritance.h"
44 #include "clang/AST/CharUnits.h"
45 #include "clang/AST/CurrentSourceLocExprScope.h"
46 #include "clang/AST/Expr.h"
47 #include "clang/AST/OSLog.h"
48 #include "clang/AST/OptionalDiagnostic.h"
49 #include "clang/AST/RecordLayout.h"
50 #include "clang/AST/StmtVisitor.h"
51 #include "clang/AST/TypeLoc.h"
52 #include "clang/Basic/Builtins.h"
53 #include "clang/Basic/TargetInfo.h"
54 #include "llvm/ADT/APFixedPoint.h"
55 #include "llvm/ADT/Optional.h"
56 #include "llvm/ADT/SmallBitVector.h"
57 #include "llvm/Support/Debug.h"
58 #include "llvm/Support/SaveAndRestore.h"
59 #include "llvm/Support/raw_ostream.h"
60 #include <cstring>
61 #include <functional>
62 
63 #define DEBUG_TYPE "exprconstant"
64 
65 using namespace clang;
66 using llvm::APFixedPoint;
67 using llvm::APInt;
68 using llvm::APSInt;
69 using llvm::APFloat;
70 using llvm::FixedPointSemantics;
71 using llvm::Optional;
72 
73 namespace {
74   struct LValue;
75   class CallStackFrame;
76   class EvalInfo;
77 
78   using SourceLocExprScopeGuard =
79       CurrentSourceLocExprScope::SourceLocExprScopeGuard;
80 
getType(APValue::LValueBase B)81   static QualType getType(APValue::LValueBase B) {
82     return B.getType();
83   }
84 
85   /// Get an LValue path entry, which is known to not be an array index, as a
86   /// field declaration.
getAsField(APValue::LValuePathEntry E)87   static const FieldDecl *getAsField(APValue::LValuePathEntry E) {
88     return dyn_cast_or_null<FieldDecl>(E.getAsBaseOrMember().getPointer());
89   }
90   /// Get an LValue path entry, which is known to not be an array index, as a
91   /// base class declaration.
getAsBaseClass(APValue::LValuePathEntry E)92   static const CXXRecordDecl *getAsBaseClass(APValue::LValuePathEntry E) {
93     return dyn_cast_or_null<CXXRecordDecl>(E.getAsBaseOrMember().getPointer());
94   }
95   /// Determine whether this LValue path entry for a base class names a virtual
96   /// base class.
isVirtualBaseClass(APValue::LValuePathEntry E)97   static bool isVirtualBaseClass(APValue::LValuePathEntry E) {
98     return E.getAsBaseOrMember().getInt();
99   }
100 
101   /// Given an expression, determine the type used to store the result of
102   /// evaluating that expression.
getStorageType(const ASTContext & Ctx,const Expr * E)103   static QualType getStorageType(const ASTContext &Ctx, const Expr *E) {
104     if (E->isPRValue())
105       return E->getType();
106     return Ctx.getLValueReferenceType(E->getType());
107   }
108 
109   /// Given a CallExpr, try to get the alloc_size attribute. May return null.
getAllocSizeAttr(const CallExpr * CE)110   static const AllocSizeAttr *getAllocSizeAttr(const CallExpr *CE) {
111     if (const FunctionDecl *DirectCallee = CE->getDirectCallee())
112       return DirectCallee->getAttr<AllocSizeAttr>();
113     if (const Decl *IndirectCallee = CE->getCalleeDecl())
114       return IndirectCallee->getAttr<AllocSizeAttr>();
115     return nullptr;
116   }
117 
118   /// Attempts to unwrap a CallExpr (with an alloc_size attribute) from an Expr.
119   /// This will look through a single cast.
120   ///
121   /// Returns null if we couldn't unwrap a function with alloc_size.
tryUnwrapAllocSizeCall(const Expr * E)122   static const CallExpr *tryUnwrapAllocSizeCall(const Expr *E) {
123     if (!E->getType()->isPointerType())
124       return nullptr;
125 
126     E = E->IgnoreParens();
127     // If we're doing a variable assignment from e.g. malloc(N), there will
128     // probably be a cast of some kind. In exotic cases, we might also see a
129     // top-level ExprWithCleanups. Ignore them either way.
130     if (const auto *FE = dyn_cast<FullExpr>(E))
131       E = FE->getSubExpr()->IgnoreParens();
132 
133     if (const auto *Cast = dyn_cast<CastExpr>(E))
134       E = Cast->getSubExpr()->IgnoreParens();
135 
136     if (const auto *CE = dyn_cast<CallExpr>(E))
137       return getAllocSizeAttr(CE) ? CE : nullptr;
138     return nullptr;
139   }
140 
141   /// Determines whether or not the given Base contains a call to a function
142   /// with the alloc_size attribute.
isBaseAnAllocSizeCall(APValue::LValueBase Base)143   static bool isBaseAnAllocSizeCall(APValue::LValueBase Base) {
144     const auto *E = Base.dyn_cast<const Expr *>();
145     return E && E->getType()->isPointerType() && tryUnwrapAllocSizeCall(E);
146   }
147 
148   /// Determines whether the given kind of constant expression is only ever
149   /// used for name mangling. If so, it's permitted to reference things that we
150   /// can't generate code for (in particular, dllimported functions).
isForManglingOnly(ConstantExprKind Kind)151   static bool isForManglingOnly(ConstantExprKind Kind) {
152     switch (Kind) {
153     case ConstantExprKind::Normal:
154     case ConstantExprKind::ClassTemplateArgument:
155     case ConstantExprKind::ImmediateInvocation:
156       // Note that non-type template arguments of class type are emitted as
157       // template parameter objects.
158       return false;
159 
160     case ConstantExprKind::NonClassTemplateArgument:
161       return true;
162     }
163     llvm_unreachable("unknown ConstantExprKind");
164   }
165 
isTemplateArgument(ConstantExprKind Kind)166   static bool isTemplateArgument(ConstantExprKind Kind) {
167     switch (Kind) {
168     case ConstantExprKind::Normal:
169     case ConstantExprKind::ImmediateInvocation:
170       return false;
171 
172     case ConstantExprKind::ClassTemplateArgument:
173     case ConstantExprKind::NonClassTemplateArgument:
174       return true;
175     }
176     llvm_unreachable("unknown ConstantExprKind");
177   }
178 
179   /// The bound to claim that an array of unknown bound has.
180   /// The value in MostDerivedArraySize is undefined in this case. So, set it
181   /// to an arbitrary value that's likely to loudly break things if it's used.
182   static const uint64_t AssumedSizeForUnsizedArray =
183       std::numeric_limits<uint64_t>::max() / 2;
184 
185   /// Determines if an LValue with the given LValueBase will have an unsized
186   /// array in its designator.
187   /// Find the path length and type of the most-derived subobject in the given
188   /// path, and find the size of the containing array, if any.
189   static unsigned
findMostDerivedSubobject(ASTContext & Ctx,APValue::LValueBase Base,ArrayRef<APValue::LValuePathEntry> Path,uint64_t & ArraySize,QualType & Type,bool & IsArray,bool & FirstEntryIsUnsizedArray)190   findMostDerivedSubobject(ASTContext &Ctx, APValue::LValueBase Base,
191                            ArrayRef<APValue::LValuePathEntry> Path,
192                            uint64_t &ArraySize, QualType &Type, bool &IsArray,
193                            bool &FirstEntryIsUnsizedArray) {
194     // This only accepts LValueBases from APValues, and APValues don't support
195     // arrays that lack size info.
196     assert(!isBaseAnAllocSizeCall(Base) &&
197            "Unsized arrays shouldn't appear here");
198     unsigned MostDerivedLength = 0;
199     Type = getType(Base);
200 
201     for (unsigned I = 0, N = Path.size(); I != N; ++I) {
202       if (Type->isArrayType()) {
203         const ArrayType *AT = Ctx.getAsArrayType(Type);
204         Type = AT->getElementType();
205         MostDerivedLength = I + 1;
206         IsArray = true;
207 
208         if (auto *CAT = dyn_cast<ConstantArrayType>(AT)) {
209           ArraySize = CAT->getSize().getZExtValue();
210         } else {
211           assert(I == 0 && "unexpected unsized array designator");
212           FirstEntryIsUnsizedArray = true;
213           ArraySize = AssumedSizeForUnsizedArray;
214         }
215       } else if (Type->isAnyComplexType()) {
216         const ComplexType *CT = Type->castAs<ComplexType>();
217         Type = CT->getElementType();
218         ArraySize = 2;
219         MostDerivedLength = I + 1;
220         IsArray = true;
221       } else if (const FieldDecl *FD = getAsField(Path[I])) {
222         Type = FD->getType();
223         ArraySize = 0;
224         MostDerivedLength = I + 1;
225         IsArray = false;
226       } else {
227         // Path[I] describes a base class.
228         ArraySize = 0;
229         IsArray = false;
230       }
231     }
232     return MostDerivedLength;
233   }
234 
235   /// A path from a glvalue to a subobject of that glvalue.
236   struct SubobjectDesignator {
237     /// True if the subobject was named in a manner not supported by C++11. Such
238     /// lvalues can still be folded, but they are not core constant expressions
239     /// and we cannot perform lvalue-to-rvalue conversions on them.
240     unsigned Invalid : 1;
241 
242     /// Is this a pointer one past the end of an object?
243     unsigned IsOnePastTheEnd : 1;
244 
245     /// Indicator of whether the first entry is an unsized array.
246     unsigned FirstEntryIsAnUnsizedArray : 1;
247 
248     /// Indicator of whether the most-derived object is an array element.
249     unsigned MostDerivedIsArrayElement : 1;
250 
251     /// The length of the path to the most-derived object of which this is a
252     /// subobject.
253     unsigned MostDerivedPathLength : 28;
254 
255     /// The size of the array of which the most-derived object is an element.
256     /// This will always be 0 if the most-derived object is not an array
257     /// element. 0 is not an indicator of whether or not the most-derived object
258     /// is an array, however, because 0-length arrays are allowed.
259     ///
260     /// If the current array is an unsized array, the value of this is
261     /// undefined.
262     uint64_t MostDerivedArraySize;
263 
264     /// The type of the most derived object referred to by this address.
265     QualType MostDerivedType;
266 
267     typedef APValue::LValuePathEntry PathEntry;
268 
269     /// The entries on the path from the glvalue to the designated subobject.
270     SmallVector<PathEntry, 8> Entries;
271 
SubobjectDesignator__anonb66d72d20111::SubobjectDesignator272     SubobjectDesignator() : Invalid(true) {}
273 
SubobjectDesignator__anonb66d72d20111::SubobjectDesignator274     explicit SubobjectDesignator(QualType T)
275         : Invalid(false), IsOnePastTheEnd(false),
276           FirstEntryIsAnUnsizedArray(false), MostDerivedIsArrayElement(false),
277           MostDerivedPathLength(0), MostDerivedArraySize(0),
278           MostDerivedType(T) {}
279 
SubobjectDesignator__anonb66d72d20111::SubobjectDesignator280     SubobjectDesignator(ASTContext &Ctx, const APValue &V)
281         : Invalid(!V.isLValue() || !V.hasLValuePath()), IsOnePastTheEnd(false),
282           FirstEntryIsAnUnsizedArray(false), MostDerivedIsArrayElement(false),
283           MostDerivedPathLength(0), MostDerivedArraySize(0) {
284       assert(V.isLValue() && "Non-LValue used to make an LValue designator?");
285       if (!Invalid) {
286         IsOnePastTheEnd = V.isLValueOnePastTheEnd();
287         ArrayRef<PathEntry> VEntries = V.getLValuePath();
288         Entries.insert(Entries.end(), VEntries.begin(), VEntries.end());
289         if (V.getLValueBase()) {
290           bool IsArray = false;
291           bool FirstIsUnsizedArray = false;
292           MostDerivedPathLength = findMostDerivedSubobject(
293               Ctx, V.getLValueBase(), V.getLValuePath(), MostDerivedArraySize,
294               MostDerivedType, IsArray, FirstIsUnsizedArray);
295           MostDerivedIsArrayElement = IsArray;
296           FirstEntryIsAnUnsizedArray = FirstIsUnsizedArray;
297         }
298       }
299     }
300 
truncate__anonb66d72d20111::SubobjectDesignator301     void truncate(ASTContext &Ctx, APValue::LValueBase Base,
302                   unsigned NewLength) {
303       if (Invalid)
304         return;
305 
306       assert(Base && "cannot truncate path for null pointer");
307       assert(NewLength <= Entries.size() && "not a truncation");
308 
309       if (NewLength == Entries.size())
310         return;
311       Entries.resize(NewLength);
312 
313       bool IsArray = false;
314       bool FirstIsUnsizedArray = false;
315       MostDerivedPathLength = findMostDerivedSubobject(
316           Ctx, Base, Entries, MostDerivedArraySize, MostDerivedType, IsArray,
317           FirstIsUnsizedArray);
318       MostDerivedIsArrayElement = IsArray;
319       FirstEntryIsAnUnsizedArray = FirstIsUnsizedArray;
320     }
321 
setInvalid__anonb66d72d20111::SubobjectDesignator322     void setInvalid() {
323       Invalid = true;
324       Entries.clear();
325     }
326 
327     /// Determine whether the most derived subobject is an array without a
328     /// known bound.
isMostDerivedAnUnsizedArray__anonb66d72d20111::SubobjectDesignator329     bool isMostDerivedAnUnsizedArray() const {
330       assert(!Invalid && "Calling this makes no sense on invalid designators");
331       return Entries.size() == 1 && FirstEntryIsAnUnsizedArray;
332     }
333 
334     /// Determine what the most derived array's size is. Results in an assertion
335     /// failure if the most derived array lacks a size.
getMostDerivedArraySize__anonb66d72d20111::SubobjectDesignator336     uint64_t getMostDerivedArraySize() const {
337       assert(!isMostDerivedAnUnsizedArray() && "Unsized array has no size");
338       return MostDerivedArraySize;
339     }
340 
341     /// Determine whether this is a one-past-the-end pointer.
isOnePastTheEnd__anonb66d72d20111::SubobjectDesignator342     bool isOnePastTheEnd() const {
343       assert(!Invalid);
344       if (IsOnePastTheEnd)
345         return true;
346       if (!isMostDerivedAnUnsizedArray() && MostDerivedIsArrayElement &&
347           Entries[MostDerivedPathLength - 1].getAsArrayIndex() ==
348               MostDerivedArraySize)
349         return true;
350       return false;
351     }
352 
353     /// Get the range of valid index adjustments in the form
354     ///   {maximum value that can be subtracted from this pointer,
355     ///    maximum value that can be added to this pointer}
validIndexAdjustments__anonb66d72d20111::SubobjectDesignator356     std::pair<uint64_t, uint64_t> validIndexAdjustments() {
357       if (Invalid || isMostDerivedAnUnsizedArray())
358         return {0, 0};
359 
360       // [expr.add]p4: For the purposes of these operators, a pointer to a
361       // nonarray object behaves the same as a pointer to the first element of
362       // an array of length one with the type of the object as its element type.
363       bool IsArray = MostDerivedPathLength == Entries.size() &&
364                      MostDerivedIsArrayElement;
365       uint64_t ArrayIndex = IsArray ? Entries.back().getAsArrayIndex()
366                                     : (uint64_t)IsOnePastTheEnd;
367       uint64_t ArraySize =
368           IsArray ? getMostDerivedArraySize() : (uint64_t)1;
369       return {ArrayIndex, ArraySize - ArrayIndex};
370     }
371 
372     /// Check that this refers to a valid subobject.
isValidSubobject__anonb66d72d20111::SubobjectDesignator373     bool isValidSubobject() const {
374       if (Invalid)
375         return false;
376       return !isOnePastTheEnd();
377     }
378     /// Check that this refers to a valid subobject, and if not, produce a
379     /// relevant diagnostic and set the designator as invalid.
380     bool checkSubobject(EvalInfo &Info, const Expr *E, CheckSubobjectKind CSK);
381 
382     /// Get the type of the designated object.
getType__anonb66d72d20111::SubobjectDesignator383     QualType getType(ASTContext &Ctx) const {
384       assert(!Invalid && "invalid designator has no subobject type");
385       return MostDerivedPathLength == Entries.size()
386                  ? MostDerivedType
387                  : Ctx.getRecordType(getAsBaseClass(Entries.back()));
388     }
389 
390     /// Update this designator to refer to the first element within this array.
addArrayUnchecked__anonb66d72d20111::SubobjectDesignator391     void addArrayUnchecked(const ConstantArrayType *CAT) {
392       Entries.push_back(PathEntry::ArrayIndex(0));
393 
394       // This is a most-derived object.
395       MostDerivedType = CAT->getElementType();
396       MostDerivedIsArrayElement = true;
397       MostDerivedArraySize = CAT->getSize().getZExtValue();
398       MostDerivedPathLength = Entries.size();
399     }
400     /// Update this designator to refer to the first element within the array of
401     /// elements of type T. This is an array of unknown size.
addUnsizedArrayUnchecked__anonb66d72d20111::SubobjectDesignator402     void addUnsizedArrayUnchecked(QualType ElemTy) {
403       Entries.push_back(PathEntry::ArrayIndex(0));
404 
405       MostDerivedType = ElemTy;
406       MostDerivedIsArrayElement = true;
407       // The value in MostDerivedArraySize is undefined in this case. So, set it
408       // to an arbitrary value that's likely to loudly break things if it's
409       // used.
410       MostDerivedArraySize = AssumedSizeForUnsizedArray;
411       MostDerivedPathLength = Entries.size();
412     }
413     /// Update this designator to refer to the given base or member of this
414     /// object.
addDeclUnchecked__anonb66d72d20111::SubobjectDesignator415     void addDeclUnchecked(const Decl *D, bool Virtual = false) {
416       Entries.push_back(APValue::BaseOrMemberType(D, Virtual));
417 
418       // If this isn't a base class, it's a new most-derived object.
419       if (const FieldDecl *FD = dyn_cast<FieldDecl>(D)) {
420         MostDerivedType = FD->getType();
421         MostDerivedIsArrayElement = false;
422         MostDerivedArraySize = 0;
423         MostDerivedPathLength = Entries.size();
424       }
425     }
426     /// Update this designator to refer to the given complex component.
addComplexUnchecked__anonb66d72d20111::SubobjectDesignator427     void addComplexUnchecked(QualType EltTy, bool Imag) {
428       Entries.push_back(PathEntry::ArrayIndex(Imag));
429 
430       // This is technically a most-derived object, though in practice this
431       // is unlikely to matter.
432       MostDerivedType = EltTy;
433       MostDerivedIsArrayElement = true;
434       MostDerivedArraySize = 2;
435       MostDerivedPathLength = Entries.size();
436     }
437     void diagnoseUnsizedArrayPointerArithmetic(EvalInfo &Info, const Expr *E);
438     void diagnosePointerArithmetic(EvalInfo &Info, const Expr *E,
439                                    const APSInt &N);
440     /// Add N to the address of this subobject.
adjustIndex__anonb66d72d20111::SubobjectDesignator441     void adjustIndex(EvalInfo &Info, const Expr *E, APSInt N) {
442       if (Invalid || !N) return;
443       uint64_t TruncatedN = N.extOrTrunc(64).getZExtValue();
444       if (isMostDerivedAnUnsizedArray()) {
445         diagnoseUnsizedArrayPointerArithmetic(Info, E);
446         // Can't verify -- trust that the user is doing the right thing (or if
447         // not, trust that the caller will catch the bad behavior).
448         // FIXME: Should we reject if this overflows, at least?
449         Entries.back() = PathEntry::ArrayIndex(
450             Entries.back().getAsArrayIndex() + TruncatedN);
451         return;
452       }
453 
454       // [expr.add]p4: For the purposes of these operators, a pointer to a
455       // nonarray object behaves the same as a pointer to the first element of
456       // an array of length one with the type of the object as its element type.
457       bool IsArray = MostDerivedPathLength == Entries.size() &&
458                      MostDerivedIsArrayElement;
459       uint64_t ArrayIndex = IsArray ? Entries.back().getAsArrayIndex()
460                                     : (uint64_t)IsOnePastTheEnd;
461       uint64_t ArraySize =
462           IsArray ? getMostDerivedArraySize() : (uint64_t)1;
463 
464       if (N < -(int64_t)ArrayIndex || N > ArraySize - ArrayIndex) {
465         // Calculate the actual index in a wide enough type, so we can include
466         // it in the note.
467         N = N.extend(std::max<unsigned>(N.getBitWidth() + 1, 65));
468         (llvm::APInt&)N += ArrayIndex;
469         assert(N.ugt(ArraySize) && "bounds check failed for in-bounds index");
470         diagnosePointerArithmetic(Info, E, N);
471         setInvalid();
472         return;
473       }
474 
475       ArrayIndex += TruncatedN;
476       assert(ArrayIndex <= ArraySize &&
477              "bounds check succeeded for out-of-bounds index");
478 
479       if (IsArray)
480         Entries.back() = PathEntry::ArrayIndex(ArrayIndex);
481       else
482         IsOnePastTheEnd = (ArrayIndex != 0);
483     }
484   };
485 
486   /// A scope at the end of which an object can need to be destroyed.
487   enum class ScopeKind {
488     Block,
489     FullExpression,
490     Call
491   };
492 
493   /// A reference to a particular call and its arguments.
494   struct CallRef {
CallRef__anonb66d72d20111::CallRef495     CallRef() : OrigCallee(), CallIndex(0), Version() {}
CallRef__anonb66d72d20111::CallRef496     CallRef(const FunctionDecl *Callee, unsigned CallIndex, unsigned Version)
497         : OrigCallee(Callee), CallIndex(CallIndex), Version(Version) {}
498 
operator bool__anonb66d72d20111::CallRef499     explicit operator bool() const { return OrigCallee; }
500 
501     /// Get the parameter that the caller initialized, corresponding to the
502     /// given parameter in the callee.
getOrigParam__anonb66d72d20111::CallRef503     const ParmVarDecl *getOrigParam(const ParmVarDecl *PVD) const {
504       return OrigCallee ? OrigCallee->getParamDecl(PVD->getFunctionScopeIndex())
505                         : PVD;
506     }
507 
508     /// The callee at the point where the arguments were evaluated. This might
509     /// be different from the actual callee (a different redeclaration, or a
510     /// virtual override), but this function's parameters are the ones that
511     /// appear in the parameter map.
512     const FunctionDecl *OrigCallee;
513     /// The call index of the frame that holds the argument values.
514     unsigned CallIndex;
515     /// The version of the parameters corresponding to this call.
516     unsigned Version;
517   };
518 
519   /// A stack frame in the constexpr call stack.
520   class CallStackFrame : public interp::Frame {
521   public:
522     EvalInfo &Info;
523 
524     /// Parent - The caller of this stack frame.
525     CallStackFrame *Caller;
526 
527     /// Callee - The function which was called.
528     const FunctionDecl *Callee;
529 
530     /// This - The binding for the this pointer in this call, if any.
531     const LValue *This;
532 
533     /// Information on how to find the arguments to this call. Our arguments
534     /// are stored in our parent's CallStackFrame, using the ParmVarDecl* as a
535     /// key and this value as the version.
536     CallRef Arguments;
537 
538     /// Source location information about the default argument or default
539     /// initializer expression we're evaluating, if any.
540     CurrentSourceLocExprScope CurSourceLocExprScope;
541 
542     // Note that we intentionally use std::map here so that references to
543     // values are stable.
544     typedef std::pair<const void *, unsigned> MapKeyTy;
545     typedef std::map<MapKeyTy, APValue> MapTy;
546     /// Temporaries - Temporary lvalues materialized within this stack frame.
547     MapTy Temporaries;
548 
549     /// CallLoc - The location of the call expression for this call.
550     SourceLocation CallLoc;
551 
552     /// Index - The call index of this call.
553     unsigned Index;
554 
555     /// The stack of integers for tracking version numbers for temporaries.
556     SmallVector<unsigned, 2> TempVersionStack = {1};
557     unsigned CurTempVersion = TempVersionStack.back();
558 
getTempVersion() const559     unsigned getTempVersion() const { return TempVersionStack.back(); }
560 
pushTempVersion()561     void pushTempVersion() {
562       TempVersionStack.push_back(++CurTempVersion);
563     }
564 
popTempVersion()565     void popTempVersion() {
566       TempVersionStack.pop_back();
567     }
568 
createCall(const FunctionDecl * Callee)569     CallRef createCall(const FunctionDecl *Callee) {
570       return {Callee, Index, ++CurTempVersion};
571     }
572 
573     // FIXME: Adding this to every 'CallStackFrame' may have a nontrivial impact
574     // on the overall stack usage of deeply-recursing constexpr evaluations.
575     // (We should cache this map rather than recomputing it repeatedly.)
576     // But let's try this and see how it goes; we can look into caching the map
577     // as a later change.
578 
579     /// LambdaCaptureFields - Mapping from captured variables/this to
580     /// corresponding data members in the closure class.
581     llvm::DenseMap<const VarDecl *, FieldDecl *> LambdaCaptureFields;
582     FieldDecl *LambdaThisCaptureField;
583 
584     CallStackFrame(EvalInfo &Info, SourceLocation CallLoc,
585                    const FunctionDecl *Callee, const LValue *This,
586                    CallRef Arguments);
587     ~CallStackFrame();
588 
589     // Return the temporary for Key whose version number is Version.
getTemporary(const void * Key,unsigned Version)590     APValue *getTemporary(const void *Key, unsigned Version) {
591       MapKeyTy KV(Key, Version);
592       auto LB = Temporaries.lower_bound(KV);
593       if (LB != Temporaries.end() && LB->first == KV)
594         return &LB->second;
595       // Pair (Key,Version) wasn't found in the map. Check that no elements
596       // in the map have 'Key' as their key.
597       assert((LB == Temporaries.end() || LB->first.first != Key) &&
598              (LB == Temporaries.begin() || std::prev(LB)->first.first != Key) &&
599              "Element with key 'Key' found in map");
600       return nullptr;
601     }
602 
603     // Return the current temporary for Key in the map.
getCurrentTemporary(const void * Key)604     APValue *getCurrentTemporary(const void *Key) {
605       auto UB = Temporaries.upper_bound(MapKeyTy(Key, UINT_MAX));
606       if (UB != Temporaries.begin() && std::prev(UB)->first.first == Key)
607         return &std::prev(UB)->second;
608       return nullptr;
609     }
610 
611     // Return the version number of the current temporary for Key.
getCurrentTemporaryVersion(const void * Key) const612     unsigned getCurrentTemporaryVersion(const void *Key) const {
613       auto UB = Temporaries.upper_bound(MapKeyTy(Key, UINT_MAX));
614       if (UB != Temporaries.begin() && std::prev(UB)->first.first == Key)
615         return std::prev(UB)->first.second;
616       return 0;
617     }
618 
619     /// Allocate storage for an object of type T in this stack frame.
620     /// Populates LV with a handle to the created object. Key identifies
621     /// the temporary within the stack frame, and must not be reused without
622     /// bumping the temporary version number.
623     template<typename KeyT>
624     APValue &createTemporary(const KeyT *Key, QualType T,
625                              ScopeKind Scope, LValue &LV);
626 
627     /// Allocate storage for a parameter of a function call made in this frame.
628     APValue &createParam(CallRef Args, const ParmVarDecl *PVD, LValue &LV);
629 
630     void describe(llvm::raw_ostream &OS) override;
631 
getCaller() const632     Frame *getCaller() const override { return Caller; }
getCallLocation() const633     SourceLocation getCallLocation() const override { return CallLoc; }
getCallee() const634     const FunctionDecl *getCallee() const override { return Callee; }
635 
isStdFunction() const636     bool isStdFunction() const {
637       for (const DeclContext *DC = Callee; DC; DC = DC->getParent())
638         if (DC->isStdNamespace())
639           return true;
640       return false;
641     }
642 
643   private:
644     APValue &createLocal(APValue::LValueBase Base, const void *Key, QualType T,
645                          ScopeKind Scope);
646   };
647 
648   /// Temporarily override 'this'.
649   class ThisOverrideRAII {
650   public:
ThisOverrideRAII(CallStackFrame & Frame,const LValue * NewThis,bool Enable)651     ThisOverrideRAII(CallStackFrame &Frame, const LValue *NewThis, bool Enable)
652         : Frame(Frame), OldThis(Frame.This) {
653       if (Enable)
654         Frame.This = NewThis;
655     }
~ThisOverrideRAII()656     ~ThisOverrideRAII() {
657       Frame.This = OldThis;
658     }
659   private:
660     CallStackFrame &Frame;
661     const LValue *OldThis;
662   };
663 }
664 
665 static bool HandleDestruction(EvalInfo &Info, const Expr *E,
666                               const LValue &This, QualType ThisType);
667 static bool HandleDestruction(EvalInfo &Info, SourceLocation Loc,
668                               APValue::LValueBase LVBase, APValue &Value,
669                               QualType T);
670 
671 namespace {
672   /// A cleanup, and a flag indicating whether it is lifetime-extended.
673   class Cleanup {
674     llvm::PointerIntPair<APValue*, 2, ScopeKind> Value;
675     APValue::LValueBase Base;
676     QualType T;
677 
678   public:
Cleanup(APValue * Val,APValue::LValueBase Base,QualType T,ScopeKind Scope)679     Cleanup(APValue *Val, APValue::LValueBase Base, QualType T,
680             ScopeKind Scope)
681         : Value(Val, Scope), Base(Base), T(T) {}
682 
683     /// Determine whether this cleanup should be performed at the end of the
684     /// given kind of scope.
isDestroyedAtEndOf(ScopeKind K) const685     bool isDestroyedAtEndOf(ScopeKind K) const {
686       return (int)Value.getInt() >= (int)K;
687     }
endLifetime(EvalInfo & Info,bool RunDestructors)688     bool endLifetime(EvalInfo &Info, bool RunDestructors) {
689       if (RunDestructors) {
690         SourceLocation Loc;
691         if (const ValueDecl *VD = Base.dyn_cast<const ValueDecl*>())
692           Loc = VD->getLocation();
693         else if (const Expr *E = Base.dyn_cast<const Expr*>())
694           Loc = E->getExprLoc();
695         return HandleDestruction(Info, Loc, Base, *Value.getPointer(), T);
696       }
697       *Value.getPointer() = APValue();
698       return true;
699     }
700 
hasSideEffect()701     bool hasSideEffect() {
702       return T.isDestructedType();
703     }
704   };
705 
706   /// A reference to an object whose construction we are currently evaluating.
707   struct ObjectUnderConstruction {
708     APValue::LValueBase Base;
709     ArrayRef<APValue::LValuePathEntry> Path;
operator ==(const ObjectUnderConstruction & LHS,const ObjectUnderConstruction & RHS)710     friend bool operator==(const ObjectUnderConstruction &LHS,
711                            const ObjectUnderConstruction &RHS) {
712       return LHS.Base == RHS.Base && LHS.Path == RHS.Path;
713     }
hash_value(const ObjectUnderConstruction & Obj)714     friend llvm::hash_code hash_value(const ObjectUnderConstruction &Obj) {
715       return llvm::hash_combine(Obj.Base, Obj.Path);
716     }
717   };
718   enum class ConstructionPhase {
719     None,
720     Bases,
721     AfterBases,
722     AfterFields,
723     Destroying,
724     DestroyingBases
725   };
726 }
727 
728 namespace llvm {
729 template<> struct DenseMapInfo<ObjectUnderConstruction> {
730   using Base = DenseMapInfo<APValue::LValueBase>;
getEmptyKeyllvm::DenseMapInfo731   static ObjectUnderConstruction getEmptyKey() {
732     return {Base::getEmptyKey(), {}}; }
getTombstoneKeyllvm::DenseMapInfo733   static ObjectUnderConstruction getTombstoneKey() {
734     return {Base::getTombstoneKey(), {}};
735   }
getHashValuellvm::DenseMapInfo736   static unsigned getHashValue(const ObjectUnderConstruction &Object) {
737     return hash_value(Object);
738   }
isEqualllvm::DenseMapInfo739   static bool isEqual(const ObjectUnderConstruction &LHS,
740                       const ObjectUnderConstruction &RHS) {
741     return LHS == RHS;
742   }
743 };
744 }
745 
746 namespace {
747   /// A dynamically-allocated heap object.
748   struct DynAlloc {
749     /// The value of this heap-allocated object.
750     APValue Value;
751     /// The allocating expression; used for diagnostics. Either a CXXNewExpr
752     /// or a CallExpr (the latter is for direct calls to operator new inside
753     /// std::allocator<T>::allocate).
754     const Expr *AllocExpr = nullptr;
755 
756     enum Kind {
757       New,
758       ArrayNew,
759       StdAllocator
760     };
761 
762     /// Get the kind of the allocation. This must match between allocation
763     /// and deallocation.
getKind__anonb66d72d20311::DynAlloc764     Kind getKind() const {
765       if (auto *NE = dyn_cast<CXXNewExpr>(AllocExpr))
766         return NE->isArray() ? ArrayNew : New;
767       assert(isa<CallExpr>(AllocExpr));
768       return StdAllocator;
769     }
770   };
771 
772   struct DynAllocOrder {
operator ()__anonb66d72d20311::DynAllocOrder773     bool operator()(DynamicAllocLValue L, DynamicAllocLValue R) const {
774       return L.getIndex() < R.getIndex();
775     }
776   };
777 
778   /// EvalInfo - This is a private struct used by the evaluator to capture
779   /// information about a subexpression as it is folded.  It retains information
780   /// about the AST context, but also maintains information about the folded
781   /// expression.
782   ///
783   /// If an expression could be evaluated, it is still possible it is not a C
784   /// "integer constant expression" or constant expression.  If not, this struct
785   /// captures information about how and why not.
786   ///
787   /// One bit of information passed *into* the request for constant folding
788   /// indicates whether the subexpression is "evaluated" or not according to C
789   /// rules.  For example, the RHS of (0 && foo()) is not evaluated.  We can
790   /// evaluate the expression regardless of what the RHS is, but C only allows
791   /// certain things in certain situations.
792   class EvalInfo : public interp::State {
793   public:
794     ASTContext &Ctx;
795 
796     /// EvalStatus - Contains information about the evaluation.
797     Expr::EvalStatus &EvalStatus;
798 
799     /// CurrentCall - The top of the constexpr call stack.
800     CallStackFrame *CurrentCall;
801 
802     /// CallStackDepth - The number of calls in the call stack right now.
803     unsigned CallStackDepth;
804 
805     /// NextCallIndex - The next call index to assign.
806     unsigned NextCallIndex;
807 
808     /// StepsLeft - The remaining number of evaluation steps we're permitted
809     /// to perform. This is essentially a limit for the number of statements
810     /// we will evaluate.
811     unsigned StepsLeft;
812 
813     /// Enable the experimental new constant interpreter. If an expression is
814     /// not supported by the interpreter, an error is triggered.
815     bool EnableNewConstInterp;
816 
817     /// BottomFrame - The frame in which evaluation started. This must be
818     /// initialized after CurrentCall and CallStackDepth.
819     CallStackFrame BottomFrame;
820 
821     /// A stack of values whose lifetimes end at the end of some surrounding
822     /// evaluation frame.
823     llvm::SmallVector<Cleanup, 16> CleanupStack;
824 
825     /// EvaluatingDecl - This is the declaration whose initializer is being
826     /// evaluated, if any.
827     APValue::LValueBase EvaluatingDecl;
828 
829     enum class EvaluatingDeclKind {
830       None,
831       /// We're evaluating the construction of EvaluatingDecl.
832       Ctor,
833       /// We're evaluating the destruction of EvaluatingDecl.
834       Dtor,
835     };
836     EvaluatingDeclKind IsEvaluatingDecl = EvaluatingDeclKind::None;
837 
838     /// EvaluatingDeclValue - This is the value being constructed for the
839     /// declaration whose initializer is being evaluated, if any.
840     APValue *EvaluatingDeclValue;
841 
842     /// Set of objects that are currently being constructed.
843     llvm::DenseMap<ObjectUnderConstruction, ConstructionPhase>
844         ObjectsUnderConstruction;
845 
846     /// Current heap allocations, along with the location where each was
847     /// allocated. We use std::map here because we need stable addresses
848     /// for the stored APValues.
849     std::map<DynamicAllocLValue, DynAlloc, DynAllocOrder> HeapAllocs;
850 
851     /// The number of heap allocations performed so far in this evaluation.
852     unsigned NumHeapAllocs = 0;
853 
854     struct EvaluatingConstructorRAII {
855       EvalInfo &EI;
856       ObjectUnderConstruction Object;
857       bool DidInsert;
EvaluatingConstructorRAII__anonb66d72d20311::EvalInfo::EvaluatingConstructorRAII858       EvaluatingConstructorRAII(EvalInfo &EI, ObjectUnderConstruction Object,
859                                 bool HasBases)
860           : EI(EI), Object(Object) {
861         DidInsert =
862             EI.ObjectsUnderConstruction
863                 .insert({Object, HasBases ? ConstructionPhase::Bases
864                                           : ConstructionPhase::AfterBases})
865                 .second;
866       }
finishedConstructingBases__anonb66d72d20311::EvalInfo::EvaluatingConstructorRAII867       void finishedConstructingBases() {
868         EI.ObjectsUnderConstruction[Object] = ConstructionPhase::AfterBases;
869       }
finishedConstructingFields__anonb66d72d20311::EvalInfo::EvaluatingConstructorRAII870       void finishedConstructingFields() {
871         EI.ObjectsUnderConstruction[Object] = ConstructionPhase::AfterFields;
872       }
~EvaluatingConstructorRAII__anonb66d72d20311::EvalInfo::EvaluatingConstructorRAII873       ~EvaluatingConstructorRAII() {
874         if (DidInsert) EI.ObjectsUnderConstruction.erase(Object);
875       }
876     };
877 
878     struct EvaluatingDestructorRAII {
879       EvalInfo &EI;
880       ObjectUnderConstruction Object;
881       bool DidInsert;
EvaluatingDestructorRAII__anonb66d72d20311::EvalInfo::EvaluatingDestructorRAII882       EvaluatingDestructorRAII(EvalInfo &EI, ObjectUnderConstruction Object)
883           : EI(EI), Object(Object) {
884         DidInsert = EI.ObjectsUnderConstruction
885                         .insert({Object, ConstructionPhase::Destroying})
886                         .second;
887       }
startedDestroyingBases__anonb66d72d20311::EvalInfo::EvaluatingDestructorRAII888       void startedDestroyingBases() {
889         EI.ObjectsUnderConstruction[Object] =
890             ConstructionPhase::DestroyingBases;
891       }
~EvaluatingDestructorRAII__anonb66d72d20311::EvalInfo::EvaluatingDestructorRAII892       ~EvaluatingDestructorRAII() {
893         if (DidInsert)
894           EI.ObjectsUnderConstruction.erase(Object);
895       }
896     };
897 
898     ConstructionPhase
isEvaluatingCtorDtor(APValue::LValueBase Base,ArrayRef<APValue::LValuePathEntry> Path)899     isEvaluatingCtorDtor(APValue::LValueBase Base,
900                          ArrayRef<APValue::LValuePathEntry> Path) {
901       return ObjectsUnderConstruction.lookup({Base, Path});
902     }
903 
904     /// If we're currently speculatively evaluating, the outermost call stack
905     /// depth at which we can mutate state, otherwise 0.
906     unsigned SpeculativeEvaluationDepth = 0;
907 
908     /// The current array initialization index, if we're performing array
909     /// initialization.
910     uint64_t ArrayInitIndex = -1;
911 
912     /// HasActiveDiagnostic - Was the previous diagnostic stored? If so, further
913     /// notes attached to it will also be stored, otherwise they will not be.
914     bool HasActiveDiagnostic;
915 
916     /// Have we emitted a diagnostic explaining why we couldn't constant
917     /// fold (not just why it's not strictly a constant expression)?
918     bool HasFoldFailureDiagnostic;
919 
920     /// Whether or not we're in a context where the front end requires a
921     /// constant value.
922     bool InConstantContext;
923 
924     /// Whether we're checking that an expression is a potential constant
925     /// expression. If so, do not fail on constructs that could become constant
926     /// later on (such as a use of an undefined global).
927     bool CheckingPotentialConstantExpression = false;
928 
929     /// Whether we're checking for an expression that has undefined behavior.
930     /// If so, we will produce warnings if we encounter an operation that is
931     /// always undefined.
932     ///
933     /// Note that we still need to evaluate the expression normally when this
934     /// is set; this is used when evaluating ICEs in C.
935     bool CheckingForUndefinedBehavior = false;
936 
937     enum EvaluationMode {
938       /// Evaluate as a constant expression. Stop if we find that the expression
939       /// is not a constant expression.
940       EM_ConstantExpression,
941 
942       /// Evaluate as a constant expression. Stop if we find that the expression
943       /// is not a constant expression. Some expressions can be retried in the
944       /// optimizer if we don't constant fold them here, but in an unevaluated
945       /// context we try to fold them immediately since the optimizer never
946       /// gets a chance to look at it.
947       EM_ConstantExpressionUnevaluated,
948 
949       /// Fold the expression to a constant. Stop if we hit a side-effect that
950       /// we can't model.
951       EM_ConstantFold,
952 
953       /// Evaluate in any way we know how. Don't worry about side-effects that
954       /// can't be modeled.
955       EM_IgnoreSideEffects,
956     } EvalMode;
957 
958     /// Are we checking whether the expression is a potential constant
959     /// expression?
checkingPotentialConstantExpression() const960     bool checkingPotentialConstantExpression() const override  {
961       return CheckingPotentialConstantExpression;
962     }
963 
964     /// Are we checking an expression for overflow?
965     // FIXME: We should check for any kind of undefined or suspicious behavior
966     // in such constructs, not just overflow.
checkingForUndefinedBehavior() const967     bool checkingForUndefinedBehavior() const override {
968       return CheckingForUndefinedBehavior;
969     }
970 
EvalInfo(const ASTContext & C,Expr::EvalStatus & S,EvaluationMode Mode)971     EvalInfo(const ASTContext &C, Expr::EvalStatus &S, EvaluationMode Mode)
972         : Ctx(const_cast<ASTContext &>(C)), EvalStatus(S), CurrentCall(nullptr),
973           CallStackDepth(0), NextCallIndex(1),
974           StepsLeft(C.getLangOpts().ConstexprStepLimit),
975           EnableNewConstInterp(C.getLangOpts().EnableNewConstInterp),
976           BottomFrame(*this, SourceLocation(), nullptr, nullptr, CallRef()),
977           EvaluatingDecl((const ValueDecl *)nullptr),
978           EvaluatingDeclValue(nullptr), HasActiveDiagnostic(false),
979           HasFoldFailureDiagnostic(false), InConstantContext(false),
980           EvalMode(Mode) {}
981 
~EvalInfo()982     ~EvalInfo() {
983       discardCleanups();
984     }
985 
setEvaluatingDecl(APValue::LValueBase Base,APValue & Value,EvaluatingDeclKind EDK=EvaluatingDeclKind::Ctor)986     void setEvaluatingDecl(APValue::LValueBase Base, APValue &Value,
987                            EvaluatingDeclKind EDK = EvaluatingDeclKind::Ctor) {
988       EvaluatingDecl = Base;
989       IsEvaluatingDecl = EDK;
990       EvaluatingDeclValue = &Value;
991     }
992 
CheckCallLimit(SourceLocation Loc)993     bool CheckCallLimit(SourceLocation Loc) {
994       // Don't perform any constexpr calls (other than the call we're checking)
995       // when checking a potential constant expression.
996       if (checkingPotentialConstantExpression() && CallStackDepth > 1)
997         return false;
998       if (NextCallIndex == 0) {
999         // NextCallIndex has wrapped around.
1000         FFDiag(Loc, diag::note_constexpr_call_limit_exceeded);
1001         return false;
1002       }
1003       if (CallStackDepth <= getLangOpts().ConstexprCallDepth)
1004         return true;
1005       FFDiag(Loc, diag::note_constexpr_depth_limit_exceeded)
1006         << getLangOpts().ConstexprCallDepth;
1007       return false;
1008     }
1009 
1010     std::pair<CallStackFrame *, unsigned>
getCallFrameAndDepth(unsigned CallIndex)1011     getCallFrameAndDepth(unsigned CallIndex) {
1012       assert(CallIndex && "no call index in getCallFrameAndDepth");
1013       // We will eventually hit BottomFrame, which has Index 1, so Frame can't
1014       // be null in this loop.
1015       unsigned Depth = CallStackDepth;
1016       CallStackFrame *Frame = CurrentCall;
1017       while (Frame->Index > CallIndex) {
1018         Frame = Frame->Caller;
1019         --Depth;
1020       }
1021       if (Frame->Index == CallIndex)
1022         return {Frame, Depth};
1023       return {nullptr, 0};
1024     }
1025 
nextStep(const Stmt * S)1026     bool nextStep(const Stmt *S) {
1027       if (!StepsLeft) {
1028         FFDiag(S->getBeginLoc(), diag::note_constexpr_step_limit_exceeded);
1029         return false;
1030       }
1031       --StepsLeft;
1032       return true;
1033     }
1034 
1035     APValue *createHeapAlloc(const Expr *E, QualType T, LValue &LV);
1036 
lookupDynamicAlloc(DynamicAllocLValue DA)1037     Optional<DynAlloc*> lookupDynamicAlloc(DynamicAllocLValue DA) {
1038       Optional<DynAlloc*> Result;
1039       auto It = HeapAllocs.find(DA);
1040       if (It != HeapAllocs.end())
1041         Result = &It->second;
1042       return Result;
1043     }
1044 
1045     /// Get the allocated storage for the given parameter of the given call.
getParamSlot(CallRef Call,const ParmVarDecl * PVD)1046     APValue *getParamSlot(CallRef Call, const ParmVarDecl *PVD) {
1047       CallStackFrame *Frame = getCallFrameAndDepth(Call.CallIndex).first;
1048       return Frame ? Frame->getTemporary(Call.getOrigParam(PVD), Call.Version)
1049                    : nullptr;
1050     }
1051 
1052     /// Information about a stack frame for std::allocator<T>::[de]allocate.
1053     struct StdAllocatorCaller {
1054       unsigned FrameIndex;
1055       QualType ElemType;
operator bool__anonb66d72d20311::EvalInfo::StdAllocatorCaller1056       explicit operator bool() const { return FrameIndex != 0; };
1057     };
1058 
getStdAllocatorCaller(StringRef FnName) const1059     StdAllocatorCaller getStdAllocatorCaller(StringRef FnName) const {
1060       for (const CallStackFrame *Call = CurrentCall; Call != &BottomFrame;
1061            Call = Call->Caller) {
1062         const auto *MD = dyn_cast_or_null<CXXMethodDecl>(Call->Callee);
1063         if (!MD)
1064           continue;
1065         const IdentifierInfo *FnII = MD->getIdentifier();
1066         if (!FnII || !FnII->isStr(FnName))
1067           continue;
1068 
1069         const auto *CTSD =
1070             dyn_cast<ClassTemplateSpecializationDecl>(MD->getParent());
1071         if (!CTSD)
1072           continue;
1073 
1074         const IdentifierInfo *ClassII = CTSD->getIdentifier();
1075         const TemplateArgumentList &TAL = CTSD->getTemplateArgs();
1076         if (CTSD->isInStdNamespace() && ClassII &&
1077             ClassII->isStr("allocator") && TAL.size() >= 1 &&
1078             TAL[0].getKind() == TemplateArgument::Type)
1079           return {Call->Index, TAL[0].getAsType()};
1080       }
1081 
1082       return {};
1083     }
1084 
performLifetimeExtension()1085     void performLifetimeExtension() {
1086       // Disable the cleanups for lifetime-extended temporaries.
1087       CleanupStack.erase(std::remove_if(CleanupStack.begin(),
1088                                         CleanupStack.end(),
1089                                         [](Cleanup &C) {
1090                                           return !C.isDestroyedAtEndOf(
1091                                               ScopeKind::FullExpression);
1092                                         }),
1093                          CleanupStack.end());
1094      }
1095 
1096     /// Throw away any remaining cleanups at the end of evaluation. If any
1097     /// cleanups would have had a side-effect, note that as an unmodeled
1098     /// side-effect and return false. Otherwise, return true.
discardCleanups()1099     bool discardCleanups() {
1100       for (Cleanup &C : CleanupStack) {
1101         if (C.hasSideEffect() && !noteSideEffect()) {
1102           CleanupStack.clear();
1103           return false;
1104         }
1105       }
1106       CleanupStack.clear();
1107       return true;
1108     }
1109 
1110   private:
getCurrentFrame()1111     interp::Frame *getCurrentFrame() override { return CurrentCall; }
getBottomFrame() const1112     const interp::Frame *getBottomFrame() const override { return &BottomFrame; }
1113 
hasActiveDiagnostic()1114     bool hasActiveDiagnostic() override { return HasActiveDiagnostic; }
setActiveDiagnostic(bool Flag)1115     void setActiveDiagnostic(bool Flag) override { HasActiveDiagnostic = Flag; }
1116 
setFoldFailureDiagnostic(bool Flag)1117     void setFoldFailureDiagnostic(bool Flag) override {
1118       HasFoldFailureDiagnostic = Flag;
1119     }
1120 
getEvalStatus() const1121     Expr::EvalStatus &getEvalStatus() const override { return EvalStatus; }
1122 
getCtx() const1123     ASTContext &getCtx() const override { return Ctx; }
1124 
1125     // If we have a prior diagnostic, it will be noting that the expression
1126     // isn't a constant expression. This diagnostic is more important,
1127     // unless we require this evaluation to produce a constant expression.
1128     //
1129     // FIXME: We might want to show both diagnostics to the user in
1130     // EM_ConstantFold mode.
hasPriorDiagnostic()1131     bool hasPriorDiagnostic() override {
1132       if (!EvalStatus.Diag->empty()) {
1133         switch (EvalMode) {
1134         case EM_ConstantFold:
1135         case EM_IgnoreSideEffects:
1136           if (!HasFoldFailureDiagnostic)
1137             break;
1138           // We've already failed to fold something. Keep that diagnostic.
1139           LLVM_FALLTHROUGH;
1140         case EM_ConstantExpression:
1141         case EM_ConstantExpressionUnevaluated:
1142           setActiveDiagnostic(false);
1143           return true;
1144         }
1145       }
1146       return false;
1147     }
1148 
getCallStackDepth()1149     unsigned getCallStackDepth() override { return CallStackDepth; }
1150 
1151   public:
1152     /// Should we continue evaluation after encountering a side-effect that we
1153     /// couldn't model?
keepEvaluatingAfterSideEffect()1154     bool keepEvaluatingAfterSideEffect() {
1155       switch (EvalMode) {
1156       case EM_IgnoreSideEffects:
1157         return true;
1158 
1159       case EM_ConstantExpression:
1160       case EM_ConstantExpressionUnevaluated:
1161       case EM_ConstantFold:
1162         // By default, assume any side effect might be valid in some other
1163         // evaluation of this expression from a different context.
1164         return checkingPotentialConstantExpression() ||
1165                checkingForUndefinedBehavior();
1166       }
1167       llvm_unreachable("Missed EvalMode case");
1168     }
1169 
1170     /// Note that we have had a side-effect, and determine whether we should
1171     /// keep evaluating.
noteSideEffect()1172     bool noteSideEffect() {
1173       EvalStatus.HasSideEffects = true;
1174       return keepEvaluatingAfterSideEffect();
1175     }
1176 
1177     /// Should we continue evaluation after encountering undefined behavior?
keepEvaluatingAfterUndefinedBehavior()1178     bool keepEvaluatingAfterUndefinedBehavior() {
1179       switch (EvalMode) {
1180       case EM_IgnoreSideEffects:
1181       case EM_ConstantFold:
1182         return true;
1183 
1184       case EM_ConstantExpression:
1185       case EM_ConstantExpressionUnevaluated:
1186         return checkingForUndefinedBehavior();
1187       }
1188       llvm_unreachable("Missed EvalMode case");
1189     }
1190 
1191     /// Note that we hit something that was technically undefined behavior, but
1192     /// that we can evaluate past it (such as signed overflow or floating-point
1193     /// division by zero.)
noteUndefinedBehavior()1194     bool noteUndefinedBehavior() override {
1195       EvalStatus.HasUndefinedBehavior = true;
1196       return keepEvaluatingAfterUndefinedBehavior();
1197     }
1198 
1199     /// Should we continue evaluation as much as possible after encountering a
1200     /// construct which can't be reduced to a value?
keepEvaluatingAfterFailure() const1201     bool keepEvaluatingAfterFailure() const override {
1202       if (!StepsLeft)
1203         return false;
1204 
1205       switch (EvalMode) {
1206       case EM_ConstantExpression:
1207       case EM_ConstantExpressionUnevaluated:
1208       case EM_ConstantFold:
1209       case EM_IgnoreSideEffects:
1210         return checkingPotentialConstantExpression() ||
1211                checkingForUndefinedBehavior();
1212       }
1213       llvm_unreachable("Missed EvalMode case");
1214     }
1215 
1216     /// Notes that we failed to evaluate an expression that other expressions
1217     /// directly depend on, and determine if we should keep evaluating. This
1218     /// should only be called if we actually intend to keep evaluating.
1219     ///
1220     /// Call noteSideEffect() instead if we may be able to ignore the value that
1221     /// we failed to evaluate, e.g. if we failed to evaluate Foo() in:
1222     ///
1223     /// (Foo(), 1)      // use noteSideEffect
1224     /// (Foo() || true) // use noteSideEffect
1225     /// Foo() + 1       // use noteFailure
noteFailure()1226     LLVM_NODISCARD bool noteFailure() {
1227       // Failure when evaluating some expression often means there is some
1228       // subexpression whose evaluation was skipped. Therefore, (because we
1229       // don't track whether we skipped an expression when unwinding after an
1230       // evaluation failure) every evaluation failure that bubbles up from a
1231       // subexpression implies that a side-effect has potentially happened. We
1232       // skip setting the HasSideEffects flag to true until we decide to
1233       // continue evaluating after that point, which happens here.
1234       bool KeepGoing = keepEvaluatingAfterFailure();
1235       EvalStatus.HasSideEffects |= KeepGoing;
1236       return KeepGoing;
1237     }
1238 
1239     class ArrayInitLoopIndex {
1240       EvalInfo &Info;
1241       uint64_t OuterIndex;
1242 
1243     public:
ArrayInitLoopIndex(EvalInfo & Info)1244       ArrayInitLoopIndex(EvalInfo &Info)
1245           : Info(Info), OuterIndex(Info.ArrayInitIndex) {
1246         Info.ArrayInitIndex = 0;
1247       }
~ArrayInitLoopIndex()1248       ~ArrayInitLoopIndex() { Info.ArrayInitIndex = OuterIndex; }
1249 
operator uint64_t&()1250       operator uint64_t&() { return Info.ArrayInitIndex; }
1251     };
1252   };
1253 
1254   /// Object used to treat all foldable expressions as constant expressions.
1255   struct FoldConstant {
1256     EvalInfo &Info;
1257     bool Enabled;
1258     bool HadNoPriorDiags;
1259     EvalInfo::EvaluationMode OldMode;
1260 
FoldConstant__anonb66d72d20311::FoldConstant1261     explicit FoldConstant(EvalInfo &Info, bool Enabled)
1262       : Info(Info),
1263         Enabled(Enabled),
1264         HadNoPriorDiags(Info.EvalStatus.Diag &&
1265                         Info.EvalStatus.Diag->empty() &&
1266                         !Info.EvalStatus.HasSideEffects),
1267         OldMode(Info.EvalMode) {
1268       if (Enabled)
1269         Info.EvalMode = EvalInfo::EM_ConstantFold;
1270     }
keepDiagnostics__anonb66d72d20311::FoldConstant1271     void keepDiagnostics() { Enabled = false; }
~FoldConstant__anonb66d72d20311::FoldConstant1272     ~FoldConstant() {
1273       if (Enabled && HadNoPriorDiags && !Info.EvalStatus.Diag->empty() &&
1274           !Info.EvalStatus.HasSideEffects)
1275         Info.EvalStatus.Diag->clear();
1276       Info.EvalMode = OldMode;
1277     }
1278   };
1279 
1280   /// RAII object used to set the current evaluation mode to ignore
1281   /// side-effects.
1282   struct IgnoreSideEffectsRAII {
1283     EvalInfo &Info;
1284     EvalInfo::EvaluationMode OldMode;
IgnoreSideEffectsRAII__anonb66d72d20311::IgnoreSideEffectsRAII1285     explicit IgnoreSideEffectsRAII(EvalInfo &Info)
1286         : Info(Info), OldMode(Info.EvalMode) {
1287       Info.EvalMode = EvalInfo::EM_IgnoreSideEffects;
1288     }
1289 
~IgnoreSideEffectsRAII__anonb66d72d20311::IgnoreSideEffectsRAII1290     ~IgnoreSideEffectsRAII() { Info.EvalMode = OldMode; }
1291   };
1292 
1293   /// RAII object used to optionally suppress diagnostics and side-effects from
1294   /// a speculative evaluation.
1295   class SpeculativeEvaluationRAII {
1296     EvalInfo *Info = nullptr;
1297     Expr::EvalStatus OldStatus;
1298     unsigned OldSpeculativeEvaluationDepth;
1299 
moveFromAndCancel(SpeculativeEvaluationRAII && Other)1300     void moveFromAndCancel(SpeculativeEvaluationRAII &&Other) {
1301       Info = Other.Info;
1302       OldStatus = Other.OldStatus;
1303       OldSpeculativeEvaluationDepth = Other.OldSpeculativeEvaluationDepth;
1304       Other.Info = nullptr;
1305     }
1306 
maybeRestoreState()1307     void maybeRestoreState() {
1308       if (!Info)
1309         return;
1310 
1311       Info->EvalStatus = OldStatus;
1312       Info->SpeculativeEvaluationDepth = OldSpeculativeEvaluationDepth;
1313     }
1314 
1315   public:
1316     SpeculativeEvaluationRAII() = default;
1317 
SpeculativeEvaluationRAII(EvalInfo & Info,SmallVectorImpl<PartialDiagnosticAt> * NewDiag=nullptr)1318     SpeculativeEvaluationRAII(
1319         EvalInfo &Info, SmallVectorImpl<PartialDiagnosticAt> *NewDiag = nullptr)
1320         : Info(&Info), OldStatus(Info.EvalStatus),
1321           OldSpeculativeEvaluationDepth(Info.SpeculativeEvaluationDepth) {
1322       Info.EvalStatus.Diag = NewDiag;
1323       Info.SpeculativeEvaluationDepth = Info.CallStackDepth + 1;
1324     }
1325 
1326     SpeculativeEvaluationRAII(const SpeculativeEvaluationRAII &Other) = delete;
SpeculativeEvaluationRAII(SpeculativeEvaluationRAII && Other)1327     SpeculativeEvaluationRAII(SpeculativeEvaluationRAII &&Other) {
1328       moveFromAndCancel(std::move(Other));
1329     }
1330 
operator =(SpeculativeEvaluationRAII && Other)1331     SpeculativeEvaluationRAII &operator=(SpeculativeEvaluationRAII &&Other) {
1332       maybeRestoreState();
1333       moveFromAndCancel(std::move(Other));
1334       return *this;
1335     }
1336 
~SpeculativeEvaluationRAII()1337     ~SpeculativeEvaluationRAII() { maybeRestoreState(); }
1338   };
1339 
1340   /// RAII object wrapping a full-expression or block scope, and handling
1341   /// the ending of the lifetime of temporaries created within it.
1342   template<ScopeKind Kind>
1343   class ScopeRAII {
1344     EvalInfo &Info;
1345     unsigned OldStackSize;
1346   public:
ScopeRAII(EvalInfo & Info)1347     ScopeRAII(EvalInfo &Info)
1348         : Info(Info), OldStackSize(Info.CleanupStack.size()) {
1349       // Push a new temporary version. This is needed to distinguish between
1350       // temporaries created in different iterations of a loop.
1351       Info.CurrentCall->pushTempVersion();
1352     }
destroy(bool RunDestructors=true)1353     bool destroy(bool RunDestructors = true) {
1354       bool OK = cleanup(Info, RunDestructors, OldStackSize);
1355       OldStackSize = -1U;
1356       return OK;
1357     }
~ScopeRAII()1358     ~ScopeRAII() {
1359       if (OldStackSize != -1U)
1360         destroy(false);
1361       // Body moved to a static method to encourage the compiler to inline away
1362       // instances of this class.
1363       Info.CurrentCall->popTempVersion();
1364     }
1365   private:
cleanup(EvalInfo & Info,bool RunDestructors,unsigned OldStackSize)1366     static bool cleanup(EvalInfo &Info, bool RunDestructors,
1367                         unsigned OldStackSize) {
1368       assert(OldStackSize <= Info.CleanupStack.size() &&
1369              "running cleanups out of order?");
1370 
1371       // Run all cleanups for a block scope, and non-lifetime-extended cleanups
1372       // for a full-expression scope.
1373       bool Success = true;
1374       for (unsigned I = Info.CleanupStack.size(); I > OldStackSize; --I) {
1375         if (Info.CleanupStack[I - 1].isDestroyedAtEndOf(Kind)) {
1376           if (!Info.CleanupStack[I - 1].endLifetime(Info, RunDestructors)) {
1377             Success = false;
1378             break;
1379           }
1380         }
1381       }
1382 
1383       // Compact any retained cleanups.
1384       auto NewEnd = Info.CleanupStack.begin() + OldStackSize;
1385       if (Kind != ScopeKind::Block)
1386         NewEnd =
1387             std::remove_if(NewEnd, Info.CleanupStack.end(), [](Cleanup &C) {
1388               return C.isDestroyedAtEndOf(Kind);
1389             });
1390       Info.CleanupStack.erase(NewEnd, Info.CleanupStack.end());
1391       return Success;
1392     }
1393   };
1394   typedef ScopeRAII<ScopeKind::Block> BlockScopeRAII;
1395   typedef ScopeRAII<ScopeKind::FullExpression> FullExpressionRAII;
1396   typedef ScopeRAII<ScopeKind::Call> CallScopeRAII;
1397 }
1398 
checkSubobject(EvalInfo & Info,const Expr * E,CheckSubobjectKind CSK)1399 bool SubobjectDesignator::checkSubobject(EvalInfo &Info, const Expr *E,
1400                                          CheckSubobjectKind CSK) {
1401   if (Invalid)
1402     return false;
1403   if (isOnePastTheEnd()) {
1404     Info.CCEDiag(E, diag::note_constexpr_past_end_subobject)
1405       << CSK;
1406     setInvalid();
1407     return false;
1408   }
1409   // Note, we do not diagnose if isMostDerivedAnUnsizedArray(), because there
1410   // must actually be at least one array element; even a VLA cannot have a
1411   // bound of zero. And if our index is nonzero, we already had a CCEDiag.
1412   return true;
1413 }
1414 
diagnoseUnsizedArrayPointerArithmetic(EvalInfo & Info,const Expr * E)1415 void SubobjectDesignator::diagnoseUnsizedArrayPointerArithmetic(EvalInfo &Info,
1416                                                                 const Expr *E) {
1417   Info.CCEDiag(E, diag::note_constexpr_unsized_array_indexed);
1418   // Do not set the designator as invalid: we can represent this situation,
1419   // and correct handling of __builtin_object_size requires us to do so.
1420 }
1421 
diagnosePointerArithmetic(EvalInfo & Info,const Expr * E,const APSInt & N)1422 void SubobjectDesignator::diagnosePointerArithmetic(EvalInfo &Info,
1423                                                     const Expr *E,
1424                                                     const APSInt &N) {
1425   // If we're complaining, we must be able to statically determine the size of
1426   // the most derived array.
1427   if (MostDerivedPathLength == Entries.size() && MostDerivedIsArrayElement)
1428     Info.CCEDiag(E, diag::note_constexpr_array_index)
1429       << N << /*array*/ 0
1430       << static_cast<unsigned>(getMostDerivedArraySize());
1431   else
1432     Info.CCEDiag(E, diag::note_constexpr_array_index)
1433       << N << /*non-array*/ 1;
1434   setInvalid();
1435 }
1436 
CallStackFrame(EvalInfo & Info,SourceLocation CallLoc,const FunctionDecl * Callee,const LValue * This,CallRef Call)1437 CallStackFrame::CallStackFrame(EvalInfo &Info, SourceLocation CallLoc,
1438                                const FunctionDecl *Callee, const LValue *This,
1439                                CallRef Call)
1440     : Info(Info), Caller(Info.CurrentCall), Callee(Callee), This(This),
1441       Arguments(Call), CallLoc(CallLoc), Index(Info.NextCallIndex++) {
1442   Info.CurrentCall = this;
1443   ++Info.CallStackDepth;
1444 }
1445 
~CallStackFrame()1446 CallStackFrame::~CallStackFrame() {
1447   assert(Info.CurrentCall == this && "calls retired out of order");
1448   --Info.CallStackDepth;
1449   Info.CurrentCall = Caller;
1450 }
1451 
isRead(AccessKinds AK)1452 static bool isRead(AccessKinds AK) {
1453   return AK == AK_Read || AK == AK_ReadObjectRepresentation;
1454 }
1455 
isModification(AccessKinds AK)1456 static bool isModification(AccessKinds AK) {
1457   switch (AK) {
1458   case AK_Read:
1459   case AK_ReadObjectRepresentation:
1460   case AK_MemberCall:
1461   case AK_DynamicCast:
1462   case AK_TypeId:
1463     return false;
1464   case AK_Assign:
1465   case AK_Increment:
1466   case AK_Decrement:
1467   case AK_Construct:
1468   case AK_Destroy:
1469     return true;
1470   }
1471   llvm_unreachable("unknown access kind");
1472 }
1473 
isAnyAccess(AccessKinds AK)1474 static bool isAnyAccess(AccessKinds AK) {
1475   return isRead(AK) || isModification(AK);
1476 }
1477 
1478 /// Is this an access per the C++ definition?
isFormalAccess(AccessKinds AK)1479 static bool isFormalAccess(AccessKinds AK) {
1480   return isAnyAccess(AK) && AK != AK_Construct && AK != AK_Destroy;
1481 }
1482 
1483 /// Is this kind of axcess valid on an indeterminate object value?
isValidIndeterminateAccess(AccessKinds AK)1484 static bool isValidIndeterminateAccess(AccessKinds AK) {
1485   switch (AK) {
1486   case AK_Read:
1487   case AK_Increment:
1488   case AK_Decrement:
1489     // These need the object's value.
1490     return false;
1491 
1492   case AK_ReadObjectRepresentation:
1493   case AK_Assign:
1494   case AK_Construct:
1495   case AK_Destroy:
1496     // Construction and destruction don't need the value.
1497     return true;
1498 
1499   case AK_MemberCall:
1500   case AK_DynamicCast:
1501   case AK_TypeId:
1502     // These aren't really meaningful on scalars.
1503     return true;
1504   }
1505   llvm_unreachable("unknown access kind");
1506 }
1507 
1508 namespace {
1509   struct ComplexValue {
1510   private:
1511     bool IsInt;
1512 
1513   public:
1514     APSInt IntReal, IntImag;
1515     APFloat FloatReal, FloatImag;
1516 
ComplexValue__anonb66d72d20611::ComplexValue1517     ComplexValue() : FloatReal(APFloat::Bogus()), FloatImag(APFloat::Bogus()) {}
1518 
makeComplexFloat__anonb66d72d20611::ComplexValue1519     void makeComplexFloat() { IsInt = false; }
isComplexFloat__anonb66d72d20611::ComplexValue1520     bool isComplexFloat() const { return !IsInt; }
getComplexFloatReal__anonb66d72d20611::ComplexValue1521     APFloat &getComplexFloatReal() { return FloatReal; }
getComplexFloatImag__anonb66d72d20611::ComplexValue1522     APFloat &getComplexFloatImag() { return FloatImag; }
1523 
makeComplexInt__anonb66d72d20611::ComplexValue1524     void makeComplexInt() { IsInt = true; }
isComplexInt__anonb66d72d20611::ComplexValue1525     bool isComplexInt() const { return IsInt; }
getComplexIntReal__anonb66d72d20611::ComplexValue1526     APSInt &getComplexIntReal() { return IntReal; }
getComplexIntImag__anonb66d72d20611::ComplexValue1527     APSInt &getComplexIntImag() { return IntImag; }
1528 
moveInto__anonb66d72d20611::ComplexValue1529     void moveInto(APValue &v) const {
1530       if (isComplexFloat())
1531         v = APValue(FloatReal, FloatImag);
1532       else
1533         v = APValue(IntReal, IntImag);
1534     }
setFrom__anonb66d72d20611::ComplexValue1535     void setFrom(const APValue &v) {
1536       assert(v.isComplexFloat() || v.isComplexInt());
1537       if (v.isComplexFloat()) {
1538         makeComplexFloat();
1539         FloatReal = v.getComplexFloatReal();
1540         FloatImag = v.getComplexFloatImag();
1541       } else {
1542         makeComplexInt();
1543         IntReal = v.getComplexIntReal();
1544         IntImag = v.getComplexIntImag();
1545       }
1546     }
1547   };
1548 
1549   struct LValue {
1550     APValue::LValueBase Base;
1551     CharUnits Offset;
1552     SubobjectDesignator Designator;
1553     bool IsNullPtr : 1;
1554     bool InvalidBase : 1;
1555 
getLValueBase__anonb66d72d20611::LValue1556     const APValue::LValueBase getLValueBase() const { return Base; }
getLValueOffset__anonb66d72d20611::LValue1557     CharUnits &getLValueOffset() { return Offset; }
getLValueOffset__anonb66d72d20611::LValue1558     const CharUnits &getLValueOffset() const { return Offset; }
getLValueDesignator__anonb66d72d20611::LValue1559     SubobjectDesignator &getLValueDesignator() { return Designator; }
getLValueDesignator__anonb66d72d20611::LValue1560     const SubobjectDesignator &getLValueDesignator() const { return Designator;}
isNullPointer__anonb66d72d20611::LValue1561     bool isNullPointer() const { return IsNullPtr;}
1562 
getLValueCallIndex__anonb66d72d20611::LValue1563     unsigned getLValueCallIndex() const { return Base.getCallIndex(); }
getLValueVersion__anonb66d72d20611::LValue1564     unsigned getLValueVersion() const { return Base.getVersion(); }
1565 
moveInto__anonb66d72d20611::LValue1566     void moveInto(APValue &V) const {
1567       if (Designator.Invalid)
1568         V = APValue(Base, Offset, APValue::NoLValuePath(), IsNullPtr);
1569       else {
1570         assert(!InvalidBase && "APValues can't handle invalid LValue bases");
1571         V = APValue(Base, Offset, Designator.Entries,
1572                     Designator.IsOnePastTheEnd, IsNullPtr);
1573       }
1574     }
setFrom__anonb66d72d20611::LValue1575     void setFrom(ASTContext &Ctx, const APValue &V) {
1576       assert(V.isLValue() && "Setting LValue from a non-LValue?");
1577       Base = V.getLValueBase();
1578       Offset = V.getLValueOffset();
1579       InvalidBase = false;
1580       Designator = SubobjectDesignator(Ctx, V);
1581       IsNullPtr = V.isNullPointer();
1582     }
1583 
set__anonb66d72d20611::LValue1584     void set(APValue::LValueBase B, bool BInvalid = false) {
1585 #ifndef NDEBUG
1586       // We only allow a few types of invalid bases. Enforce that here.
1587       if (BInvalid) {
1588         const auto *E = B.get<const Expr *>();
1589         assert((isa<MemberExpr>(E) || tryUnwrapAllocSizeCall(E)) &&
1590                "Unexpected type of invalid base");
1591       }
1592 #endif
1593 
1594       Base = B;
1595       Offset = CharUnits::fromQuantity(0);
1596       InvalidBase = BInvalid;
1597       Designator = SubobjectDesignator(getType(B));
1598       IsNullPtr = false;
1599     }
1600 
setNull__anonb66d72d20611::LValue1601     void setNull(ASTContext &Ctx, QualType PointerTy) {
1602       Base = (const ValueDecl *)nullptr;
1603       Offset =
1604           CharUnits::fromQuantity(Ctx.getTargetNullPointerValue(PointerTy));
1605       InvalidBase = false;
1606       Designator = SubobjectDesignator(PointerTy->getPointeeType());
1607       IsNullPtr = true;
1608     }
1609 
setInvalid__anonb66d72d20611::LValue1610     void setInvalid(APValue::LValueBase B, unsigned I = 0) {
1611       set(B, true);
1612     }
1613 
toString__anonb66d72d20611::LValue1614     std::string toString(ASTContext &Ctx, QualType T) const {
1615       APValue Printable;
1616       moveInto(Printable);
1617       return Printable.getAsString(Ctx, T);
1618     }
1619 
1620   private:
1621     // Check that this LValue is not based on a null pointer. If it is, produce
1622     // a diagnostic and mark the designator as invalid.
1623     template <typename GenDiagType>
checkNullPointerDiagnosingWith__anonb66d72d20611::LValue1624     bool checkNullPointerDiagnosingWith(const GenDiagType &GenDiag) {
1625       if (Designator.Invalid)
1626         return false;
1627       if (IsNullPtr) {
1628         GenDiag();
1629         Designator.setInvalid();
1630         return false;
1631       }
1632       return true;
1633     }
1634 
1635   public:
checkNullPointer__anonb66d72d20611::LValue1636     bool checkNullPointer(EvalInfo &Info, const Expr *E,
1637                           CheckSubobjectKind CSK) {
1638       return checkNullPointerDiagnosingWith([&Info, E, CSK] {
1639         Info.CCEDiag(E, diag::note_constexpr_null_subobject) << CSK;
1640       });
1641     }
1642 
checkNullPointerForFoldAccess__anonb66d72d20611::LValue1643     bool checkNullPointerForFoldAccess(EvalInfo &Info, const Expr *E,
1644                                        AccessKinds AK) {
1645       return checkNullPointerDiagnosingWith([&Info, E, AK] {
1646         Info.FFDiag(E, diag::note_constexpr_access_null) << AK;
1647       });
1648     }
1649 
1650     // Check this LValue refers to an object. If not, set the designator to be
1651     // invalid and emit a diagnostic.
checkSubobject__anonb66d72d20611::LValue1652     bool checkSubobject(EvalInfo &Info, const Expr *E, CheckSubobjectKind CSK) {
1653       return (CSK == CSK_ArrayToPointer || checkNullPointer(Info, E, CSK)) &&
1654              Designator.checkSubobject(Info, E, CSK);
1655     }
1656 
addDecl__anonb66d72d20611::LValue1657     void addDecl(EvalInfo &Info, const Expr *E,
1658                  const Decl *D, bool Virtual = false) {
1659       if (checkSubobject(Info, E, isa<FieldDecl>(D) ? CSK_Field : CSK_Base))
1660         Designator.addDeclUnchecked(D, Virtual);
1661     }
addUnsizedArray__anonb66d72d20611::LValue1662     void addUnsizedArray(EvalInfo &Info, const Expr *E, QualType ElemTy) {
1663       if (!Designator.Entries.empty()) {
1664         Info.CCEDiag(E, diag::note_constexpr_unsupported_unsized_array);
1665         Designator.setInvalid();
1666         return;
1667       }
1668       if (checkSubobject(Info, E, CSK_ArrayToPointer)) {
1669         assert(getType(Base)->isPointerType() || getType(Base)->isArrayType());
1670         Designator.FirstEntryIsAnUnsizedArray = true;
1671         Designator.addUnsizedArrayUnchecked(ElemTy);
1672       }
1673     }
addArray__anonb66d72d20611::LValue1674     void addArray(EvalInfo &Info, const Expr *E, const ConstantArrayType *CAT) {
1675       if (checkSubobject(Info, E, CSK_ArrayToPointer))
1676         Designator.addArrayUnchecked(CAT);
1677     }
addComplex__anonb66d72d20611::LValue1678     void addComplex(EvalInfo &Info, const Expr *E, QualType EltTy, bool Imag) {
1679       if (checkSubobject(Info, E, Imag ? CSK_Imag : CSK_Real))
1680         Designator.addComplexUnchecked(EltTy, Imag);
1681     }
clearIsNullPointer__anonb66d72d20611::LValue1682     void clearIsNullPointer() {
1683       IsNullPtr = false;
1684     }
adjustOffsetAndIndex__anonb66d72d20611::LValue1685     void adjustOffsetAndIndex(EvalInfo &Info, const Expr *E,
1686                               const APSInt &Index, CharUnits ElementSize) {
1687       // An index of 0 has no effect. (In C, adding 0 to a null pointer is UB,
1688       // but we're not required to diagnose it and it's valid in C++.)
1689       if (!Index)
1690         return;
1691 
1692       // Compute the new offset in the appropriate width, wrapping at 64 bits.
1693       // FIXME: When compiling for a 32-bit target, we should use 32-bit
1694       // offsets.
1695       uint64_t Offset64 = Offset.getQuantity();
1696       uint64_t ElemSize64 = ElementSize.getQuantity();
1697       uint64_t Index64 = Index.extOrTrunc(64).getZExtValue();
1698       Offset = CharUnits::fromQuantity(Offset64 + ElemSize64 * Index64);
1699 
1700       if (checkNullPointer(Info, E, CSK_ArrayIndex))
1701         Designator.adjustIndex(Info, E, Index);
1702       clearIsNullPointer();
1703     }
adjustOffset__anonb66d72d20611::LValue1704     void adjustOffset(CharUnits N) {
1705       Offset += N;
1706       if (N.getQuantity())
1707         clearIsNullPointer();
1708     }
1709   };
1710 
1711   struct MemberPtr {
MemberPtr__anonb66d72d20611::MemberPtr1712     MemberPtr() {}
MemberPtr__anonb66d72d20611::MemberPtr1713     explicit MemberPtr(const ValueDecl *Decl) :
1714       DeclAndIsDerivedMember(Decl, false), Path() {}
1715 
1716     /// The member or (direct or indirect) field referred to by this member
1717     /// pointer, or 0 if this is a null member pointer.
getDecl__anonb66d72d20611::MemberPtr1718     const ValueDecl *getDecl() const {
1719       return DeclAndIsDerivedMember.getPointer();
1720     }
1721     /// Is this actually a member of some type derived from the relevant class?
isDerivedMember__anonb66d72d20611::MemberPtr1722     bool isDerivedMember() const {
1723       return DeclAndIsDerivedMember.getInt();
1724     }
1725     /// Get the class which the declaration actually lives in.
getContainingRecord__anonb66d72d20611::MemberPtr1726     const CXXRecordDecl *getContainingRecord() const {
1727       return cast<CXXRecordDecl>(
1728           DeclAndIsDerivedMember.getPointer()->getDeclContext());
1729     }
1730 
moveInto__anonb66d72d20611::MemberPtr1731     void moveInto(APValue &V) const {
1732       V = APValue(getDecl(), isDerivedMember(), Path);
1733     }
setFrom__anonb66d72d20611::MemberPtr1734     void setFrom(const APValue &V) {
1735       assert(V.isMemberPointer());
1736       DeclAndIsDerivedMember.setPointer(V.getMemberPointerDecl());
1737       DeclAndIsDerivedMember.setInt(V.isMemberPointerToDerivedMember());
1738       Path.clear();
1739       ArrayRef<const CXXRecordDecl*> P = V.getMemberPointerPath();
1740       Path.insert(Path.end(), P.begin(), P.end());
1741     }
1742 
1743     /// DeclAndIsDerivedMember - The member declaration, and a flag indicating
1744     /// whether the member is a member of some class derived from the class type
1745     /// of the member pointer.
1746     llvm::PointerIntPair<const ValueDecl*, 1, bool> DeclAndIsDerivedMember;
1747     /// Path - The path of base/derived classes from the member declaration's
1748     /// class (exclusive) to the class type of the member pointer (inclusive).
1749     SmallVector<const CXXRecordDecl*, 4> Path;
1750 
1751     /// Perform a cast towards the class of the Decl (either up or down the
1752     /// hierarchy).
castBack__anonb66d72d20611::MemberPtr1753     bool castBack(const CXXRecordDecl *Class) {
1754       assert(!Path.empty());
1755       const CXXRecordDecl *Expected;
1756       if (Path.size() >= 2)
1757         Expected = Path[Path.size() - 2];
1758       else
1759         Expected = getContainingRecord();
1760       if (Expected->getCanonicalDecl() != Class->getCanonicalDecl()) {
1761         // C++11 [expr.static.cast]p12: In a conversion from (D::*) to (B::*),
1762         // if B does not contain the original member and is not a base or
1763         // derived class of the class containing the original member, the result
1764         // of the cast is undefined.
1765         // C++11 [conv.mem]p2 does not cover this case for a cast from (B::*) to
1766         // (D::*). We consider that to be a language defect.
1767         return false;
1768       }
1769       Path.pop_back();
1770       return true;
1771     }
1772     /// Perform a base-to-derived member pointer cast.
castToDerived__anonb66d72d20611::MemberPtr1773     bool castToDerived(const CXXRecordDecl *Derived) {
1774       if (!getDecl())
1775         return true;
1776       if (!isDerivedMember()) {
1777         Path.push_back(Derived);
1778         return true;
1779       }
1780       if (!castBack(Derived))
1781         return false;
1782       if (Path.empty())
1783         DeclAndIsDerivedMember.setInt(false);
1784       return true;
1785     }
1786     /// Perform a derived-to-base member pointer cast.
castToBase__anonb66d72d20611::MemberPtr1787     bool castToBase(const CXXRecordDecl *Base) {
1788       if (!getDecl())
1789         return true;
1790       if (Path.empty())
1791         DeclAndIsDerivedMember.setInt(true);
1792       if (isDerivedMember()) {
1793         Path.push_back(Base);
1794         return true;
1795       }
1796       return castBack(Base);
1797     }
1798   };
1799 
1800   /// Compare two member pointers, which are assumed to be of the same type.
operator ==(const MemberPtr & LHS,const MemberPtr & RHS)1801   static bool operator==(const MemberPtr &LHS, const MemberPtr &RHS) {
1802     if (!LHS.getDecl() || !RHS.getDecl())
1803       return !LHS.getDecl() && !RHS.getDecl();
1804     if (LHS.getDecl()->getCanonicalDecl() != RHS.getDecl()->getCanonicalDecl())
1805       return false;
1806     return LHS.Path == RHS.Path;
1807   }
1808 }
1809 
1810 static bool Evaluate(APValue &Result, EvalInfo &Info, const Expr *E);
1811 static bool EvaluateInPlace(APValue &Result, EvalInfo &Info,
1812                             const LValue &This, const Expr *E,
1813                             bool AllowNonLiteralTypes = false);
1814 static bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info,
1815                            bool InvalidBaseOK = false);
1816 static bool EvaluatePointer(const Expr *E, LValue &Result, EvalInfo &Info,
1817                             bool InvalidBaseOK = false);
1818 static bool EvaluateMemberPointer(const Expr *E, MemberPtr &Result,
1819                                   EvalInfo &Info);
1820 static bool EvaluateTemporary(const Expr *E, LValue &Result, EvalInfo &Info);
1821 static bool EvaluateInteger(const Expr *E, APSInt &Result, EvalInfo &Info);
1822 static bool EvaluateIntegerOrLValue(const Expr *E, APValue &Result,
1823                                     EvalInfo &Info);
1824 static bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
1825 static bool EvaluateComplex(const Expr *E, ComplexValue &Res, EvalInfo &Info);
1826 static bool EvaluateAtomic(const Expr *E, const LValue *This, APValue &Result,
1827                            EvalInfo &Info);
1828 static bool EvaluateAsRValue(EvalInfo &Info, const Expr *E, APValue &Result);
1829 
1830 /// Evaluate an integer or fixed point expression into an APResult.
1831 static bool EvaluateFixedPointOrInteger(const Expr *E, APFixedPoint &Result,
1832                                         EvalInfo &Info);
1833 
1834 /// Evaluate only a fixed point expression into an APResult.
1835 static bool EvaluateFixedPoint(const Expr *E, APFixedPoint &Result,
1836                                EvalInfo &Info);
1837 
1838 //===----------------------------------------------------------------------===//
1839 // Misc utilities
1840 //===----------------------------------------------------------------------===//
1841 
1842 /// Negate an APSInt in place, converting it to a signed form if necessary, and
1843 /// preserving its value (by extending by up to one bit as needed).
negateAsSigned(APSInt & Int)1844 static void negateAsSigned(APSInt &Int) {
1845   if (Int.isUnsigned() || Int.isMinSignedValue()) {
1846     Int = Int.extend(Int.getBitWidth() + 1);
1847     Int.setIsSigned(true);
1848   }
1849   Int = -Int;
1850 }
1851 
1852 template<typename KeyT>
createTemporary(const KeyT * Key,QualType T,ScopeKind Scope,LValue & LV)1853 APValue &CallStackFrame::createTemporary(const KeyT *Key, QualType T,
1854                                          ScopeKind Scope, LValue &LV) {
1855   unsigned Version = getTempVersion();
1856   APValue::LValueBase Base(Key, Index, Version);
1857   LV.set(Base);
1858   return createLocal(Base, Key, T, Scope);
1859 }
1860 
1861 /// Allocate storage for a parameter of a function call made in this frame.
createParam(CallRef Args,const ParmVarDecl * PVD,LValue & LV)1862 APValue &CallStackFrame::createParam(CallRef Args, const ParmVarDecl *PVD,
1863                                      LValue &LV) {
1864   assert(Args.CallIndex == Index && "creating parameter in wrong frame");
1865   APValue::LValueBase Base(PVD, Index, Args.Version);
1866   LV.set(Base);
1867   // We always destroy parameters at the end of the call, even if we'd allow
1868   // them to live to the end of the full-expression at runtime, in order to
1869   // give portable results and match other compilers.
1870   return createLocal(Base, PVD, PVD->getType(), ScopeKind::Call);
1871 }
1872 
createLocal(APValue::LValueBase Base,const void * Key,QualType T,ScopeKind Scope)1873 APValue &CallStackFrame::createLocal(APValue::LValueBase Base, const void *Key,
1874                                      QualType T, ScopeKind Scope) {
1875   assert(Base.getCallIndex() == Index && "lvalue for wrong frame");
1876   unsigned Version = Base.getVersion();
1877   APValue &Result = Temporaries[MapKeyTy(Key, Version)];
1878   assert(Result.isAbsent() && "local created multiple times");
1879 
1880   // If we're creating a local immediately in the operand of a speculative
1881   // evaluation, don't register a cleanup to be run outside the speculative
1882   // evaluation context, since we won't actually be able to initialize this
1883   // object.
1884   if (Index <= Info.SpeculativeEvaluationDepth) {
1885     if (T.isDestructedType())
1886       Info.noteSideEffect();
1887   } else {
1888     Info.CleanupStack.push_back(Cleanup(&Result, Base, T, Scope));
1889   }
1890   return Result;
1891 }
1892 
createHeapAlloc(const Expr * E,QualType T,LValue & LV)1893 APValue *EvalInfo::createHeapAlloc(const Expr *E, QualType T, LValue &LV) {
1894   if (NumHeapAllocs > DynamicAllocLValue::getMaxIndex()) {
1895     FFDiag(E, diag::note_constexpr_heap_alloc_limit_exceeded);
1896     return nullptr;
1897   }
1898 
1899   DynamicAllocLValue DA(NumHeapAllocs++);
1900   LV.set(APValue::LValueBase::getDynamicAlloc(DA, T));
1901   auto Result = HeapAllocs.emplace(std::piecewise_construct,
1902                                    std::forward_as_tuple(DA), std::tuple<>());
1903   assert(Result.second && "reused a heap alloc index?");
1904   Result.first->second.AllocExpr = E;
1905   return &Result.first->second.Value;
1906 }
1907 
1908 /// Produce a string describing the given constexpr call.
describe(raw_ostream & Out)1909 void CallStackFrame::describe(raw_ostream &Out) {
1910   unsigned ArgIndex = 0;
1911   bool IsMemberCall = isa<CXXMethodDecl>(Callee) &&
1912                       !isa<CXXConstructorDecl>(Callee) &&
1913                       cast<CXXMethodDecl>(Callee)->isInstance();
1914 
1915   if (!IsMemberCall)
1916     Out << *Callee << '(';
1917 
1918   if (This && IsMemberCall) {
1919     APValue Val;
1920     This->moveInto(Val);
1921     Val.printPretty(Out, Info.Ctx,
1922                     This->Designator.MostDerivedType);
1923     // FIXME: Add parens around Val if needed.
1924     Out << "->" << *Callee << '(';
1925     IsMemberCall = false;
1926   }
1927 
1928   for (FunctionDecl::param_const_iterator I = Callee->param_begin(),
1929        E = Callee->param_end(); I != E; ++I, ++ArgIndex) {
1930     if (ArgIndex > (unsigned)IsMemberCall)
1931       Out << ", ";
1932 
1933     const ParmVarDecl *Param = *I;
1934     APValue *V = Info.getParamSlot(Arguments, Param);
1935     if (V)
1936       V->printPretty(Out, Info.Ctx, Param->getType());
1937     else
1938       Out << "<...>";
1939 
1940     if (ArgIndex == 0 && IsMemberCall)
1941       Out << "->" << *Callee << '(';
1942   }
1943 
1944   Out << ')';
1945 }
1946 
1947 /// Evaluate an expression to see if it had side-effects, and discard its
1948 /// result.
1949 /// \return \c true if the caller should keep evaluating.
EvaluateIgnoredValue(EvalInfo & Info,const Expr * E)1950 static bool EvaluateIgnoredValue(EvalInfo &Info, const Expr *E) {
1951   assert(!E->isValueDependent());
1952   APValue Scratch;
1953   if (!Evaluate(Scratch, Info, E))
1954     // We don't need the value, but we might have skipped a side effect here.
1955     return Info.noteSideEffect();
1956   return true;
1957 }
1958 
1959 /// Should this call expression be treated as a string literal?
IsStringLiteralCall(const CallExpr * E)1960 static bool IsStringLiteralCall(const CallExpr *E) {
1961   unsigned Builtin = E->getBuiltinCallee();
1962   return (Builtin == Builtin::BI__builtin___CFStringMakeConstantString ||
1963           Builtin == Builtin::BI__builtin___NSStringMakeConstantString);
1964 }
1965 
IsGlobalLValue(APValue::LValueBase B)1966 static bool IsGlobalLValue(APValue::LValueBase B) {
1967   // C++11 [expr.const]p3 An address constant expression is a prvalue core
1968   // constant expression of pointer type that evaluates to...
1969 
1970   // ... a null pointer value, or a prvalue core constant expression of type
1971   // std::nullptr_t.
1972   if (!B) return true;
1973 
1974   if (const ValueDecl *D = B.dyn_cast<const ValueDecl*>()) {
1975     // ... the address of an object with static storage duration,
1976     if (const VarDecl *VD = dyn_cast<VarDecl>(D))
1977       return VD->hasGlobalStorage();
1978     if (isa<TemplateParamObjectDecl>(D))
1979       return true;
1980     // ... the address of a function,
1981     // ... the address of a GUID [MS extension],
1982     return isa<FunctionDecl>(D) || isa<MSGuidDecl>(D);
1983   }
1984 
1985   if (B.is<TypeInfoLValue>() || B.is<DynamicAllocLValue>())
1986     return true;
1987 
1988   const Expr *E = B.get<const Expr*>();
1989   switch (E->getStmtClass()) {
1990   default:
1991     return false;
1992   case Expr::CompoundLiteralExprClass: {
1993     const CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
1994     return CLE->isFileScope() && CLE->isLValue();
1995   }
1996   case Expr::MaterializeTemporaryExprClass:
1997     // A materialized temporary might have been lifetime-extended to static
1998     // storage duration.
1999     return cast<MaterializeTemporaryExpr>(E)->getStorageDuration() == SD_Static;
2000   // A string literal has static storage duration.
2001   case Expr::StringLiteralClass:
2002   case Expr::PredefinedExprClass:
2003   case Expr::ObjCStringLiteralClass:
2004   case Expr::ObjCEncodeExprClass:
2005     return true;
2006   case Expr::ObjCBoxedExprClass:
2007     return cast<ObjCBoxedExpr>(E)->isExpressibleAsConstantInitializer();
2008   case Expr::CallExprClass:
2009     return IsStringLiteralCall(cast<CallExpr>(E));
2010   // For GCC compatibility, &&label has static storage duration.
2011   case Expr::AddrLabelExprClass:
2012     return true;
2013   // A Block literal expression may be used as the initialization value for
2014   // Block variables at global or local static scope.
2015   case Expr::BlockExprClass:
2016     return !cast<BlockExpr>(E)->getBlockDecl()->hasCaptures();
2017   case Expr::ImplicitValueInitExprClass:
2018     // FIXME:
2019     // We can never form an lvalue with an implicit value initialization as its
2020     // base through expression evaluation, so these only appear in one case: the
2021     // implicit variable declaration we invent when checking whether a constexpr
2022     // constructor can produce a constant expression. We must assume that such
2023     // an expression might be a global lvalue.
2024     return true;
2025   }
2026 }
2027 
GetLValueBaseDecl(const LValue & LVal)2028 static const ValueDecl *GetLValueBaseDecl(const LValue &LVal) {
2029   return LVal.Base.dyn_cast<const ValueDecl*>();
2030 }
2031 
IsLiteralLValue(const LValue & Value)2032 static bool IsLiteralLValue(const LValue &Value) {
2033   if (Value.getLValueCallIndex())
2034     return false;
2035   const Expr *E = Value.Base.dyn_cast<const Expr*>();
2036   return E && !isa<MaterializeTemporaryExpr>(E);
2037 }
2038 
IsWeakLValue(const LValue & Value)2039 static bool IsWeakLValue(const LValue &Value) {
2040   const ValueDecl *Decl = GetLValueBaseDecl(Value);
2041   return Decl && Decl->isWeak();
2042 }
2043 
isZeroSized(const LValue & Value)2044 static bool isZeroSized(const LValue &Value) {
2045   const ValueDecl *Decl = GetLValueBaseDecl(Value);
2046   if (Decl && isa<VarDecl>(Decl)) {
2047     QualType Ty = Decl->getType();
2048     if (Ty->isArrayType())
2049       return Ty->isIncompleteType() ||
2050              Decl->getASTContext().getTypeSize(Ty) == 0;
2051   }
2052   return false;
2053 }
2054 
HasSameBase(const LValue & A,const LValue & B)2055 static bool HasSameBase(const LValue &A, const LValue &B) {
2056   if (!A.getLValueBase())
2057     return !B.getLValueBase();
2058   if (!B.getLValueBase())
2059     return false;
2060 
2061   if (A.getLValueBase().getOpaqueValue() !=
2062       B.getLValueBase().getOpaqueValue())
2063     return false;
2064 
2065   return A.getLValueCallIndex() == B.getLValueCallIndex() &&
2066          A.getLValueVersion() == B.getLValueVersion();
2067 }
2068 
NoteLValueLocation(EvalInfo & Info,APValue::LValueBase Base)2069 static void NoteLValueLocation(EvalInfo &Info, APValue::LValueBase Base) {
2070   assert(Base && "no location for a null lvalue");
2071   const ValueDecl *VD = Base.dyn_cast<const ValueDecl*>();
2072 
2073   // For a parameter, find the corresponding call stack frame (if it still
2074   // exists), and point at the parameter of the function definition we actually
2075   // invoked.
2076   if (auto *PVD = dyn_cast_or_null<ParmVarDecl>(VD)) {
2077     unsigned Idx = PVD->getFunctionScopeIndex();
2078     for (CallStackFrame *F = Info.CurrentCall; F; F = F->Caller) {
2079       if (F->Arguments.CallIndex == Base.getCallIndex() &&
2080           F->Arguments.Version == Base.getVersion() && F->Callee &&
2081           Idx < F->Callee->getNumParams()) {
2082         VD = F->Callee->getParamDecl(Idx);
2083         break;
2084       }
2085     }
2086   }
2087 
2088   if (VD)
2089     Info.Note(VD->getLocation(), diag::note_declared_at);
2090   else if (const Expr *E = Base.dyn_cast<const Expr*>())
2091     Info.Note(E->getExprLoc(), diag::note_constexpr_temporary_here);
2092   else if (DynamicAllocLValue DA = Base.dyn_cast<DynamicAllocLValue>()) {
2093     // FIXME: Produce a note for dangling pointers too.
2094     if (Optional<DynAlloc*> Alloc = Info.lookupDynamicAlloc(DA))
2095       Info.Note((*Alloc)->AllocExpr->getExprLoc(),
2096                 diag::note_constexpr_dynamic_alloc_here);
2097   }
2098   // We have no information to show for a typeid(T) object.
2099 }
2100 
2101 enum class CheckEvaluationResultKind {
2102   ConstantExpression,
2103   FullyInitialized,
2104 };
2105 
2106 /// Materialized temporaries that we've already checked to determine if they're
2107 /// initializsed by a constant expression.
2108 using CheckedTemporaries =
2109     llvm::SmallPtrSet<const MaterializeTemporaryExpr *, 8>;
2110 
2111 static bool CheckEvaluationResult(CheckEvaluationResultKind CERK,
2112                                   EvalInfo &Info, SourceLocation DiagLoc,
2113                                   QualType Type, const APValue &Value,
2114                                   ConstantExprKind Kind,
2115                                   SourceLocation SubobjectLoc,
2116                                   CheckedTemporaries &CheckedTemps);
2117 
2118 /// Check that this reference or pointer core constant expression is a valid
2119 /// value for an address or reference constant expression. Return true if we
2120 /// can fold this expression, whether or not it's a constant expression.
CheckLValueConstantExpression(EvalInfo & Info,SourceLocation Loc,QualType Type,const LValue & LVal,ConstantExprKind Kind,CheckedTemporaries & CheckedTemps)2121 static bool CheckLValueConstantExpression(EvalInfo &Info, SourceLocation Loc,
2122                                           QualType Type, const LValue &LVal,
2123                                           ConstantExprKind Kind,
2124                                           CheckedTemporaries &CheckedTemps) {
2125   bool IsReferenceType = Type->isReferenceType();
2126 
2127   APValue::LValueBase Base = LVal.getLValueBase();
2128   const SubobjectDesignator &Designator = LVal.getLValueDesignator();
2129 
2130   const Expr *BaseE = Base.dyn_cast<const Expr *>();
2131   const ValueDecl *BaseVD = Base.dyn_cast<const ValueDecl*>();
2132 
2133   // Additional restrictions apply in a template argument. We only enforce the
2134   // C++20 restrictions here; additional syntactic and semantic restrictions
2135   // are applied elsewhere.
2136   if (isTemplateArgument(Kind)) {
2137     int InvalidBaseKind = -1;
2138     StringRef Ident;
2139     if (Base.is<TypeInfoLValue>())
2140       InvalidBaseKind = 0;
2141     else if (isa_and_nonnull<StringLiteral>(BaseE))
2142       InvalidBaseKind = 1;
2143     else if (isa_and_nonnull<MaterializeTemporaryExpr>(BaseE) ||
2144              isa_and_nonnull<LifetimeExtendedTemporaryDecl>(BaseVD))
2145       InvalidBaseKind = 2;
2146     else if (auto *PE = dyn_cast_or_null<PredefinedExpr>(BaseE)) {
2147       InvalidBaseKind = 3;
2148       Ident = PE->getIdentKindName();
2149     }
2150 
2151     if (InvalidBaseKind != -1) {
2152       Info.FFDiag(Loc, diag::note_constexpr_invalid_template_arg)
2153           << IsReferenceType << !Designator.Entries.empty() << InvalidBaseKind
2154           << Ident;
2155       return false;
2156     }
2157   }
2158 
2159   if (auto *FD = dyn_cast_or_null<FunctionDecl>(BaseVD)) {
2160     if (FD->isConsteval()) {
2161       Info.FFDiag(Loc, diag::note_consteval_address_accessible)
2162           << !Type->isAnyPointerType();
2163       Info.Note(FD->getLocation(), diag::note_declared_at);
2164       return false;
2165     }
2166   }
2167 
2168   // Check that the object is a global. Note that the fake 'this' object we
2169   // manufacture when checking potential constant expressions is conservatively
2170   // assumed to be global here.
2171   if (!IsGlobalLValue(Base)) {
2172     if (Info.getLangOpts().CPlusPlus11) {
2173       const ValueDecl *VD = Base.dyn_cast<const ValueDecl*>();
2174       Info.FFDiag(Loc, diag::note_constexpr_non_global, 1)
2175         << IsReferenceType << !Designator.Entries.empty()
2176         << !!VD << VD;
2177 
2178       auto *VarD = dyn_cast_or_null<VarDecl>(VD);
2179       if (VarD && VarD->isConstexpr()) {
2180         // Non-static local constexpr variables have unintuitive semantics:
2181         //   constexpr int a = 1;
2182         //   constexpr const int *p = &a;
2183         // ... is invalid because the address of 'a' is not constant. Suggest
2184         // adding a 'static' in this case.
2185         Info.Note(VarD->getLocation(), diag::note_constexpr_not_static)
2186             << VarD
2187             << FixItHint::CreateInsertion(VarD->getBeginLoc(), "static ");
2188       } else {
2189         NoteLValueLocation(Info, Base);
2190       }
2191     } else {
2192       Info.FFDiag(Loc);
2193     }
2194     // Don't allow references to temporaries to escape.
2195     return false;
2196   }
2197   assert((Info.checkingPotentialConstantExpression() ||
2198           LVal.getLValueCallIndex() == 0) &&
2199          "have call index for global lvalue");
2200 
2201   if (Base.is<DynamicAllocLValue>()) {
2202     Info.FFDiag(Loc, diag::note_constexpr_dynamic_alloc)
2203         << IsReferenceType << !Designator.Entries.empty();
2204     NoteLValueLocation(Info, Base);
2205     return false;
2206   }
2207 
2208   if (BaseVD) {
2209     if (const VarDecl *Var = dyn_cast<const VarDecl>(BaseVD)) {
2210       // Check if this is a thread-local variable.
2211       if (Var->getTLSKind())
2212         // FIXME: Diagnostic!
2213         return false;
2214 
2215       // A dllimport variable never acts like a constant, unless we're
2216       // evaluating a value for use only in name mangling.
2217       if (!isForManglingOnly(Kind) && Var->hasAttr<DLLImportAttr>())
2218         // FIXME: Diagnostic!
2219         return false;
2220     }
2221     if (const auto *FD = dyn_cast<const FunctionDecl>(BaseVD)) {
2222       // __declspec(dllimport) must be handled very carefully:
2223       // We must never initialize an expression with the thunk in C++.
2224       // Doing otherwise would allow the same id-expression to yield
2225       // different addresses for the same function in different translation
2226       // units.  However, this means that we must dynamically initialize the
2227       // expression with the contents of the import address table at runtime.
2228       //
2229       // The C language has no notion of ODR; furthermore, it has no notion of
2230       // dynamic initialization.  This means that we are permitted to
2231       // perform initialization with the address of the thunk.
2232       if (Info.getLangOpts().CPlusPlus && !isForManglingOnly(Kind) &&
2233           FD->hasAttr<DLLImportAttr>())
2234         // FIXME: Diagnostic!
2235         return false;
2236     }
2237   } else if (const auto *MTE =
2238                  dyn_cast_or_null<MaterializeTemporaryExpr>(BaseE)) {
2239     if (CheckedTemps.insert(MTE).second) {
2240       QualType TempType = getType(Base);
2241       if (TempType.isDestructedType()) {
2242         Info.FFDiag(MTE->getExprLoc(),
2243                     diag::note_constexpr_unsupported_temporary_nontrivial_dtor)
2244             << TempType;
2245         return false;
2246       }
2247 
2248       APValue *V = MTE->getOrCreateValue(false);
2249       assert(V && "evasluation result refers to uninitialised temporary");
2250       if (!CheckEvaluationResult(CheckEvaluationResultKind::ConstantExpression,
2251                                  Info, MTE->getExprLoc(), TempType, *V,
2252                                  Kind, SourceLocation(), CheckedTemps))
2253         return false;
2254     }
2255   }
2256 
2257   // Allow address constant expressions to be past-the-end pointers. This is
2258   // an extension: the standard requires them to point to an object.
2259   if (!IsReferenceType)
2260     return true;
2261 
2262   // A reference constant expression must refer to an object.
2263   if (!Base) {
2264     // FIXME: diagnostic
2265     Info.CCEDiag(Loc);
2266     return true;
2267   }
2268 
2269   // Does this refer one past the end of some object?
2270   if (!Designator.Invalid && Designator.isOnePastTheEnd()) {
2271     Info.FFDiag(Loc, diag::note_constexpr_past_end, 1)
2272       << !Designator.Entries.empty() << !!BaseVD << BaseVD;
2273     NoteLValueLocation(Info, Base);
2274   }
2275 
2276   return true;
2277 }
2278 
2279 /// Member pointers are constant expressions unless they point to a
2280 /// non-virtual dllimport member function.
CheckMemberPointerConstantExpression(EvalInfo & Info,SourceLocation Loc,QualType Type,const APValue & Value,ConstantExprKind Kind)2281 static bool CheckMemberPointerConstantExpression(EvalInfo &Info,
2282                                                  SourceLocation Loc,
2283                                                  QualType Type,
2284                                                  const APValue &Value,
2285                                                  ConstantExprKind Kind) {
2286   const ValueDecl *Member = Value.getMemberPointerDecl();
2287   const auto *FD = dyn_cast_or_null<CXXMethodDecl>(Member);
2288   if (!FD)
2289     return true;
2290   if (FD->isConsteval()) {
2291     Info.FFDiag(Loc, diag::note_consteval_address_accessible) << /*pointer*/ 0;
2292     Info.Note(FD->getLocation(), diag::note_declared_at);
2293     return false;
2294   }
2295   return isForManglingOnly(Kind) || FD->isVirtual() ||
2296          !FD->hasAttr<DLLImportAttr>();
2297 }
2298 
2299 /// Check that this core constant expression is of literal type, and if not,
2300 /// produce an appropriate diagnostic.
CheckLiteralType(EvalInfo & Info,const Expr * E,const LValue * This=nullptr)2301 static bool CheckLiteralType(EvalInfo &Info, const Expr *E,
2302                              const LValue *This = nullptr) {
2303   if (!E->isPRValue() || E->getType()->isLiteralType(Info.Ctx))
2304     return true;
2305 
2306   // C++1y: A constant initializer for an object o [...] may also invoke
2307   // constexpr constructors for o and its subobjects even if those objects
2308   // are of non-literal class types.
2309   //
2310   // C++11 missed this detail for aggregates, so classes like this:
2311   //   struct foo_t { union { int i; volatile int j; } u; };
2312   // are not (obviously) initializable like so:
2313   //   __attribute__((__require_constant_initialization__))
2314   //   static const foo_t x = {{0}};
2315   // because "i" is a subobject with non-literal initialization (due to the
2316   // volatile member of the union). See:
2317   //   http://www.open-std.org/jtc1/sc22/wg21/docs/cwg_active.html#1677
2318   // Therefore, we use the C++1y behavior.
2319   if (This && Info.EvaluatingDecl == This->getLValueBase())
2320     return true;
2321 
2322   // Prvalue constant expressions must be of literal types.
2323   if (Info.getLangOpts().CPlusPlus11)
2324     Info.FFDiag(E, diag::note_constexpr_nonliteral)
2325       << E->getType();
2326   else
2327     Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
2328   return false;
2329 }
2330 
CheckEvaluationResult(CheckEvaluationResultKind CERK,EvalInfo & Info,SourceLocation DiagLoc,QualType Type,const APValue & Value,ConstantExprKind Kind,SourceLocation SubobjectLoc,CheckedTemporaries & CheckedTemps)2331 static bool CheckEvaluationResult(CheckEvaluationResultKind CERK,
2332                                   EvalInfo &Info, SourceLocation DiagLoc,
2333                                   QualType Type, const APValue &Value,
2334                                   ConstantExprKind Kind,
2335                                   SourceLocation SubobjectLoc,
2336                                   CheckedTemporaries &CheckedTemps) {
2337   if (!Value.hasValue()) {
2338     Info.FFDiag(DiagLoc, diag::note_constexpr_uninitialized)
2339       << true << Type;
2340     if (SubobjectLoc.isValid())
2341       Info.Note(SubobjectLoc, diag::note_constexpr_subobject_declared_here);
2342     return false;
2343   }
2344 
2345   // We allow _Atomic(T) to be initialized from anything that T can be
2346   // initialized from.
2347   if (const AtomicType *AT = Type->getAs<AtomicType>())
2348     Type = AT->getValueType();
2349 
2350   // Core issue 1454: For a literal constant expression of array or class type,
2351   // each subobject of its value shall have been initialized by a constant
2352   // expression.
2353   if (Value.isArray()) {
2354     QualType EltTy = Type->castAsArrayTypeUnsafe()->getElementType();
2355     for (unsigned I = 0, N = Value.getArrayInitializedElts(); I != N; ++I) {
2356       if (!CheckEvaluationResult(CERK, Info, DiagLoc, EltTy,
2357                                  Value.getArrayInitializedElt(I), Kind,
2358                                  SubobjectLoc, CheckedTemps))
2359         return false;
2360     }
2361     if (!Value.hasArrayFiller())
2362       return true;
2363     return CheckEvaluationResult(CERK, Info, DiagLoc, EltTy,
2364                                  Value.getArrayFiller(), Kind, SubobjectLoc,
2365                                  CheckedTemps);
2366   }
2367   if (Value.isUnion() && Value.getUnionField()) {
2368     return CheckEvaluationResult(
2369         CERK, Info, DiagLoc, Value.getUnionField()->getType(),
2370         Value.getUnionValue(), Kind, Value.getUnionField()->getLocation(),
2371         CheckedTemps);
2372   }
2373   if (Value.isStruct()) {
2374     RecordDecl *RD = Type->castAs<RecordType>()->getDecl();
2375     if (const CXXRecordDecl *CD = dyn_cast<CXXRecordDecl>(RD)) {
2376       unsigned BaseIndex = 0;
2377       for (const CXXBaseSpecifier &BS : CD->bases()) {
2378         if (!CheckEvaluationResult(CERK, Info, DiagLoc, BS.getType(),
2379                                    Value.getStructBase(BaseIndex), Kind,
2380                                    BS.getBeginLoc(), CheckedTemps))
2381           return false;
2382         ++BaseIndex;
2383       }
2384     }
2385     for (const auto *I : RD->fields()) {
2386       if (I->isUnnamedBitfield())
2387         continue;
2388 
2389       if (!CheckEvaluationResult(CERK, Info, DiagLoc, I->getType(),
2390                                  Value.getStructField(I->getFieldIndex()),
2391                                  Kind, I->getLocation(), CheckedTemps))
2392         return false;
2393     }
2394   }
2395 
2396   if (Value.isLValue() &&
2397       CERK == CheckEvaluationResultKind::ConstantExpression) {
2398     LValue LVal;
2399     LVal.setFrom(Info.Ctx, Value);
2400     return CheckLValueConstantExpression(Info, DiagLoc, Type, LVal, Kind,
2401                                          CheckedTemps);
2402   }
2403 
2404   if (Value.isMemberPointer() &&
2405       CERK == CheckEvaluationResultKind::ConstantExpression)
2406     return CheckMemberPointerConstantExpression(Info, DiagLoc, Type, Value, Kind);
2407 
2408   // Everything else is fine.
2409   return true;
2410 }
2411 
2412 /// Check that this core constant expression value is a valid value for a
2413 /// constant expression. If not, report an appropriate diagnostic. Does not
2414 /// check that the expression is of literal type.
CheckConstantExpression(EvalInfo & Info,SourceLocation DiagLoc,QualType Type,const APValue & Value,ConstantExprKind Kind)2415 static bool CheckConstantExpression(EvalInfo &Info, SourceLocation DiagLoc,
2416                                     QualType Type, const APValue &Value,
2417                                     ConstantExprKind Kind) {
2418   // Nothing to check for a constant expression of type 'cv void'.
2419   if (Type->isVoidType())
2420     return true;
2421 
2422   CheckedTemporaries CheckedTemps;
2423   return CheckEvaluationResult(CheckEvaluationResultKind::ConstantExpression,
2424                                Info, DiagLoc, Type, Value, Kind,
2425                                SourceLocation(), CheckedTemps);
2426 }
2427 
2428 /// Check that this evaluated value is fully-initialized and can be loaded by
2429 /// an lvalue-to-rvalue conversion.
CheckFullyInitialized(EvalInfo & Info,SourceLocation DiagLoc,QualType Type,const APValue & Value)2430 static bool CheckFullyInitialized(EvalInfo &Info, SourceLocation DiagLoc,
2431                                   QualType Type, const APValue &Value) {
2432   CheckedTemporaries CheckedTemps;
2433   return CheckEvaluationResult(
2434       CheckEvaluationResultKind::FullyInitialized, Info, DiagLoc, Type, Value,
2435       ConstantExprKind::Normal, SourceLocation(), CheckedTemps);
2436 }
2437 
2438 /// Enforce C++2a [expr.const]/4.17, which disallows new-expressions unless
2439 /// "the allocated storage is deallocated within the evaluation".
CheckMemoryLeaks(EvalInfo & Info)2440 static bool CheckMemoryLeaks(EvalInfo &Info) {
2441   if (!Info.HeapAllocs.empty()) {
2442     // We can still fold to a constant despite a compile-time memory leak,
2443     // so long as the heap allocation isn't referenced in the result (we check
2444     // that in CheckConstantExpression).
2445     Info.CCEDiag(Info.HeapAllocs.begin()->second.AllocExpr,
2446                  diag::note_constexpr_memory_leak)
2447         << unsigned(Info.HeapAllocs.size() - 1);
2448   }
2449   return true;
2450 }
2451 
EvalPointerValueAsBool(const APValue & Value,bool & Result)2452 static bool EvalPointerValueAsBool(const APValue &Value, bool &Result) {
2453   // A null base expression indicates a null pointer.  These are always
2454   // evaluatable, and they are false unless the offset is zero.
2455   if (!Value.getLValueBase()) {
2456     Result = !Value.getLValueOffset().isZero();
2457     return true;
2458   }
2459 
2460   // We have a non-null base.  These are generally known to be true, but if it's
2461   // a weak declaration it can be null at runtime.
2462   Result = true;
2463   const ValueDecl *Decl = Value.getLValueBase().dyn_cast<const ValueDecl*>();
2464   return !Decl || !Decl->isWeak();
2465 }
2466 
HandleConversionToBool(const APValue & Val,bool & Result)2467 static bool HandleConversionToBool(const APValue &Val, bool &Result) {
2468   switch (Val.getKind()) {
2469   case APValue::None:
2470   case APValue::Indeterminate:
2471     return false;
2472   case APValue::Int:
2473     Result = Val.getInt().getBoolValue();
2474     return true;
2475   case APValue::FixedPoint:
2476     Result = Val.getFixedPoint().getBoolValue();
2477     return true;
2478   case APValue::Float:
2479     Result = !Val.getFloat().isZero();
2480     return true;
2481   case APValue::ComplexInt:
2482     Result = Val.getComplexIntReal().getBoolValue() ||
2483              Val.getComplexIntImag().getBoolValue();
2484     return true;
2485   case APValue::ComplexFloat:
2486     Result = !Val.getComplexFloatReal().isZero() ||
2487              !Val.getComplexFloatImag().isZero();
2488     return true;
2489   case APValue::LValue:
2490     return EvalPointerValueAsBool(Val, Result);
2491   case APValue::MemberPointer:
2492     Result = Val.getMemberPointerDecl();
2493     return true;
2494   case APValue::Vector:
2495   case APValue::Array:
2496   case APValue::Struct:
2497   case APValue::Union:
2498   case APValue::AddrLabelDiff:
2499     return false;
2500   }
2501 
2502   llvm_unreachable("unknown APValue kind");
2503 }
2504 
EvaluateAsBooleanCondition(const Expr * E,bool & Result,EvalInfo & Info)2505 static bool EvaluateAsBooleanCondition(const Expr *E, bool &Result,
2506                                        EvalInfo &Info) {
2507   assert(!E->isValueDependent());
2508   assert(E->isPRValue() && "missing lvalue-to-rvalue conv in bool condition");
2509   APValue Val;
2510   if (!Evaluate(Val, Info, E))
2511     return false;
2512   return HandleConversionToBool(Val, Result);
2513 }
2514 
2515 template<typename T>
HandleOverflow(EvalInfo & Info,const Expr * E,const T & SrcValue,QualType DestType)2516 static bool HandleOverflow(EvalInfo &Info, const Expr *E,
2517                            const T &SrcValue, QualType DestType) {
2518   Info.CCEDiag(E, diag::note_constexpr_overflow)
2519     << SrcValue << DestType;
2520   return Info.noteUndefinedBehavior();
2521 }
2522 
HandleFloatToIntCast(EvalInfo & Info,const Expr * E,QualType SrcType,const APFloat & Value,QualType DestType,APSInt & Result)2523 static bool HandleFloatToIntCast(EvalInfo &Info, const Expr *E,
2524                                  QualType SrcType, const APFloat &Value,
2525                                  QualType DestType, APSInt &Result) {
2526   unsigned DestWidth = Info.Ctx.getIntWidth(DestType);
2527   // Determine whether we are converting to unsigned or signed.
2528   bool DestSigned = DestType->isSignedIntegerOrEnumerationType();
2529 
2530   Result = APSInt(DestWidth, !DestSigned);
2531   bool ignored;
2532   if (Value.convertToInteger(Result, llvm::APFloat::rmTowardZero, &ignored)
2533       & APFloat::opInvalidOp)
2534     return HandleOverflow(Info, E, Value, DestType);
2535   return true;
2536 }
2537 
2538 /// Get rounding mode used for evaluation of the specified expression.
2539 /// \param[out] DynamicRM Is set to true is the requested rounding mode is
2540 ///                       dynamic.
2541 /// If rounding mode is unknown at compile time, still try to evaluate the
2542 /// expression. If the result is exact, it does not depend on rounding mode.
2543 /// So return "tonearest" mode instead of "dynamic".
getActiveRoundingMode(EvalInfo & Info,const Expr * E,bool & DynamicRM)2544 static llvm::RoundingMode getActiveRoundingMode(EvalInfo &Info, const Expr *E,
2545                                                 bool &DynamicRM) {
2546   llvm::RoundingMode RM =
2547       E->getFPFeaturesInEffect(Info.Ctx.getLangOpts()).getRoundingMode();
2548   DynamicRM = (RM == llvm::RoundingMode::Dynamic);
2549   if (DynamicRM)
2550     RM = llvm::RoundingMode::NearestTiesToEven;
2551   return RM;
2552 }
2553 
2554 /// Check if the given evaluation result is allowed for constant evaluation.
checkFloatingPointResult(EvalInfo & Info,const Expr * E,APFloat::opStatus St)2555 static bool checkFloatingPointResult(EvalInfo &Info, const Expr *E,
2556                                      APFloat::opStatus St) {
2557   // In a constant context, assume that any dynamic rounding mode or FP
2558   // exception state matches the default floating-point environment.
2559   if (Info.InConstantContext)
2560     return true;
2561 
2562   FPOptions FPO = E->getFPFeaturesInEffect(Info.Ctx.getLangOpts());
2563   if ((St & APFloat::opInexact) &&
2564       FPO.getRoundingMode() == llvm::RoundingMode::Dynamic) {
2565     // Inexact result means that it depends on rounding mode. If the requested
2566     // mode is dynamic, the evaluation cannot be made in compile time.
2567     Info.FFDiag(E, diag::note_constexpr_dynamic_rounding);
2568     return false;
2569   }
2570 
2571   if ((St != APFloat::opOK) &&
2572       (FPO.getRoundingMode() == llvm::RoundingMode::Dynamic ||
2573        FPO.getFPExceptionMode() != LangOptions::FPE_Ignore ||
2574        FPO.getAllowFEnvAccess())) {
2575     Info.FFDiag(E, diag::note_constexpr_float_arithmetic_strict);
2576     return false;
2577   }
2578 
2579   if ((St & APFloat::opStatus::opInvalidOp) &&
2580       FPO.getFPExceptionMode() != LangOptions::FPE_Ignore) {
2581     // There is no usefully definable result.
2582     Info.FFDiag(E);
2583     return false;
2584   }
2585 
2586   // FIXME: if:
2587   // - evaluation triggered other FP exception, and
2588   // - exception mode is not "ignore", and
2589   // - the expression being evaluated is not a part of global variable
2590   //   initializer,
2591   // the evaluation probably need to be rejected.
2592   return true;
2593 }
2594 
HandleFloatToFloatCast(EvalInfo & Info,const Expr * E,QualType SrcType,QualType DestType,APFloat & Result)2595 static bool HandleFloatToFloatCast(EvalInfo &Info, const Expr *E,
2596                                    QualType SrcType, QualType DestType,
2597                                    APFloat &Result) {
2598   assert(isa<CastExpr>(E) || isa<CompoundAssignOperator>(E));
2599   bool DynamicRM;
2600   llvm::RoundingMode RM = getActiveRoundingMode(Info, E, DynamicRM);
2601   APFloat::opStatus St;
2602   APFloat Value = Result;
2603   bool ignored;
2604   St = Result.convert(Info.Ctx.getFloatTypeSemantics(DestType), RM, &ignored);
2605   return checkFloatingPointResult(Info, E, St);
2606 }
2607 
HandleIntToIntCast(EvalInfo & Info,const Expr * E,QualType DestType,QualType SrcType,const APSInt & Value)2608 static APSInt HandleIntToIntCast(EvalInfo &Info, const Expr *E,
2609                                  QualType DestType, QualType SrcType,
2610                                  const APSInt &Value) {
2611   unsigned DestWidth = Info.Ctx.getIntWidth(DestType);
2612   // Figure out if this is a truncate, extend or noop cast.
2613   // If the input is signed, do a sign extend, noop, or truncate.
2614   APSInt Result = Value.extOrTrunc(DestWidth);
2615   Result.setIsUnsigned(DestType->isUnsignedIntegerOrEnumerationType());
2616   if (DestType->isBooleanType())
2617     Result = Value.getBoolValue();
2618   return Result;
2619 }
2620 
HandleIntToFloatCast(EvalInfo & Info,const Expr * E,const FPOptions FPO,QualType SrcType,const APSInt & Value,QualType DestType,APFloat & Result)2621 static bool HandleIntToFloatCast(EvalInfo &Info, const Expr *E,
2622                                  const FPOptions FPO,
2623                                  QualType SrcType, const APSInt &Value,
2624                                  QualType DestType, APFloat &Result) {
2625   Result = APFloat(Info.Ctx.getFloatTypeSemantics(DestType), 1);
2626   APFloat::opStatus St = Result.convertFromAPInt(Value, Value.isSigned(),
2627        APFloat::rmNearestTiesToEven);
2628   if (!Info.InConstantContext && St != llvm::APFloatBase::opOK &&
2629       FPO.isFPConstrained()) {
2630     Info.FFDiag(E, diag::note_constexpr_float_arithmetic_strict);
2631     return false;
2632   }
2633   return true;
2634 }
2635 
truncateBitfieldValue(EvalInfo & Info,const Expr * E,APValue & Value,const FieldDecl * FD)2636 static bool truncateBitfieldValue(EvalInfo &Info, const Expr *E,
2637                                   APValue &Value, const FieldDecl *FD) {
2638   assert(FD->isBitField() && "truncateBitfieldValue on non-bitfield");
2639 
2640   if (!Value.isInt()) {
2641     // Trying to store a pointer-cast-to-integer into a bitfield.
2642     // FIXME: In this case, we should provide the diagnostic for casting
2643     // a pointer to an integer.
2644     assert(Value.isLValue() && "integral value neither int nor lvalue?");
2645     Info.FFDiag(E);
2646     return false;
2647   }
2648 
2649   APSInt &Int = Value.getInt();
2650   unsigned OldBitWidth = Int.getBitWidth();
2651   unsigned NewBitWidth = FD->getBitWidthValue(Info.Ctx);
2652   if (NewBitWidth < OldBitWidth)
2653     Int = Int.trunc(NewBitWidth).extend(OldBitWidth);
2654   return true;
2655 }
2656 
EvalAndBitcastToAPInt(EvalInfo & Info,const Expr * E,llvm::APInt & Res)2657 static bool EvalAndBitcastToAPInt(EvalInfo &Info, const Expr *E,
2658                                   llvm::APInt &Res) {
2659   APValue SVal;
2660   if (!Evaluate(SVal, Info, E))
2661     return false;
2662   if (SVal.isInt()) {
2663     Res = SVal.getInt();
2664     return true;
2665   }
2666   if (SVal.isFloat()) {
2667     Res = SVal.getFloat().bitcastToAPInt();
2668     return true;
2669   }
2670   if (SVal.isVector()) {
2671     QualType VecTy = E->getType();
2672     unsigned VecSize = Info.Ctx.getTypeSize(VecTy);
2673     QualType EltTy = VecTy->castAs<VectorType>()->getElementType();
2674     unsigned EltSize = Info.Ctx.getTypeSize(EltTy);
2675     bool BigEndian = Info.Ctx.getTargetInfo().isBigEndian();
2676     Res = llvm::APInt::getNullValue(VecSize);
2677     for (unsigned i = 0; i < SVal.getVectorLength(); i++) {
2678       APValue &Elt = SVal.getVectorElt(i);
2679       llvm::APInt EltAsInt;
2680       if (Elt.isInt()) {
2681         EltAsInt = Elt.getInt();
2682       } else if (Elt.isFloat()) {
2683         EltAsInt = Elt.getFloat().bitcastToAPInt();
2684       } else {
2685         // Don't try to handle vectors of anything other than int or float
2686         // (not sure if it's possible to hit this case).
2687         Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
2688         return false;
2689       }
2690       unsigned BaseEltSize = EltAsInt.getBitWidth();
2691       if (BigEndian)
2692         Res |= EltAsInt.zextOrTrunc(VecSize).rotr(i*EltSize+BaseEltSize);
2693       else
2694         Res |= EltAsInt.zextOrTrunc(VecSize).rotl(i*EltSize);
2695     }
2696     return true;
2697   }
2698   // Give up if the input isn't an int, float, or vector.  For example, we
2699   // reject "(v4i16)(intptr_t)&a".
2700   Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
2701   return false;
2702 }
2703 
2704 /// Perform the given integer operation, which is known to need at most BitWidth
2705 /// bits, and check for overflow in the original type (if that type was not an
2706 /// unsigned type).
2707 template<typename Operation>
CheckedIntArithmetic(EvalInfo & Info,const Expr * E,const APSInt & LHS,const APSInt & RHS,unsigned BitWidth,Operation Op,APSInt & Result)2708 static bool CheckedIntArithmetic(EvalInfo &Info, const Expr *E,
2709                                  const APSInt &LHS, const APSInt &RHS,
2710                                  unsigned BitWidth, Operation Op,
2711                                  APSInt &Result) {
2712   if (LHS.isUnsigned()) {
2713     Result = Op(LHS, RHS);
2714     return true;
2715   }
2716 
2717   APSInt Value(Op(LHS.extend(BitWidth), RHS.extend(BitWidth)), false);
2718   Result = Value.trunc(LHS.getBitWidth());
2719   if (Result.extend(BitWidth) != Value) {
2720     if (Info.checkingForUndefinedBehavior())
2721       Info.Ctx.getDiagnostics().Report(E->getExprLoc(),
2722                                        diag::warn_integer_constant_overflow)
2723           << toString(Result, 10) << E->getType();
2724     return HandleOverflow(Info, E, Value, E->getType());
2725   }
2726   return true;
2727 }
2728 
2729 /// Perform the given binary integer operation.
handleIntIntBinOp(EvalInfo & Info,const Expr * E,const APSInt & LHS,BinaryOperatorKind Opcode,APSInt RHS,APSInt & Result)2730 static bool handleIntIntBinOp(EvalInfo &Info, const Expr *E, const APSInt &LHS,
2731                               BinaryOperatorKind Opcode, APSInt RHS,
2732                               APSInt &Result) {
2733   switch (Opcode) {
2734   default:
2735     Info.FFDiag(E);
2736     return false;
2737   case BO_Mul:
2738     return CheckedIntArithmetic(Info, E, LHS, RHS, LHS.getBitWidth() * 2,
2739                                 std::multiplies<APSInt>(), Result);
2740   case BO_Add:
2741     return CheckedIntArithmetic(Info, E, LHS, RHS, LHS.getBitWidth() + 1,
2742                                 std::plus<APSInt>(), Result);
2743   case BO_Sub:
2744     return CheckedIntArithmetic(Info, E, LHS, RHS, LHS.getBitWidth() + 1,
2745                                 std::minus<APSInt>(), Result);
2746   case BO_And: Result = LHS & RHS; return true;
2747   case BO_Xor: Result = LHS ^ RHS; return true;
2748   case BO_Or:  Result = LHS | RHS; return true;
2749   case BO_Div:
2750   case BO_Rem:
2751     if (RHS == 0) {
2752       Info.FFDiag(E, diag::note_expr_divide_by_zero);
2753       return false;
2754     }
2755     Result = (Opcode == BO_Rem ? LHS % RHS : LHS / RHS);
2756     // Check for overflow case: INT_MIN / -1 or INT_MIN % -1. APSInt supports
2757     // this operation and gives the two's complement result.
2758     if (RHS.isNegative() && RHS.isAllOnesValue() &&
2759         LHS.isSigned() && LHS.isMinSignedValue())
2760       return HandleOverflow(Info, E, -LHS.extend(LHS.getBitWidth() + 1),
2761                             E->getType());
2762     return true;
2763   case BO_Shl: {
2764     if (Info.getLangOpts().OpenCL)
2765       // OpenCL 6.3j: shift values are effectively % word size of LHS.
2766       RHS &= APSInt(llvm::APInt(RHS.getBitWidth(),
2767                     static_cast<uint64_t>(LHS.getBitWidth() - 1)),
2768                     RHS.isUnsigned());
2769     else if (RHS.isSigned() && RHS.isNegative()) {
2770       // During constant-folding, a negative shift is an opposite shift. Such
2771       // a shift is not a constant expression.
2772       Info.CCEDiag(E, diag::note_constexpr_negative_shift) << RHS;
2773       RHS = -RHS;
2774       goto shift_right;
2775     }
2776   shift_left:
2777     // C++11 [expr.shift]p1: Shift width must be less than the bit width of
2778     // the shifted type.
2779     unsigned SA = (unsigned) RHS.getLimitedValue(LHS.getBitWidth()-1);
2780     if (SA != RHS) {
2781       Info.CCEDiag(E, diag::note_constexpr_large_shift)
2782         << RHS << E->getType() << LHS.getBitWidth();
2783     } else if (LHS.isSigned() && !Info.getLangOpts().CPlusPlus20) {
2784       // C++11 [expr.shift]p2: A signed left shift must have a non-negative
2785       // operand, and must not overflow the corresponding unsigned type.
2786       // C++2a [expr.shift]p2: E1 << E2 is the unique value congruent to
2787       // E1 x 2^E2 module 2^N.
2788       if (LHS.isNegative())
2789         Info.CCEDiag(E, diag::note_constexpr_lshift_of_negative) << LHS;
2790       else if (LHS.countLeadingZeros() < SA)
2791         Info.CCEDiag(E, diag::note_constexpr_lshift_discards);
2792     }
2793     Result = LHS << SA;
2794     return true;
2795   }
2796   case BO_Shr: {
2797     if (Info.getLangOpts().OpenCL)
2798       // OpenCL 6.3j: shift values are effectively % word size of LHS.
2799       RHS &= APSInt(llvm::APInt(RHS.getBitWidth(),
2800                     static_cast<uint64_t>(LHS.getBitWidth() - 1)),
2801                     RHS.isUnsigned());
2802     else if (RHS.isSigned() && RHS.isNegative()) {
2803       // During constant-folding, a negative shift is an opposite shift. Such a
2804       // shift is not a constant expression.
2805       Info.CCEDiag(E, diag::note_constexpr_negative_shift) << RHS;
2806       RHS = -RHS;
2807       goto shift_left;
2808     }
2809   shift_right:
2810     // C++11 [expr.shift]p1: Shift width must be less than the bit width of the
2811     // shifted type.
2812     unsigned SA = (unsigned) RHS.getLimitedValue(LHS.getBitWidth()-1);
2813     if (SA != RHS)
2814       Info.CCEDiag(E, diag::note_constexpr_large_shift)
2815         << RHS << E->getType() << LHS.getBitWidth();
2816     Result = LHS >> SA;
2817     return true;
2818   }
2819 
2820   case BO_LT: Result = LHS < RHS; return true;
2821   case BO_GT: Result = LHS > RHS; return true;
2822   case BO_LE: Result = LHS <= RHS; return true;
2823   case BO_GE: Result = LHS >= RHS; return true;
2824   case BO_EQ: Result = LHS == RHS; return true;
2825   case BO_NE: Result = LHS != RHS; return true;
2826   case BO_Cmp:
2827     llvm_unreachable("BO_Cmp should be handled elsewhere");
2828   }
2829 }
2830 
2831 /// Perform the given binary floating-point operation, in-place, on LHS.
handleFloatFloatBinOp(EvalInfo & Info,const BinaryOperator * E,APFloat & LHS,BinaryOperatorKind Opcode,const APFloat & RHS)2832 static bool handleFloatFloatBinOp(EvalInfo &Info, const BinaryOperator *E,
2833                                   APFloat &LHS, BinaryOperatorKind Opcode,
2834                                   const APFloat &RHS) {
2835   bool DynamicRM;
2836   llvm::RoundingMode RM = getActiveRoundingMode(Info, E, DynamicRM);
2837   APFloat::opStatus St;
2838   switch (Opcode) {
2839   default:
2840     Info.FFDiag(E);
2841     return false;
2842   case BO_Mul:
2843     St = LHS.multiply(RHS, RM);
2844     break;
2845   case BO_Add:
2846     St = LHS.add(RHS, RM);
2847     break;
2848   case BO_Sub:
2849     St = LHS.subtract(RHS, RM);
2850     break;
2851   case BO_Div:
2852     // [expr.mul]p4:
2853     //   If the second operand of / or % is zero the behavior is undefined.
2854     if (RHS.isZero())
2855       Info.CCEDiag(E, diag::note_expr_divide_by_zero);
2856     St = LHS.divide(RHS, RM);
2857     break;
2858   }
2859 
2860   // [expr.pre]p4:
2861   //   If during the evaluation of an expression, the result is not
2862   //   mathematically defined [...], the behavior is undefined.
2863   // FIXME: C++ rules require us to not conform to IEEE 754 here.
2864   if (LHS.isNaN()) {
2865     Info.CCEDiag(E, diag::note_constexpr_float_arithmetic) << LHS.isNaN();
2866     return Info.noteUndefinedBehavior();
2867   }
2868 
2869   return checkFloatingPointResult(Info, E, St);
2870 }
2871 
handleLogicalOpForVector(const APInt & LHSValue,BinaryOperatorKind Opcode,const APInt & RHSValue,APInt & Result)2872 static bool handleLogicalOpForVector(const APInt &LHSValue,
2873                                      BinaryOperatorKind Opcode,
2874                                      const APInt &RHSValue, APInt &Result) {
2875   bool LHS = (LHSValue != 0);
2876   bool RHS = (RHSValue != 0);
2877 
2878   if (Opcode == BO_LAnd)
2879     Result = LHS && RHS;
2880   else
2881     Result = LHS || RHS;
2882   return true;
2883 }
handleLogicalOpForVector(const APFloat & LHSValue,BinaryOperatorKind Opcode,const APFloat & RHSValue,APInt & Result)2884 static bool handleLogicalOpForVector(const APFloat &LHSValue,
2885                                      BinaryOperatorKind Opcode,
2886                                      const APFloat &RHSValue, APInt &Result) {
2887   bool LHS = !LHSValue.isZero();
2888   bool RHS = !RHSValue.isZero();
2889 
2890   if (Opcode == BO_LAnd)
2891     Result = LHS && RHS;
2892   else
2893     Result = LHS || RHS;
2894   return true;
2895 }
2896 
handleLogicalOpForVector(const APValue & LHSValue,BinaryOperatorKind Opcode,const APValue & RHSValue,APInt & Result)2897 static bool handleLogicalOpForVector(const APValue &LHSValue,
2898                                      BinaryOperatorKind Opcode,
2899                                      const APValue &RHSValue, APInt &Result) {
2900   // The result is always an int type, however operands match the first.
2901   if (LHSValue.getKind() == APValue::Int)
2902     return handleLogicalOpForVector(LHSValue.getInt(), Opcode,
2903                                     RHSValue.getInt(), Result);
2904   assert(LHSValue.getKind() == APValue::Float && "Should be no other options");
2905   return handleLogicalOpForVector(LHSValue.getFloat(), Opcode,
2906                                   RHSValue.getFloat(), Result);
2907 }
2908 
2909 template <typename APTy>
2910 static bool
handleCompareOpForVectorHelper(const APTy & LHSValue,BinaryOperatorKind Opcode,const APTy & RHSValue,APInt & Result)2911 handleCompareOpForVectorHelper(const APTy &LHSValue, BinaryOperatorKind Opcode,
2912                                const APTy &RHSValue, APInt &Result) {
2913   switch (Opcode) {
2914   default:
2915     llvm_unreachable("unsupported binary operator");
2916   case BO_EQ:
2917     Result = (LHSValue == RHSValue);
2918     break;
2919   case BO_NE:
2920     Result = (LHSValue != RHSValue);
2921     break;
2922   case BO_LT:
2923     Result = (LHSValue < RHSValue);
2924     break;
2925   case BO_GT:
2926     Result = (LHSValue > RHSValue);
2927     break;
2928   case BO_LE:
2929     Result = (LHSValue <= RHSValue);
2930     break;
2931   case BO_GE:
2932     Result = (LHSValue >= RHSValue);
2933     break;
2934   }
2935 
2936   return true;
2937 }
2938 
handleCompareOpForVector(const APValue & LHSValue,BinaryOperatorKind Opcode,const APValue & RHSValue,APInt & Result)2939 static bool handleCompareOpForVector(const APValue &LHSValue,
2940                                      BinaryOperatorKind Opcode,
2941                                      const APValue &RHSValue, APInt &Result) {
2942   // The result is always an int type, however operands match the first.
2943   if (LHSValue.getKind() == APValue::Int)
2944     return handleCompareOpForVectorHelper(LHSValue.getInt(), Opcode,
2945                                           RHSValue.getInt(), Result);
2946   assert(LHSValue.getKind() == APValue::Float && "Should be no other options");
2947   return handleCompareOpForVectorHelper(LHSValue.getFloat(), Opcode,
2948                                         RHSValue.getFloat(), Result);
2949 }
2950 
2951 // Perform binary operations for vector types, in place on the LHS.
handleVectorVectorBinOp(EvalInfo & Info,const BinaryOperator * E,BinaryOperatorKind Opcode,APValue & LHSValue,const APValue & RHSValue)2952 static bool handleVectorVectorBinOp(EvalInfo &Info, const BinaryOperator *E,
2953                                     BinaryOperatorKind Opcode,
2954                                     APValue &LHSValue,
2955                                     const APValue &RHSValue) {
2956   assert(Opcode != BO_PtrMemD && Opcode != BO_PtrMemI &&
2957          "Operation not supported on vector types");
2958 
2959   const auto *VT = E->getType()->castAs<VectorType>();
2960   unsigned NumElements = VT->getNumElements();
2961   QualType EltTy = VT->getElementType();
2962 
2963   // In the cases (typically C as I've observed) where we aren't evaluating
2964   // constexpr but are checking for cases where the LHS isn't yet evaluatable,
2965   // just give up.
2966   if (!LHSValue.isVector()) {
2967     assert(LHSValue.isLValue() &&
2968            "A vector result that isn't a vector OR uncalculated LValue");
2969     Info.FFDiag(E);
2970     return false;
2971   }
2972 
2973   assert(LHSValue.getVectorLength() == NumElements &&
2974          RHSValue.getVectorLength() == NumElements && "Different vector sizes");
2975 
2976   SmallVector<APValue, 4> ResultElements;
2977 
2978   for (unsigned EltNum = 0; EltNum < NumElements; ++EltNum) {
2979     APValue LHSElt = LHSValue.getVectorElt(EltNum);
2980     APValue RHSElt = RHSValue.getVectorElt(EltNum);
2981 
2982     if (EltTy->isIntegerType()) {
2983       APSInt EltResult{Info.Ctx.getIntWidth(EltTy),
2984                        EltTy->isUnsignedIntegerType()};
2985       bool Success = true;
2986 
2987       if (BinaryOperator::isLogicalOp(Opcode))
2988         Success = handleLogicalOpForVector(LHSElt, Opcode, RHSElt, EltResult);
2989       else if (BinaryOperator::isComparisonOp(Opcode))
2990         Success = handleCompareOpForVector(LHSElt, Opcode, RHSElt, EltResult);
2991       else
2992         Success = handleIntIntBinOp(Info, E, LHSElt.getInt(), Opcode,
2993                                     RHSElt.getInt(), EltResult);
2994 
2995       if (!Success) {
2996         Info.FFDiag(E);
2997         return false;
2998       }
2999       ResultElements.emplace_back(EltResult);
3000 
3001     } else if (EltTy->isFloatingType()) {
3002       assert(LHSElt.getKind() == APValue::Float &&
3003              RHSElt.getKind() == APValue::Float &&
3004              "Mismatched LHS/RHS/Result Type");
3005       APFloat LHSFloat = LHSElt.getFloat();
3006 
3007       if (!handleFloatFloatBinOp(Info, E, LHSFloat, Opcode,
3008                                  RHSElt.getFloat())) {
3009         Info.FFDiag(E);
3010         return false;
3011       }
3012 
3013       ResultElements.emplace_back(LHSFloat);
3014     }
3015   }
3016 
3017   LHSValue = APValue(ResultElements.data(), ResultElements.size());
3018   return true;
3019 }
3020 
3021 /// Cast an lvalue referring to a base subobject to a derived class, by
3022 /// truncating the lvalue's path to the given length.
CastToDerivedClass(EvalInfo & Info,const Expr * E,LValue & Result,const RecordDecl * TruncatedType,unsigned TruncatedElements)3023 static bool CastToDerivedClass(EvalInfo &Info, const Expr *E, LValue &Result,
3024                                const RecordDecl *TruncatedType,
3025                                unsigned TruncatedElements) {
3026   SubobjectDesignator &D = Result.Designator;
3027 
3028   // Check we actually point to a derived class object.
3029   if (TruncatedElements == D.Entries.size())
3030     return true;
3031   assert(TruncatedElements >= D.MostDerivedPathLength &&
3032          "not casting to a derived class");
3033   if (!Result.checkSubobject(Info, E, CSK_Derived))
3034     return false;
3035 
3036   // Truncate the path to the subobject, and remove any derived-to-base offsets.
3037   const RecordDecl *RD = TruncatedType;
3038   for (unsigned I = TruncatedElements, N = D.Entries.size(); I != N; ++I) {
3039     if (RD->isInvalidDecl()) return false;
3040     const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
3041     const CXXRecordDecl *Base = getAsBaseClass(D.Entries[I]);
3042     if (isVirtualBaseClass(D.Entries[I]))
3043       Result.Offset -= Layout.getVBaseClassOffset(Base);
3044     else
3045       Result.Offset -= Layout.getBaseClassOffset(Base);
3046     RD = Base;
3047   }
3048   D.Entries.resize(TruncatedElements);
3049   return true;
3050 }
3051 
HandleLValueDirectBase(EvalInfo & Info,const Expr * E,LValue & Obj,const CXXRecordDecl * Derived,const CXXRecordDecl * Base,const ASTRecordLayout * RL=nullptr)3052 static bool HandleLValueDirectBase(EvalInfo &Info, const Expr *E, LValue &Obj,
3053                                    const CXXRecordDecl *Derived,
3054                                    const CXXRecordDecl *Base,
3055                                    const ASTRecordLayout *RL = nullptr) {
3056   if (!RL) {
3057     if (Derived->isInvalidDecl()) return false;
3058     RL = &Info.Ctx.getASTRecordLayout(Derived);
3059   }
3060 
3061   Obj.getLValueOffset() += RL->getBaseClassOffset(Base);
3062   Obj.addDecl(Info, E, Base, /*Virtual*/ false);
3063   return true;
3064 }
3065 
HandleLValueBase(EvalInfo & Info,const Expr * E,LValue & Obj,const CXXRecordDecl * DerivedDecl,const CXXBaseSpecifier * Base)3066 static bool HandleLValueBase(EvalInfo &Info, const Expr *E, LValue &Obj,
3067                              const CXXRecordDecl *DerivedDecl,
3068                              const CXXBaseSpecifier *Base) {
3069   const CXXRecordDecl *BaseDecl = Base->getType()->getAsCXXRecordDecl();
3070 
3071   if (!Base->isVirtual())
3072     return HandleLValueDirectBase(Info, E, Obj, DerivedDecl, BaseDecl);
3073 
3074   SubobjectDesignator &D = Obj.Designator;
3075   if (D.Invalid)
3076     return false;
3077 
3078   // Extract most-derived object and corresponding type.
3079   DerivedDecl = D.MostDerivedType->getAsCXXRecordDecl();
3080   if (!CastToDerivedClass(Info, E, Obj, DerivedDecl, D.MostDerivedPathLength))
3081     return false;
3082 
3083   // Find the virtual base class.
3084   if (DerivedDecl->isInvalidDecl()) return false;
3085   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(DerivedDecl);
3086   Obj.getLValueOffset() += Layout.getVBaseClassOffset(BaseDecl);
3087   Obj.addDecl(Info, E, BaseDecl, /*Virtual*/ true);
3088   return true;
3089 }
3090 
HandleLValueBasePath(EvalInfo & Info,const CastExpr * E,QualType Type,LValue & Result)3091 static bool HandleLValueBasePath(EvalInfo &Info, const CastExpr *E,
3092                                  QualType Type, LValue &Result) {
3093   for (CastExpr::path_const_iterator PathI = E->path_begin(),
3094                                      PathE = E->path_end();
3095        PathI != PathE; ++PathI) {
3096     if (!HandleLValueBase(Info, E, Result, Type->getAsCXXRecordDecl(),
3097                           *PathI))
3098       return false;
3099     Type = (*PathI)->getType();
3100   }
3101   return true;
3102 }
3103 
3104 /// Cast an lvalue referring to a derived class to a known base subobject.
CastToBaseClass(EvalInfo & Info,const Expr * E,LValue & Result,const CXXRecordDecl * DerivedRD,const CXXRecordDecl * BaseRD)3105 static bool CastToBaseClass(EvalInfo &Info, const Expr *E, LValue &Result,
3106                             const CXXRecordDecl *DerivedRD,
3107                             const CXXRecordDecl *BaseRD) {
3108   CXXBasePaths Paths(/*FindAmbiguities=*/false,
3109                      /*RecordPaths=*/true, /*DetectVirtual=*/false);
3110   if (!DerivedRD->isDerivedFrom(BaseRD, Paths))
3111     llvm_unreachable("Class must be derived from the passed in base class!");
3112 
3113   for (CXXBasePathElement &Elem : Paths.front())
3114     if (!HandleLValueBase(Info, E, Result, Elem.Class, Elem.Base))
3115       return false;
3116   return true;
3117 }
3118 
3119 /// Update LVal to refer to the given field, which must be a member of the type
3120 /// currently described by LVal.
HandleLValueMember(EvalInfo & Info,const Expr * E,LValue & LVal,const FieldDecl * FD,const ASTRecordLayout * RL=nullptr)3121 static bool HandleLValueMember(EvalInfo &Info, const Expr *E, LValue &LVal,
3122                                const FieldDecl *FD,
3123                                const ASTRecordLayout *RL = nullptr) {
3124   if (!RL) {
3125     if (FD->getParent()->isInvalidDecl()) return false;
3126     RL = &Info.Ctx.getASTRecordLayout(FD->getParent());
3127   }
3128 
3129   unsigned I = FD->getFieldIndex();
3130   LVal.adjustOffset(Info.Ctx.toCharUnitsFromBits(RL->getFieldOffset(I)));
3131   LVal.addDecl(Info, E, FD);
3132   return true;
3133 }
3134 
3135 /// Update LVal to refer to the given indirect field.
HandleLValueIndirectMember(EvalInfo & Info,const Expr * E,LValue & LVal,const IndirectFieldDecl * IFD)3136 static bool HandleLValueIndirectMember(EvalInfo &Info, const Expr *E,
3137                                        LValue &LVal,
3138                                        const IndirectFieldDecl *IFD) {
3139   for (const auto *C : IFD->chain())
3140     if (!HandleLValueMember(Info, E, LVal, cast<FieldDecl>(C)))
3141       return false;
3142   return true;
3143 }
3144 
3145 /// Get the size of the given type in char units.
HandleSizeof(EvalInfo & Info,SourceLocation Loc,QualType Type,CharUnits & Size)3146 static bool HandleSizeof(EvalInfo &Info, SourceLocation Loc,
3147                          QualType Type, CharUnits &Size) {
3148   // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
3149   // extension.
3150   if (Type->isVoidType() || Type->isFunctionType()) {
3151     Size = CharUnits::One();
3152     return true;
3153   }
3154 
3155   if (Type->isDependentType()) {
3156     Info.FFDiag(Loc);
3157     return false;
3158   }
3159 
3160   if (!Type->isConstantSizeType()) {
3161     // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
3162     // FIXME: Better diagnostic.
3163     Info.FFDiag(Loc);
3164     return false;
3165   }
3166 
3167   Size = Info.Ctx.getTypeSizeInChars(Type);
3168   return true;
3169 }
3170 
3171 /// Update a pointer value to model pointer arithmetic.
3172 /// \param Info - Information about the ongoing evaluation.
3173 /// \param E - The expression being evaluated, for diagnostic purposes.
3174 /// \param LVal - The pointer value to be updated.
3175 /// \param EltTy - The pointee type represented by LVal.
3176 /// \param Adjustment - The adjustment, in objects of type EltTy, to add.
HandleLValueArrayAdjustment(EvalInfo & Info,const Expr * E,LValue & LVal,QualType EltTy,APSInt Adjustment)3177 static bool HandleLValueArrayAdjustment(EvalInfo &Info, const Expr *E,
3178                                         LValue &LVal, QualType EltTy,
3179                                         APSInt Adjustment) {
3180   CharUnits SizeOfPointee;
3181   if (!HandleSizeof(Info, E->getExprLoc(), EltTy, SizeOfPointee))
3182     return false;
3183 
3184   LVal.adjustOffsetAndIndex(Info, E, Adjustment, SizeOfPointee);
3185   return true;
3186 }
3187 
HandleLValueArrayAdjustment(EvalInfo & Info,const Expr * E,LValue & LVal,QualType EltTy,int64_t Adjustment)3188 static bool HandleLValueArrayAdjustment(EvalInfo &Info, const Expr *E,
3189                                         LValue &LVal, QualType EltTy,
3190                                         int64_t Adjustment) {
3191   return HandleLValueArrayAdjustment(Info, E, LVal, EltTy,
3192                                      APSInt::get(Adjustment));
3193 }
3194 
3195 /// Update an lvalue to refer to a component of a complex number.
3196 /// \param Info - Information about the ongoing evaluation.
3197 /// \param LVal - The lvalue to be updated.
3198 /// \param EltTy - The complex number's component type.
3199 /// \param Imag - False for the real component, true for the imaginary.
HandleLValueComplexElement(EvalInfo & Info,const Expr * E,LValue & LVal,QualType EltTy,bool Imag)3200 static bool HandleLValueComplexElement(EvalInfo &Info, const Expr *E,
3201                                        LValue &LVal, QualType EltTy,
3202                                        bool Imag) {
3203   if (Imag) {
3204     CharUnits SizeOfComponent;
3205     if (!HandleSizeof(Info, E->getExprLoc(), EltTy, SizeOfComponent))
3206       return false;
3207     LVal.Offset += SizeOfComponent;
3208   }
3209   LVal.addComplex(Info, E, EltTy, Imag);
3210   return true;
3211 }
3212 
3213 /// Try to evaluate the initializer for a variable declaration.
3214 ///
3215 /// \param Info   Information about the ongoing evaluation.
3216 /// \param E      An expression to be used when printing diagnostics.
3217 /// \param VD     The variable whose initializer should be obtained.
3218 /// \param Version The version of the variable within the frame.
3219 /// \param Frame  The frame in which the variable was created. Must be null
3220 ///               if this variable is not local to the evaluation.
3221 /// \param Result Filled in with a pointer to the value of the variable.
evaluateVarDeclInit(EvalInfo & Info,const Expr * E,const VarDecl * VD,CallStackFrame * Frame,unsigned Version,APValue * & Result)3222 static bool evaluateVarDeclInit(EvalInfo &Info, const Expr *E,
3223                                 const VarDecl *VD, CallStackFrame *Frame,
3224                                 unsigned Version, APValue *&Result) {
3225   APValue::LValueBase Base(VD, Frame ? Frame->Index : 0, Version);
3226 
3227   // If this is a local variable, dig out its value.
3228   if (Frame) {
3229     Result = Frame->getTemporary(VD, Version);
3230     if (Result)
3231       return true;
3232 
3233     if (!isa<ParmVarDecl>(VD)) {
3234       // Assume variables referenced within a lambda's call operator that were
3235       // not declared within the call operator are captures and during checking
3236       // of a potential constant expression, assume they are unknown constant
3237       // expressions.
3238       assert(isLambdaCallOperator(Frame->Callee) &&
3239              (VD->getDeclContext() != Frame->Callee || VD->isInitCapture()) &&
3240              "missing value for local variable");
3241       if (Info.checkingPotentialConstantExpression())
3242         return false;
3243       // FIXME: This diagnostic is bogus; we do support captures. Is this code
3244       // still reachable at all?
3245       Info.FFDiag(E->getBeginLoc(),
3246                   diag::note_unimplemented_constexpr_lambda_feature_ast)
3247           << "captures not currently allowed";
3248       return false;
3249     }
3250   }
3251 
3252   // If we're currently evaluating the initializer of this declaration, use that
3253   // in-flight value.
3254   if (Info.EvaluatingDecl == Base) {
3255     Result = Info.EvaluatingDeclValue;
3256     return true;
3257   }
3258 
3259   if (isa<ParmVarDecl>(VD)) {
3260     // Assume parameters of a potential constant expression are usable in
3261     // constant expressions.
3262     if (!Info.checkingPotentialConstantExpression() ||
3263         !Info.CurrentCall->Callee ||
3264         !Info.CurrentCall->Callee->Equals(VD->getDeclContext())) {
3265       if (Info.getLangOpts().CPlusPlus11) {
3266         Info.FFDiag(E, diag::note_constexpr_function_param_value_unknown)
3267             << VD;
3268         NoteLValueLocation(Info, Base);
3269       } else {
3270         Info.FFDiag(E);
3271       }
3272     }
3273     return false;
3274   }
3275 
3276   // Dig out the initializer, and use the declaration which it's attached to.
3277   // FIXME: We should eventually check whether the variable has a reachable
3278   // initializing declaration.
3279   const Expr *Init = VD->getAnyInitializer(VD);
3280   if (!Init) {
3281     // Don't diagnose during potential constant expression checking; an
3282     // initializer might be added later.
3283     if (!Info.checkingPotentialConstantExpression()) {
3284       Info.FFDiag(E, diag::note_constexpr_var_init_unknown, 1)
3285         << VD;
3286       NoteLValueLocation(Info, Base);
3287     }
3288     return false;
3289   }
3290 
3291   if (Init->isValueDependent()) {
3292     // The DeclRefExpr is not value-dependent, but the variable it refers to
3293     // has a value-dependent initializer. This should only happen in
3294     // constant-folding cases, where the variable is not actually of a suitable
3295     // type for use in a constant expression (otherwise the DeclRefExpr would
3296     // have been value-dependent too), so diagnose that.
3297     assert(!VD->mightBeUsableInConstantExpressions(Info.Ctx));
3298     if (!Info.checkingPotentialConstantExpression()) {
3299       Info.FFDiag(E, Info.getLangOpts().CPlusPlus11
3300                          ? diag::note_constexpr_ltor_non_constexpr
3301                          : diag::note_constexpr_ltor_non_integral, 1)
3302           << VD << VD->getType();
3303       NoteLValueLocation(Info, Base);
3304     }
3305     return false;
3306   }
3307 
3308   // Check that we can fold the initializer. In C++, we will have already done
3309   // this in the cases where it matters for conformance.
3310   SmallVector<PartialDiagnosticAt, 8> Notes;
3311   if (!VD->evaluateValue(Notes)) {
3312     Info.FFDiag(E, diag::note_constexpr_var_init_non_constant,
3313               Notes.size() + 1) << VD;
3314     NoteLValueLocation(Info, Base);
3315     Info.addNotes(Notes);
3316     return false;
3317   }
3318 
3319   // Check that the variable is actually usable in constant expressions. For a
3320   // const integral variable or a reference, we might have a non-constant
3321   // initializer that we can nonetheless evaluate the initializer for. Such
3322   // variables are not usable in constant expressions. In C++98, the
3323   // initializer also syntactically needs to be an ICE.
3324   //
3325   // FIXME: We don't diagnose cases that aren't potentially usable in constant
3326   // expressions here; doing so would regress diagnostics for things like
3327   // reading from a volatile constexpr variable.
3328   if ((Info.getLangOpts().CPlusPlus && !VD->hasConstantInitialization() &&
3329        VD->mightBeUsableInConstantExpressions(Info.Ctx)) ||
3330       ((Info.getLangOpts().CPlusPlus || Info.getLangOpts().OpenCL) &&
3331        !Info.getLangOpts().CPlusPlus11 && !VD->hasICEInitializer(Info.Ctx))) {
3332     Info.CCEDiag(E, diag::note_constexpr_var_init_non_constant, 1) << VD;
3333     NoteLValueLocation(Info, Base);
3334   }
3335 
3336   // Never use the initializer of a weak variable, not even for constant
3337   // folding. We can't be sure that this is the definition that will be used.
3338   if (VD->isWeak()) {
3339     Info.FFDiag(E, diag::note_constexpr_var_init_weak) << VD;
3340     NoteLValueLocation(Info, Base);
3341     return false;
3342   }
3343 
3344   Result = VD->getEvaluatedValue();
3345   return true;
3346 }
3347 
3348 /// Get the base index of the given base class within an APValue representing
3349 /// the given derived class.
getBaseIndex(const CXXRecordDecl * Derived,const CXXRecordDecl * Base)3350 static unsigned getBaseIndex(const CXXRecordDecl *Derived,
3351                              const CXXRecordDecl *Base) {
3352   Base = Base->getCanonicalDecl();
3353   unsigned Index = 0;
3354   for (CXXRecordDecl::base_class_const_iterator I = Derived->bases_begin(),
3355          E = Derived->bases_end(); I != E; ++I, ++Index) {
3356     if (I->getType()->getAsCXXRecordDecl()->getCanonicalDecl() == Base)
3357       return Index;
3358   }
3359 
3360   llvm_unreachable("base class missing from derived class's bases list");
3361 }
3362 
3363 /// Extract the value of a character from a string literal.
extractStringLiteralCharacter(EvalInfo & Info,const Expr * Lit,uint64_t Index)3364 static APSInt extractStringLiteralCharacter(EvalInfo &Info, const Expr *Lit,
3365                                             uint64_t Index) {
3366   assert(!isa<SourceLocExpr>(Lit) &&
3367          "SourceLocExpr should have already been converted to a StringLiteral");
3368 
3369   // FIXME: Support MakeStringConstant
3370   if (const auto *ObjCEnc = dyn_cast<ObjCEncodeExpr>(Lit)) {
3371     std::string Str;
3372     Info.Ctx.getObjCEncodingForType(ObjCEnc->getEncodedType(), Str);
3373     assert(Index <= Str.size() && "Index too large");
3374     return APSInt::getUnsigned(Str.c_str()[Index]);
3375   }
3376 
3377   if (auto PE = dyn_cast<PredefinedExpr>(Lit))
3378     Lit = PE->getFunctionName();
3379   const StringLiteral *S = cast<StringLiteral>(Lit);
3380   const ConstantArrayType *CAT =
3381       Info.Ctx.getAsConstantArrayType(S->getType());
3382   assert(CAT && "string literal isn't an array");
3383   QualType CharType = CAT->getElementType();
3384   assert(CharType->isIntegerType() && "unexpected character type");
3385 
3386   APSInt Value(S->getCharByteWidth() * Info.Ctx.getCharWidth(),
3387                CharType->isUnsignedIntegerType());
3388   if (Index < S->getLength())
3389     Value = S->getCodeUnit(Index);
3390   return Value;
3391 }
3392 
3393 // Expand a string literal into an array of characters.
3394 //
3395 // FIXME: This is inefficient; we should probably introduce something similar
3396 // to the LLVM ConstantDataArray to make this cheaper.
expandStringLiteral(EvalInfo & Info,const StringLiteral * S,APValue & Result,QualType AllocType=QualType ())3397 static void expandStringLiteral(EvalInfo &Info, const StringLiteral *S,
3398                                 APValue &Result,
3399                                 QualType AllocType = QualType()) {
3400   const ConstantArrayType *CAT = Info.Ctx.getAsConstantArrayType(
3401       AllocType.isNull() ? S->getType() : AllocType);
3402   assert(CAT && "string literal isn't an array");
3403   QualType CharType = CAT->getElementType();
3404   assert(CharType->isIntegerType() && "unexpected character type");
3405 
3406   unsigned Elts = CAT->getSize().getZExtValue();
3407   Result = APValue(APValue::UninitArray(),
3408                    std::min(S->getLength(), Elts), Elts);
3409   APSInt Value(S->getCharByteWidth() * Info.Ctx.getCharWidth(),
3410                CharType->isUnsignedIntegerType());
3411   if (Result.hasArrayFiller())
3412     Result.getArrayFiller() = APValue(Value);
3413   for (unsigned I = 0, N = Result.getArrayInitializedElts(); I != N; ++I) {
3414     Value = S->getCodeUnit(I);
3415     Result.getArrayInitializedElt(I) = APValue(Value);
3416   }
3417 }
3418 
3419 // Expand an array so that it has more than Index filled elements.
expandArray(APValue & Array,unsigned Index)3420 static void expandArray(APValue &Array, unsigned Index) {
3421   unsigned Size = Array.getArraySize();
3422   assert(Index < Size);
3423 
3424   // Always at least double the number of elements for which we store a value.
3425   unsigned OldElts = Array.getArrayInitializedElts();
3426   unsigned NewElts = std::max(Index+1, OldElts * 2);
3427   NewElts = std::min(Size, std::max(NewElts, 8u));
3428 
3429   // Copy the data across.
3430   APValue NewValue(APValue::UninitArray(), NewElts, Size);
3431   for (unsigned I = 0; I != OldElts; ++I)
3432     NewValue.getArrayInitializedElt(I).swap(Array.getArrayInitializedElt(I));
3433   for (unsigned I = OldElts; I != NewElts; ++I)
3434     NewValue.getArrayInitializedElt(I) = Array.getArrayFiller();
3435   if (NewValue.hasArrayFiller())
3436     NewValue.getArrayFiller() = Array.getArrayFiller();
3437   Array.swap(NewValue);
3438 }
3439 
3440 /// Determine whether a type would actually be read by an lvalue-to-rvalue
3441 /// conversion. If it's of class type, we may assume that the copy operation
3442 /// is trivial. Note that this is never true for a union type with fields
3443 /// (because the copy always "reads" the active member) and always true for
3444 /// a non-class type.
3445 static bool isReadByLvalueToRvalueConversion(const CXXRecordDecl *RD);
isReadByLvalueToRvalueConversion(QualType T)3446 static bool isReadByLvalueToRvalueConversion(QualType T) {
3447   CXXRecordDecl *RD = T->getBaseElementTypeUnsafe()->getAsCXXRecordDecl();
3448   return !RD || isReadByLvalueToRvalueConversion(RD);
3449 }
isReadByLvalueToRvalueConversion(const CXXRecordDecl * RD)3450 static bool isReadByLvalueToRvalueConversion(const CXXRecordDecl *RD) {
3451   // FIXME: A trivial copy of a union copies the object representation, even if
3452   // the union is empty.
3453   if (RD->isUnion())
3454     return !RD->field_empty();
3455   if (RD->isEmpty())
3456     return false;
3457 
3458   for (auto *Field : RD->fields())
3459     if (!Field->isUnnamedBitfield() &&
3460         isReadByLvalueToRvalueConversion(Field->getType()))
3461       return true;
3462 
3463   for (auto &BaseSpec : RD->bases())
3464     if (isReadByLvalueToRvalueConversion(BaseSpec.getType()))
3465       return true;
3466 
3467   return false;
3468 }
3469 
3470 /// Diagnose an attempt to read from any unreadable field within the specified
3471 /// type, which might be a class type.
diagnoseMutableFields(EvalInfo & Info,const Expr * E,AccessKinds AK,QualType T)3472 static bool diagnoseMutableFields(EvalInfo &Info, const Expr *E, AccessKinds AK,
3473                                   QualType T) {
3474   CXXRecordDecl *RD = T->getBaseElementTypeUnsafe()->getAsCXXRecordDecl();
3475   if (!RD)
3476     return false;
3477 
3478   if (!RD->hasMutableFields())
3479     return false;
3480 
3481   for (auto *Field : RD->fields()) {
3482     // If we're actually going to read this field in some way, then it can't
3483     // be mutable. If we're in a union, then assigning to a mutable field
3484     // (even an empty one) can change the active member, so that's not OK.
3485     // FIXME: Add core issue number for the union case.
3486     if (Field->isMutable() &&
3487         (RD->isUnion() || isReadByLvalueToRvalueConversion(Field->getType()))) {
3488       Info.FFDiag(E, diag::note_constexpr_access_mutable, 1) << AK << Field;
3489       Info.Note(Field->getLocation(), diag::note_declared_at);
3490       return true;
3491     }
3492 
3493     if (diagnoseMutableFields(Info, E, AK, Field->getType()))
3494       return true;
3495   }
3496 
3497   for (auto &BaseSpec : RD->bases())
3498     if (diagnoseMutableFields(Info, E, AK, BaseSpec.getType()))
3499       return true;
3500 
3501   // All mutable fields were empty, and thus not actually read.
3502   return false;
3503 }
3504 
lifetimeStartedInEvaluation(EvalInfo & Info,APValue::LValueBase Base,bool MutableSubobject=false)3505 static bool lifetimeStartedInEvaluation(EvalInfo &Info,
3506                                         APValue::LValueBase Base,
3507                                         bool MutableSubobject = false) {
3508   // A temporary or transient heap allocation we created.
3509   if (Base.getCallIndex() || Base.is<DynamicAllocLValue>())
3510     return true;
3511 
3512   switch (Info.IsEvaluatingDecl) {
3513   case EvalInfo::EvaluatingDeclKind::None:
3514     return false;
3515 
3516   case EvalInfo::EvaluatingDeclKind::Ctor:
3517     // The variable whose initializer we're evaluating.
3518     if (Info.EvaluatingDecl == Base)
3519       return true;
3520 
3521     // A temporary lifetime-extended by the variable whose initializer we're
3522     // evaluating.
3523     if (auto *BaseE = Base.dyn_cast<const Expr *>())
3524       if (auto *BaseMTE = dyn_cast<MaterializeTemporaryExpr>(BaseE))
3525         return Info.EvaluatingDecl == BaseMTE->getExtendingDecl();
3526     return false;
3527 
3528   case EvalInfo::EvaluatingDeclKind::Dtor:
3529     // C++2a [expr.const]p6:
3530     //   [during constant destruction] the lifetime of a and its non-mutable
3531     //   subobjects (but not its mutable subobjects) [are] considered to start
3532     //   within e.
3533     if (MutableSubobject || Base != Info.EvaluatingDecl)
3534       return false;
3535     // FIXME: We can meaningfully extend this to cover non-const objects, but
3536     // we will need special handling: we should be able to access only
3537     // subobjects of such objects that are themselves declared const.
3538     QualType T = getType(Base);
3539     return T.isConstQualified() || T->isReferenceType();
3540   }
3541 
3542   llvm_unreachable("unknown evaluating decl kind");
3543 }
3544 
3545 namespace {
3546 /// A handle to a complete object (an object that is not a subobject of
3547 /// another object).
3548 struct CompleteObject {
3549   /// The identity of the object.
3550   APValue::LValueBase Base;
3551   /// The value of the complete object.
3552   APValue *Value;
3553   /// The type of the complete object.
3554   QualType Type;
3555 
CompleteObject__anonb66d72d20911::CompleteObject3556   CompleteObject() : Value(nullptr) {}
CompleteObject__anonb66d72d20911::CompleteObject3557   CompleteObject(APValue::LValueBase Base, APValue *Value, QualType Type)
3558       : Base(Base), Value(Value), Type(Type) {}
3559 
mayAccessMutableMembers__anonb66d72d20911::CompleteObject3560   bool mayAccessMutableMembers(EvalInfo &Info, AccessKinds AK) const {
3561     // If this isn't a "real" access (eg, if it's just accessing the type
3562     // info), allow it. We assume the type doesn't change dynamically for
3563     // subobjects of constexpr objects (even though we'd hit UB here if it
3564     // did). FIXME: Is this right?
3565     if (!isAnyAccess(AK))
3566       return true;
3567 
3568     // In C++14 onwards, it is permitted to read a mutable member whose
3569     // lifetime began within the evaluation.
3570     // FIXME: Should we also allow this in C++11?
3571     if (!Info.getLangOpts().CPlusPlus14)
3572       return false;
3573     return lifetimeStartedInEvaluation(Info, Base, /*MutableSubobject*/true);
3574   }
3575 
operator bool__anonb66d72d20911::CompleteObject3576   explicit operator bool() const { return !Type.isNull(); }
3577 };
3578 } // end anonymous namespace
3579 
getSubobjectType(QualType ObjType,QualType SubobjType,bool IsMutable=false)3580 static QualType getSubobjectType(QualType ObjType, QualType SubobjType,
3581                                  bool IsMutable = false) {
3582   // C++ [basic.type.qualifier]p1:
3583   // - A const object is an object of type const T or a non-mutable subobject
3584   //   of a const object.
3585   if (ObjType.isConstQualified() && !IsMutable)
3586     SubobjType.addConst();
3587   // - A volatile object is an object of type const T or a subobject of a
3588   //   volatile object.
3589   if (ObjType.isVolatileQualified())
3590     SubobjType.addVolatile();
3591   return SubobjType;
3592 }
3593 
3594 /// Find the designated sub-object of an rvalue.
3595 template<typename SubobjectHandler>
3596 typename SubobjectHandler::result_type
findSubobject(EvalInfo & Info,const Expr * E,const CompleteObject & Obj,const SubobjectDesignator & Sub,SubobjectHandler & handler)3597 findSubobject(EvalInfo &Info, const Expr *E, const CompleteObject &Obj,
3598               const SubobjectDesignator &Sub, SubobjectHandler &handler) {
3599   if (Sub.Invalid)
3600     // A diagnostic will have already been produced.
3601     return handler.failed();
3602   if (Sub.isOnePastTheEnd() || Sub.isMostDerivedAnUnsizedArray()) {
3603     if (Info.getLangOpts().CPlusPlus11)
3604       Info.FFDiag(E, Sub.isOnePastTheEnd()
3605                          ? diag::note_constexpr_access_past_end
3606                          : diag::note_constexpr_access_unsized_array)
3607           << handler.AccessKind;
3608     else
3609       Info.FFDiag(E);
3610     return handler.failed();
3611   }
3612 
3613   APValue *O = Obj.Value;
3614   QualType ObjType = Obj.Type;
3615   const FieldDecl *LastField = nullptr;
3616   const FieldDecl *VolatileField = nullptr;
3617 
3618   // Walk the designator's path to find the subobject.
3619   for (unsigned I = 0, N = Sub.Entries.size(); /**/; ++I) {
3620     // Reading an indeterminate value is undefined, but assigning over one is OK.
3621     if ((O->isAbsent() && !(handler.AccessKind == AK_Construct && I == N)) ||
3622         (O->isIndeterminate() &&
3623          !isValidIndeterminateAccess(handler.AccessKind))) {
3624       if (!Info.checkingPotentialConstantExpression())
3625         Info.FFDiag(E, diag::note_constexpr_access_uninit)
3626             << handler.AccessKind << O->isIndeterminate();
3627       return handler.failed();
3628     }
3629 
3630     // C++ [class.ctor]p5, C++ [class.dtor]p5:
3631     //    const and volatile semantics are not applied on an object under
3632     //    {con,de}struction.
3633     if ((ObjType.isConstQualified() || ObjType.isVolatileQualified()) &&
3634         ObjType->isRecordType() &&
3635         Info.isEvaluatingCtorDtor(
3636             Obj.Base, llvm::makeArrayRef(Sub.Entries.begin(),
3637                                          Sub.Entries.begin() + I)) !=
3638                           ConstructionPhase::None) {
3639       ObjType = Info.Ctx.getCanonicalType(ObjType);
3640       ObjType.removeLocalConst();
3641       ObjType.removeLocalVolatile();
3642     }
3643 
3644     // If this is our last pass, check that the final object type is OK.
3645     if (I == N || (I == N - 1 && ObjType->isAnyComplexType())) {
3646       // Accesses to volatile objects are prohibited.
3647       if (ObjType.isVolatileQualified() && isFormalAccess(handler.AccessKind)) {
3648         if (Info.getLangOpts().CPlusPlus) {
3649           int DiagKind;
3650           SourceLocation Loc;
3651           const NamedDecl *Decl = nullptr;
3652           if (VolatileField) {
3653             DiagKind = 2;
3654             Loc = VolatileField->getLocation();
3655             Decl = VolatileField;
3656           } else if (auto *VD = Obj.Base.dyn_cast<const ValueDecl*>()) {
3657             DiagKind = 1;
3658             Loc = VD->getLocation();
3659             Decl = VD;
3660           } else {
3661             DiagKind = 0;
3662             if (auto *E = Obj.Base.dyn_cast<const Expr *>())
3663               Loc = E->getExprLoc();
3664           }
3665           Info.FFDiag(E, diag::note_constexpr_access_volatile_obj, 1)
3666               << handler.AccessKind << DiagKind << Decl;
3667           Info.Note(Loc, diag::note_constexpr_volatile_here) << DiagKind;
3668         } else {
3669           Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
3670         }
3671         return handler.failed();
3672       }
3673 
3674       // If we are reading an object of class type, there may still be more
3675       // things we need to check: if there are any mutable subobjects, we
3676       // cannot perform this read. (This only happens when performing a trivial
3677       // copy or assignment.)
3678       if (ObjType->isRecordType() &&
3679           !Obj.mayAccessMutableMembers(Info, handler.AccessKind) &&
3680           diagnoseMutableFields(Info, E, handler.AccessKind, ObjType))
3681         return handler.failed();
3682     }
3683 
3684     if (I == N) {
3685       if (!handler.found(*O, ObjType))
3686         return false;
3687 
3688       // If we modified a bit-field, truncate it to the right width.
3689       if (isModification(handler.AccessKind) &&
3690           LastField && LastField->isBitField() &&
3691           !truncateBitfieldValue(Info, E, *O, LastField))
3692         return false;
3693 
3694       return true;
3695     }
3696 
3697     LastField = nullptr;
3698     if (ObjType->isArrayType()) {
3699       // Next subobject is an array element.
3700       const ConstantArrayType *CAT = Info.Ctx.getAsConstantArrayType(ObjType);
3701       assert(CAT && "vla in literal type?");
3702       uint64_t Index = Sub.Entries[I].getAsArrayIndex();
3703       if (CAT->getSize().ule(Index)) {
3704         // Note, it should not be possible to form a pointer with a valid
3705         // designator which points more than one past the end of the array.
3706         if (Info.getLangOpts().CPlusPlus11)
3707           Info.FFDiag(E, diag::note_constexpr_access_past_end)
3708             << handler.AccessKind;
3709         else
3710           Info.FFDiag(E);
3711         return handler.failed();
3712       }
3713 
3714       ObjType = CAT->getElementType();
3715 
3716       if (O->getArrayInitializedElts() > Index)
3717         O = &O->getArrayInitializedElt(Index);
3718       else if (!isRead(handler.AccessKind)) {
3719         expandArray(*O, Index);
3720         O = &O->getArrayInitializedElt(Index);
3721       } else
3722         O = &O->getArrayFiller();
3723     } else if (ObjType->isAnyComplexType()) {
3724       // Next subobject is a complex number.
3725       uint64_t Index = Sub.Entries[I].getAsArrayIndex();
3726       if (Index > 1) {
3727         if (Info.getLangOpts().CPlusPlus11)
3728           Info.FFDiag(E, diag::note_constexpr_access_past_end)
3729             << handler.AccessKind;
3730         else
3731           Info.FFDiag(E);
3732         return handler.failed();
3733       }
3734 
3735       ObjType = getSubobjectType(
3736           ObjType, ObjType->castAs<ComplexType>()->getElementType());
3737 
3738       assert(I == N - 1 && "extracting subobject of scalar?");
3739       if (O->isComplexInt()) {
3740         return handler.found(Index ? O->getComplexIntImag()
3741                                    : O->getComplexIntReal(), ObjType);
3742       } else {
3743         assert(O->isComplexFloat());
3744         return handler.found(Index ? O->getComplexFloatImag()
3745                                    : O->getComplexFloatReal(), ObjType);
3746       }
3747     } else if (const FieldDecl *Field = getAsField(Sub.Entries[I])) {
3748       if (Field->isMutable() &&
3749           !Obj.mayAccessMutableMembers(Info, handler.AccessKind)) {
3750         Info.FFDiag(E, diag::note_constexpr_access_mutable, 1)
3751           << handler.AccessKind << Field;
3752         Info.Note(Field->getLocation(), diag::note_declared_at);
3753         return handler.failed();
3754       }
3755 
3756       // Next subobject is a class, struct or union field.
3757       RecordDecl *RD = ObjType->castAs<RecordType>()->getDecl();
3758       if (RD->isUnion()) {
3759         const FieldDecl *UnionField = O->getUnionField();
3760         if (!UnionField ||
3761             UnionField->getCanonicalDecl() != Field->getCanonicalDecl()) {
3762           if (I == N - 1 && handler.AccessKind == AK_Construct) {
3763             // Placement new onto an inactive union member makes it active.
3764             O->setUnion(Field, APValue());
3765           } else {
3766             // FIXME: If O->getUnionValue() is absent, report that there's no
3767             // active union member rather than reporting the prior active union
3768             // member. We'll need to fix nullptr_t to not use APValue() as its
3769             // representation first.
3770             Info.FFDiag(E, diag::note_constexpr_access_inactive_union_member)
3771                 << handler.AccessKind << Field << !UnionField << UnionField;
3772             return handler.failed();
3773           }
3774         }
3775         O = &O->getUnionValue();
3776       } else
3777         O = &O->getStructField(Field->getFieldIndex());
3778 
3779       ObjType = getSubobjectType(ObjType, Field->getType(), Field->isMutable());
3780       LastField = Field;
3781       if (Field->getType().isVolatileQualified())
3782         VolatileField = Field;
3783     } else {
3784       // Next subobject is a base class.
3785       const CXXRecordDecl *Derived = ObjType->getAsCXXRecordDecl();
3786       const CXXRecordDecl *Base = getAsBaseClass(Sub.Entries[I]);
3787       O = &O->getStructBase(getBaseIndex(Derived, Base));
3788 
3789       ObjType = getSubobjectType(ObjType, Info.Ctx.getRecordType(Base));
3790     }
3791   }
3792 }
3793 
3794 namespace {
3795 struct ExtractSubobjectHandler {
3796   EvalInfo &Info;
3797   const Expr *E;
3798   APValue &Result;
3799   const AccessKinds AccessKind;
3800 
3801   typedef bool result_type;
failed__anonb66d72d20a11::ExtractSubobjectHandler3802   bool failed() { return false; }
found__anonb66d72d20a11::ExtractSubobjectHandler3803   bool found(APValue &Subobj, QualType SubobjType) {
3804     Result = Subobj;
3805     if (AccessKind == AK_ReadObjectRepresentation)
3806       return true;
3807     return CheckFullyInitialized(Info, E->getExprLoc(), SubobjType, Result);
3808   }
found__anonb66d72d20a11::ExtractSubobjectHandler3809   bool found(APSInt &Value, QualType SubobjType) {
3810     Result = APValue(Value);
3811     return true;
3812   }
found__anonb66d72d20a11::ExtractSubobjectHandler3813   bool found(APFloat &Value, QualType SubobjType) {
3814     Result = APValue(Value);
3815     return true;
3816   }
3817 };
3818 } // end anonymous namespace
3819 
3820 /// Extract the designated sub-object of an rvalue.
extractSubobject(EvalInfo & Info,const Expr * E,const CompleteObject & Obj,const SubobjectDesignator & Sub,APValue & Result,AccessKinds AK=AK_Read)3821 static bool extractSubobject(EvalInfo &Info, const Expr *E,
3822                              const CompleteObject &Obj,
3823                              const SubobjectDesignator &Sub, APValue &Result,
3824                              AccessKinds AK = AK_Read) {
3825   assert(AK == AK_Read || AK == AK_ReadObjectRepresentation);
3826   ExtractSubobjectHandler Handler = {Info, E, Result, AK};
3827   return findSubobject(Info, E, Obj, Sub, Handler);
3828 }
3829 
3830 namespace {
3831 struct ModifySubobjectHandler {
3832   EvalInfo &Info;
3833   APValue &NewVal;
3834   const Expr *E;
3835 
3836   typedef bool result_type;
3837   static const AccessKinds AccessKind = AK_Assign;
3838 
checkConst__anonb66d72d20b11::ModifySubobjectHandler3839   bool checkConst(QualType QT) {
3840     // Assigning to a const object has undefined behavior.
3841     if (QT.isConstQualified()) {
3842       Info.FFDiag(E, diag::note_constexpr_modify_const_type) << QT;
3843       return false;
3844     }
3845     return true;
3846   }
3847 
failed__anonb66d72d20b11::ModifySubobjectHandler3848   bool failed() { return false; }
found__anonb66d72d20b11::ModifySubobjectHandler3849   bool found(APValue &Subobj, QualType SubobjType) {
3850     if (!checkConst(SubobjType))
3851       return false;
3852     // We've been given ownership of NewVal, so just swap it in.
3853     Subobj.swap(NewVal);
3854     return true;
3855   }
found__anonb66d72d20b11::ModifySubobjectHandler3856   bool found(APSInt &Value, QualType SubobjType) {
3857     if (!checkConst(SubobjType))
3858       return false;
3859     if (!NewVal.isInt()) {
3860       // Maybe trying to write a cast pointer value into a complex?
3861       Info.FFDiag(E);
3862       return false;
3863     }
3864     Value = NewVal.getInt();
3865     return true;
3866   }
found__anonb66d72d20b11::ModifySubobjectHandler3867   bool found(APFloat &Value, QualType SubobjType) {
3868     if (!checkConst(SubobjType))
3869       return false;
3870     Value = NewVal.getFloat();
3871     return true;
3872   }
3873 };
3874 } // end anonymous namespace
3875 
3876 const AccessKinds ModifySubobjectHandler::AccessKind;
3877 
3878 /// Update the designated sub-object of an rvalue to the given value.
modifySubobject(EvalInfo & Info,const Expr * E,const CompleteObject & Obj,const SubobjectDesignator & Sub,APValue & NewVal)3879 static bool modifySubobject(EvalInfo &Info, const Expr *E,
3880                             const CompleteObject &Obj,
3881                             const SubobjectDesignator &Sub,
3882                             APValue &NewVal) {
3883   ModifySubobjectHandler Handler = { Info, NewVal, E };
3884   return findSubobject(Info, E, Obj, Sub, Handler);
3885 }
3886 
3887 /// Find the position where two subobject designators diverge, or equivalently
3888 /// the length of the common initial subsequence.
FindDesignatorMismatch(QualType ObjType,const SubobjectDesignator & A,const SubobjectDesignator & B,bool & WasArrayIndex)3889 static unsigned FindDesignatorMismatch(QualType ObjType,
3890                                        const SubobjectDesignator &A,
3891                                        const SubobjectDesignator &B,
3892                                        bool &WasArrayIndex) {
3893   unsigned I = 0, N = std::min(A.Entries.size(), B.Entries.size());
3894   for (/**/; I != N; ++I) {
3895     if (!ObjType.isNull() &&
3896         (ObjType->isArrayType() || ObjType->isAnyComplexType())) {
3897       // Next subobject is an array element.
3898       if (A.Entries[I].getAsArrayIndex() != B.Entries[I].getAsArrayIndex()) {
3899         WasArrayIndex = true;
3900         return I;
3901       }
3902       if (ObjType->isAnyComplexType())
3903         ObjType = ObjType->castAs<ComplexType>()->getElementType();
3904       else
3905         ObjType = ObjType->castAsArrayTypeUnsafe()->getElementType();
3906     } else {
3907       if (A.Entries[I].getAsBaseOrMember() !=
3908           B.Entries[I].getAsBaseOrMember()) {
3909         WasArrayIndex = false;
3910         return I;
3911       }
3912       if (const FieldDecl *FD = getAsField(A.Entries[I]))
3913         // Next subobject is a field.
3914         ObjType = FD->getType();
3915       else
3916         // Next subobject is a base class.
3917         ObjType = QualType();
3918     }
3919   }
3920   WasArrayIndex = false;
3921   return I;
3922 }
3923 
3924 /// Determine whether the given subobject designators refer to elements of the
3925 /// same array object.
AreElementsOfSameArray(QualType ObjType,const SubobjectDesignator & A,const SubobjectDesignator & B)3926 static bool AreElementsOfSameArray(QualType ObjType,
3927                                    const SubobjectDesignator &A,
3928                                    const SubobjectDesignator &B) {
3929   if (A.Entries.size() != B.Entries.size())
3930     return false;
3931 
3932   bool IsArray = A.MostDerivedIsArrayElement;
3933   if (IsArray && A.MostDerivedPathLength != A.Entries.size())
3934     // A is a subobject of the array element.
3935     return false;
3936 
3937   // If A (and B) designates an array element, the last entry will be the array
3938   // index. That doesn't have to match. Otherwise, we're in the 'implicit array
3939   // of length 1' case, and the entire path must match.
3940   bool WasArrayIndex;
3941   unsigned CommonLength = FindDesignatorMismatch(ObjType, A, B, WasArrayIndex);
3942   return CommonLength >= A.Entries.size() - IsArray;
3943 }
3944 
3945 /// Find the complete object to which an LValue refers.
findCompleteObject(EvalInfo & Info,const Expr * E,AccessKinds AK,const LValue & LVal,QualType LValType)3946 static CompleteObject findCompleteObject(EvalInfo &Info, const Expr *E,
3947                                          AccessKinds AK, const LValue &LVal,
3948                                          QualType LValType) {
3949   if (LVal.InvalidBase) {
3950     Info.FFDiag(E);
3951     return CompleteObject();
3952   }
3953 
3954   if (!LVal.Base) {
3955     Info.FFDiag(E, diag::note_constexpr_access_null) << AK;
3956     return CompleteObject();
3957   }
3958 
3959   CallStackFrame *Frame = nullptr;
3960   unsigned Depth = 0;
3961   if (LVal.getLValueCallIndex()) {
3962     std::tie(Frame, Depth) =
3963         Info.getCallFrameAndDepth(LVal.getLValueCallIndex());
3964     if (!Frame) {
3965       Info.FFDiag(E, diag::note_constexpr_lifetime_ended, 1)
3966         << AK << LVal.Base.is<const ValueDecl*>();
3967       NoteLValueLocation(Info, LVal.Base);
3968       return CompleteObject();
3969     }
3970   }
3971 
3972   bool IsAccess = isAnyAccess(AK);
3973 
3974   // C++11 DR1311: An lvalue-to-rvalue conversion on a volatile-qualified type
3975   // is not a constant expression (even if the object is non-volatile). We also
3976   // apply this rule to C++98, in order to conform to the expected 'volatile'
3977   // semantics.
3978   if (isFormalAccess(AK) && LValType.isVolatileQualified()) {
3979     if (Info.getLangOpts().CPlusPlus)
3980       Info.FFDiag(E, diag::note_constexpr_access_volatile_type)
3981         << AK << LValType;
3982     else
3983       Info.FFDiag(E);
3984     return CompleteObject();
3985   }
3986 
3987   // Compute value storage location and type of base object.
3988   APValue *BaseVal = nullptr;
3989   QualType BaseType = getType(LVal.Base);
3990 
3991   if (Info.getLangOpts().CPlusPlus14 && LVal.Base == Info.EvaluatingDecl &&
3992       lifetimeStartedInEvaluation(Info, LVal.Base)) {
3993     // This is the object whose initializer we're evaluating, so its lifetime
3994     // started in the current evaluation.
3995     BaseVal = Info.EvaluatingDeclValue;
3996   } else if (const ValueDecl *D = LVal.Base.dyn_cast<const ValueDecl *>()) {
3997     // Allow reading from a GUID declaration.
3998     if (auto *GD = dyn_cast<MSGuidDecl>(D)) {
3999       if (isModification(AK)) {
4000         // All the remaining cases do not permit modification of the object.
4001         Info.FFDiag(E, diag::note_constexpr_modify_global);
4002         return CompleteObject();
4003       }
4004       APValue &V = GD->getAsAPValue();
4005       if (V.isAbsent()) {
4006         Info.FFDiag(E, diag::note_constexpr_unsupported_layout)
4007             << GD->getType();
4008         return CompleteObject();
4009       }
4010       return CompleteObject(LVal.Base, &V, GD->getType());
4011     }
4012 
4013     // Allow reading from template parameter objects.
4014     if (auto *TPO = dyn_cast<TemplateParamObjectDecl>(D)) {
4015       if (isModification(AK)) {
4016         Info.FFDiag(E, diag::note_constexpr_modify_global);
4017         return CompleteObject();
4018       }
4019       return CompleteObject(LVal.Base, const_cast<APValue *>(&TPO->getValue()),
4020                             TPO->getType());
4021     }
4022 
4023     // In C++98, const, non-volatile integers initialized with ICEs are ICEs.
4024     // In C++11, constexpr, non-volatile variables initialized with constant
4025     // expressions are constant expressions too. Inside constexpr functions,
4026     // parameters are constant expressions even if they're non-const.
4027     // In C++1y, objects local to a constant expression (those with a Frame) are
4028     // both readable and writable inside constant expressions.
4029     // In C, such things can also be folded, although they are not ICEs.
4030     const VarDecl *VD = dyn_cast<VarDecl>(D);
4031     if (VD) {
4032       if (const VarDecl *VDef = VD->getDefinition(Info.Ctx))
4033         VD = VDef;
4034     }
4035     if (!VD || VD->isInvalidDecl()) {
4036       Info.FFDiag(E);
4037       return CompleteObject();
4038     }
4039 
4040     bool IsConstant = BaseType.isConstant(Info.Ctx);
4041 
4042     // Unless we're looking at a local variable or argument in a constexpr call,
4043     // the variable we're reading must be const.
4044     if (!Frame) {
4045       if (IsAccess && isa<ParmVarDecl>(VD)) {
4046         // Access of a parameter that's not associated with a frame isn't going
4047         // to work out, but we can leave it to evaluateVarDeclInit to provide a
4048         // suitable diagnostic.
4049       } else if (Info.getLangOpts().CPlusPlus14 &&
4050                  lifetimeStartedInEvaluation(Info, LVal.Base)) {
4051         // OK, we can read and modify an object if we're in the process of
4052         // evaluating its initializer, because its lifetime began in this
4053         // evaluation.
4054       } else if (isModification(AK)) {
4055         // All the remaining cases do not permit modification of the object.
4056         Info.FFDiag(E, diag::note_constexpr_modify_global);
4057         return CompleteObject();
4058       } else if (VD->isConstexpr()) {
4059         // OK, we can read this variable.
4060       } else if (BaseType->isIntegralOrEnumerationType()) {
4061         if (!IsConstant) {
4062           if (!IsAccess)
4063             return CompleteObject(LVal.getLValueBase(), nullptr, BaseType);
4064           if (Info.getLangOpts().CPlusPlus) {
4065             Info.FFDiag(E, diag::note_constexpr_ltor_non_const_int, 1) << VD;
4066             Info.Note(VD->getLocation(), diag::note_declared_at);
4067           } else {
4068             Info.FFDiag(E);
4069           }
4070           return CompleteObject();
4071         }
4072       } else if (!IsAccess) {
4073         return CompleteObject(LVal.getLValueBase(), nullptr, BaseType);
4074       } else if (IsConstant && Info.checkingPotentialConstantExpression() &&
4075                  BaseType->isLiteralType(Info.Ctx) && !VD->hasDefinition()) {
4076         // This variable might end up being constexpr. Don't diagnose it yet.
4077       } else if (IsConstant) {
4078         // Keep evaluating to see what we can do. In particular, we support
4079         // folding of const floating-point types, in order to make static const
4080         // data members of such types (supported as an extension) more useful.
4081         if (Info.getLangOpts().CPlusPlus) {
4082           Info.CCEDiag(E, Info.getLangOpts().CPlusPlus11
4083                               ? diag::note_constexpr_ltor_non_constexpr
4084                               : diag::note_constexpr_ltor_non_integral, 1)
4085               << VD << BaseType;
4086           Info.Note(VD->getLocation(), diag::note_declared_at);
4087         } else {
4088           Info.CCEDiag(E);
4089         }
4090       } else {
4091         // Never allow reading a non-const value.
4092         if (Info.getLangOpts().CPlusPlus) {
4093           Info.FFDiag(E, Info.getLangOpts().CPlusPlus11
4094                              ? diag::note_constexpr_ltor_non_constexpr
4095                              : diag::note_constexpr_ltor_non_integral, 1)
4096               << VD << BaseType;
4097           Info.Note(VD->getLocation(), diag::note_declared_at);
4098         } else {
4099           Info.FFDiag(E);
4100         }
4101         return CompleteObject();
4102       }
4103     }
4104 
4105     if (!evaluateVarDeclInit(Info, E, VD, Frame, LVal.getLValueVersion(), BaseVal))
4106       return CompleteObject();
4107   } else if (DynamicAllocLValue DA = LVal.Base.dyn_cast<DynamicAllocLValue>()) {
4108     Optional<DynAlloc*> Alloc = Info.lookupDynamicAlloc(DA);
4109     if (!Alloc) {
4110       Info.FFDiag(E, diag::note_constexpr_access_deleted_object) << AK;
4111       return CompleteObject();
4112     }
4113     return CompleteObject(LVal.Base, &(*Alloc)->Value,
4114                           LVal.Base.getDynamicAllocType());
4115   } else {
4116     const Expr *Base = LVal.Base.dyn_cast<const Expr*>();
4117 
4118     if (!Frame) {
4119       if (const MaterializeTemporaryExpr *MTE =
4120               dyn_cast_or_null<MaterializeTemporaryExpr>(Base)) {
4121         assert(MTE->getStorageDuration() == SD_Static &&
4122                "should have a frame for a non-global materialized temporary");
4123 
4124         // C++20 [expr.const]p4: [DR2126]
4125         //   An object or reference is usable in constant expressions if it is
4126         //   - a temporary object of non-volatile const-qualified literal type
4127         //     whose lifetime is extended to that of a variable that is usable
4128         //     in constant expressions
4129         //
4130         // C++20 [expr.const]p5:
4131         //  an lvalue-to-rvalue conversion [is not allowed unless it applies to]
4132         //   - a non-volatile glvalue that refers to an object that is usable
4133         //     in constant expressions, or
4134         //   - a non-volatile glvalue of literal type that refers to a
4135         //     non-volatile object whose lifetime began within the evaluation
4136         //     of E;
4137         //
4138         // C++11 misses the 'began within the evaluation of e' check and
4139         // instead allows all temporaries, including things like:
4140         //   int &&r = 1;
4141         //   int x = ++r;
4142         //   constexpr int k = r;
4143         // Therefore we use the C++14-onwards rules in C++11 too.
4144         //
4145         // Note that temporaries whose lifetimes began while evaluating a
4146         // variable's constructor are not usable while evaluating the
4147         // corresponding destructor, not even if they're of const-qualified
4148         // types.
4149         if (!MTE->isUsableInConstantExpressions(Info.Ctx) &&
4150             !lifetimeStartedInEvaluation(Info, LVal.Base)) {
4151           if (!IsAccess)
4152             return CompleteObject(LVal.getLValueBase(), nullptr, BaseType);
4153           Info.FFDiag(E, diag::note_constexpr_access_static_temporary, 1) << AK;
4154           Info.Note(MTE->getExprLoc(), diag::note_constexpr_temporary_here);
4155           return CompleteObject();
4156         }
4157 
4158         BaseVal = MTE->getOrCreateValue(false);
4159         assert(BaseVal && "got reference to unevaluated temporary");
4160       } else {
4161         if (!IsAccess)
4162           return CompleteObject(LVal.getLValueBase(), nullptr, BaseType);
4163         APValue Val;
4164         LVal.moveInto(Val);
4165         Info.FFDiag(E, diag::note_constexpr_access_unreadable_object)
4166             << AK
4167             << Val.getAsString(Info.Ctx,
4168                                Info.Ctx.getLValueReferenceType(LValType));
4169         NoteLValueLocation(Info, LVal.Base);
4170         return CompleteObject();
4171       }
4172     } else {
4173       BaseVal = Frame->getTemporary(Base, LVal.Base.getVersion());
4174       assert(BaseVal && "missing value for temporary");
4175     }
4176   }
4177 
4178   // In C++14, we can't safely access any mutable state when we might be
4179   // evaluating after an unmodeled side effect. Parameters are modeled as state
4180   // in the caller, but aren't visible once the call returns, so they can be
4181   // modified in a speculatively-evaluated call.
4182   //
4183   // FIXME: Not all local state is mutable. Allow local constant subobjects
4184   // to be read here (but take care with 'mutable' fields).
4185   unsigned VisibleDepth = Depth;
4186   if (llvm::isa_and_nonnull<ParmVarDecl>(
4187           LVal.Base.dyn_cast<const ValueDecl *>()))
4188     ++VisibleDepth;
4189   if ((Frame && Info.getLangOpts().CPlusPlus14 &&
4190        Info.EvalStatus.HasSideEffects) ||
4191       (isModification(AK) && VisibleDepth < Info.SpeculativeEvaluationDepth))
4192     return CompleteObject();
4193 
4194   return CompleteObject(LVal.getLValueBase(), BaseVal, BaseType);
4195 }
4196 
4197 /// Perform an lvalue-to-rvalue conversion on the given glvalue. This
4198 /// can also be used for 'lvalue-to-lvalue' conversions for looking up the
4199 /// glvalue referred to by an entity of reference type.
4200 ///
4201 /// \param Info - Information about the ongoing evaluation.
4202 /// \param Conv - The expression for which we are performing the conversion.
4203 ///               Used for diagnostics.
4204 /// \param Type - The type of the glvalue (before stripping cv-qualifiers in the
4205 ///               case of a non-class type).
4206 /// \param LVal - The glvalue on which we are attempting to perform this action.
4207 /// \param RVal - The produced value will be placed here.
4208 /// \param WantObjectRepresentation - If true, we're looking for the object
4209 ///               representation rather than the value, and in particular,
4210 ///               there is no requirement that the result be fully initialized.
4211 static bool
handleLValueToRValueConversion(EvalInfo & Info,const Expr * Conv,QualType Type,const LValue & LVal,APValue & RVal,bool WantObjectRepresentation=false)4212 handleLValueToRValueConversion(EvalInfo &Info, const Expr *Conv, QualType Type,
4213                                const LValue &LVal, APValue &RVal,
4214                                bool WantObjectRepresentation = false) {
4215   if (LVal.Designator.Invalid)
4216     return false;
4217 
4218   // Check for special cases where there is no existing APValue to look at.
4219   const Expr *Base = LVal.Base.dyn_cast<const Expr*>();
4220 
4221   AccessKinds AK =
4222       WantObjectRepresentation ? AK_ReadObjectRepresentation : AK_Read;
4223 
4224   if (Base && !LVal.getLValueCallIndex() && !Type.isVolatileQualified()) {
4225     if (const CompoundLiteralExpr *CLE = dyn_cast<CompoundLiteralExpr>(Base)) {
4226       // In C99, a CompoundLiteralExpr is an lvalue, and we defer evaluating the
4227       // initializer until now for such expressions. Such an expression can't be
4228       // an ICE in C, so this only matters for fold.
4229       if (Type.isVolatileQualified()) {
4230         Info.FFDiag(Conv);
4231         return false;
4232       }
4233       APValue Lit;
4234       if (!Evaluate(Lit, Info, CLE->getInitializer()))
4235         return false;
4236       CompleteObject LitObj(LVal.Base, &Lit, Base->getType());
4237       return extractSubobject(Info, Conv, LitObj, LVal.Designator, RVal, AK);
4238     } else if (isa<StringLiteral>(Base) || isa<PredefinedExpr>(Base)) {
4239       // Special-case character extraction so we don't have to construct an
4240       // APValue for the whole string.
4241       assert(LVal.Designator.Entries.size() <= 1 &&
4242              "Can only read characters from string literals");
4243       if (LVal.Designator.Entries.empty()) {
4244         // Fail for now for LValue to RValue conversion of an array.
4245         // (This shouldn't show up in C/C++, but it could be triggered by a
4246         // weird EvaluateAsRValue call from a tool.)
4247         Info.FFDiag(Conv);
4248         return false;
4249       }
4250       if (LVal.Designator.isOnePastTheEnd()) {
4251         if (Info.getLangOpts().CPlusPlus11)
4252           Info.FFDiag(Conv, diag::note_constexpr_access_past_end) << AK;
4253         else
4254           Info.FFDiag(Conv);
4255         return false;
4256       }
4257       uint64_t CharIndex = LVal.Designator.Entries[0].getAsArrayIndex();
4258       RVal = APValue(extractStringLiteralCharacter(Info, Base, CharIndex));
4259       return true;
4260     }
4261   }
4262 
4263   CompleteObject Obj = findCompleteObject(Info, Conv, AK, LVal, Type);
4264   return Obj && extractSubobject(Info, Conv, Obj, LVal.Designator, RVal, AK);
4265 }
4266 
4267 /// Perform an assignment of Val to LVal. Takes ownership of Val.
handleAssignment(EvalInfo & Info,const Expr * E,const LValue & LVal,QualType LValType,APValue & Val)4268 static bool handleAssignment(EvalInfo &Info, const Expr *E, const LValue &LVal,
4269                              QualType LValType, APValue &Val) {
4270   if (LVal.Designator.Invalid)
4271     return false;
4272 
4273   if (!Info.getLangOpts().CPlusPlus14) {
4274     Info.FFDiag(E);
4275     return false;
4276   }
4277 
4278   CompleteObject Obj = findCompleteObject(Info, E, AK_Assign, LVal, LValType);
4279   return Obj && modifySubobject(Info, E, Obj, LVal.Designator, Val);
4280 }
4281 
4282 namespace {
4283 struct CompoundAssignSubobjectHandler {
4284   EvalInfo &Info;
4285   const CompoundAssignOperator *E;
4286   QualType PromotedLHSType;
4287   BinaryOperatorKind Opcode;
4288   const APValue &RHS;
4289 
4290   static const AccessKinds AccessKind = AK_Assign;
4291 
4292   typedef bool result_type;
4293 
checkConst__anonb66d72d20c11::CompoundAssignSubobjectHandler4294   bool checkConst(QualType QT) {
4295     // Assigning to a const object has undefined behavior.
4296     if (QT.isConstQualified()) {
4297       Info.FFDiag(E, diag::note_constexpr_modify_const_type) << QT;
4298       return false;
4299     }
4300     return true;
4301   }
4302 
failed__anonb66d72d20c11::CompoundAssignSubobjectHandler4303   bool failed() { return false; }
found__anonb66d72d20c11::CompoundAssignSubobjectHandler4304   bool found(APValue &Subobj, QualType SubobjType) {
4305     switch (Subobj.getKind()) {
4306     case APValue::Int:
4307       return found(Subobj.getInt(), SubobjType);
4308     case APValue::Float:
4309       return found(Subobj.getFloat(), SubobjType);
4310     case APValue::ComplexInt:
4311     case APValue::ComplexFloat:
4312       // FIXME: Implement complex compound assignment.
4313       Info.FFDiag(E);
4314       return false;
4315     case APValue::LValue:
4316       return foundPointer(Subobj, SubobjType);
4317     case APValue::Vector:
4318       return foundVector(Subobj, SubobjType);
4319     default:
4320       // FIXME: can this happen?
4321       Info.FFDiag(E);
4322       return false;
4323     }
4324   }
4325 
foundVector__anonb66d72d20c11::CompoundAssignSubobjectHandler4326   bool foundVector(APValue &Value, QualType SubobjType) {
4327     if (!checkConst(SubobjType))
4328       return false;
4329 
4330     if (!SubobjType->isVectorType()) {
4331       Info.FFDiag(E);
4332       return false;
4333     }
4334     return handleVectorVectorBinOp(Info, E, Opcode, Value, RHS);
4335   }
4336 
found__anonb66d72d20c11::CompoundAssignSubobjectHandler4337   bool found(APSInt &Value, QualType SubobjType) {
4338     if (!checkConst(SubobjType))
4339       return false;
4340 
4341     if (!SubobjType->isIntegerType()) {
4342       // We don't support compound assignment on integer-cast-to-pointer
4343       // values.
4344       Info.FFDiag(E);
4345       return false;
4346     }
4347 
4348     if (RHS.isInt()) {
4349       APSInt LHS =
4350           HandleIntToIntCast(Info, E, PromotedLHSType, SubobjType, Value);
4351       if (!handleIntIntBinOp(Info, E, LHS, Opcode, RHS.getInt(), LHS))
4352         return false;
4353       Value = HandleIntToIntCast(Info, E, SubobjType, PromotedLHSType, LHS);
4354       return true;
4355     } else if (RHS.isFloat()) {
4356       const FPOptions FPO = E->getFPFeaturesInEffect(
4357                                     Info.Ctx.getLangOpts());
4358       APFloat FValue(0.0);
4359       return HandleIntToFloatCast(Info, E, FPO, SubobjType, Value,
4360                                   PromotedLHSType, FValue) &&
4361              handleFloatFloatBinOp(Info, E, FValue, Opcode, RHS.getFloat()) &&
4362              HandleFloatToIntCast(Info, E, PromotedLHSType, FValue, SubobjType,
4363                                   Value);
4364     }
4365 
4366     Info.FFDiag(E);
4367     return false;
4368   }
found__anonb66d72d20c11::CompoundAssignSubobjectHandler4369   bool found(APFloat &Value, QualType SubobjType) {
4370     return checkConst(SubobjType) &&
4371            HandleFloatToFloatCast(Info, E, SubobjType, PromotedLHSType,
4372                                   Value) &&
4373            handleFloatFloatBinOp(Info, E, Value, Opcode, RHS.getFloat()) &&
4374            HandleFloatToFloatCast(Info, E, PromotedLHSType, SubobjType, Value);
4375   }
foundPointer__anonb66d72d20c11::CompoundAssignSubobjectHandler4376   bool foundPointer(APValue &Subobj, QualType SubobjType) {
4377     if (!checkConst(SubobjType))
4378       return false;
4379 
4380     QualType PointeeType;
4381     if (const PointerType *PT = SubobjType->getAs<PointerType>())
4382       PointeeType = PT->getPointeeType();
4383 
4384     if (PointeeType.isNull() || !RHS.isInt() ||
4385         (Opcode != BO_Add && Opcode != BO_Sub)) {
4386       Info.FFDiag(E);
4387       return false;
4388     }
4389 
4390     APSInt Offset = RHS.getInt();
4391     if (Opcode == BO_Sub)
4392       negateAsSigned(Offset);
4393 
4394     LValue LVal;
4395     LVal.setFrom(Info.Ctx, Subobj);
4396     if (!HandleLValueArrayAdjustment(Info, E, LVal, PointeeType, Offset))
4397       return false;
4398     LVal.moveInto(Subobj);
4399     return true;
4400   }
4401 };
4402 } // end anonymous namespace
4403 
4404 const AccessKinds CompoundAssignSubobjectHandler::AccessKind;
4405 
4406 /// Perform a compound assignment of LVal <op>= RVal.
handleCompoundAssignment(EvalInfo & Info,const CompoundAssignOperator * E,const LValue & LVal,QualType LValType,QualType PromotedLValType,BinaryOperatorKind Opcode,const APValue & RVal)4407 static bool handleCompoundAssignment(EvalInfo &Info,
4408                                      const CompoundAssignOperator *E,
4409                                      const LValue &LVal, QualType LValType,
4410                                      QualType PromotedLValType,
4411                                      BinaryOperatorKind Opcode,
4412                                      const APValue &RVal) {
4413   if (LVal.Designator.Invalid)
4414     return false;
4415 
4416   if (!Info.getLangOpts().CPlusPlus14) {
4417     Info.FFDiag(E);
4418     return false;
4419   }
4420 
4421   CompleteObject Obj = findCompleteObject(Info, E, AK_Assign, LVal, LValType);
4422   CompoundAssignSubobjectHandler Handler = { Info, E, PromotedLValType, Opcode,
4423                                              RVal };
4424   return Obj && findSubobject(Info, E, Obj, LVal.Designator, Handler);
4425 }
4426 
4427 namespace {
4428 struct IncDecSubobjectHandler {
4429   EvalInfo &Info;
4430   const UnaryOperator *E;
4431   AccessKinds AccessKind;
4432   APValue *Old;
4433 
4434   typedef bool result_type;
4435 
checkConst__anonb66d72d20d11::IncDecSubobjectHandler4436   bool checkConst(QualType QT) {
4437     // Assigning to a const object has undefined behavior.
4438     if (QT.isConstQualified()) {
4439       Info.FFDiag(E, diag::note_constexpr_modify_const_type) << QT;
4440       return false;
4441     }
4442     return true;
4443   }
4444 
failed__anonb66d72d20d11::IncDecSubobjectHandler4445   bool failed() { return false; }
found__anonb66d72d20d11::IncDecSubobjectHandler4446   bool found(APValue &Subobj, QualType SubobjType) {
4447     // Stash the old value. Also clear Old, so we don't clobber it later
4448     // if we're post-incrementing a complex.
4449     if (Old) {
4450       *Old = Subobj;
4451       Old = nullptr;
4452     }
4453 
4454     switch (Subobj.getKind()) {
4455     case APValue::Int:
4456       return found(Subobj.getInt(), SubobjType);
4457     case APValue::Float:
4458       return found(Subobj.getFloat(), SubobjType);
4459     case APValue::ComplexInt:
4460       return found(Subobj.getComplexIntReal(),
4461                    SubobjType->castAs<ComplexType>()->getElementType()
4462                      .withCVRQualifiers(SubobjType.getCVRQualifiers()));
4463     case APValue::ComplexFloat:
4464       return found(Subobj.getComplexFloatReal(),
4465                    SubobjType->castAs<ComplexType>()->getElementType()
4466                      .withCVRQualifiers(SubobjType.getCVRQualifiers()));
4467     case APValue::LValue:
4468       return foundPointer(Subobj, SubobjType);
4469     default:
4470       // FIXME: can this happen?
4471       Info.FFDiag(E);
4472       return false;
4473     }
4474   }
found__anonb66d72d20d11::IncDecSubobjectHandler4475   bool found(APSInt &Value, QualType SubobjType) {
4476     if (!checkConst(SubobjType))
4477       return false;
4478 
4479     if (!SubobjType->isIntegerType()) {
4480       // We don't support increment / decrement on integer-cast-to-pointer
4481       // values.
4482       Info.FFDiag(E);
4483       return false;
4484     }
4485 
4486     if (Old) *Old = APValue(Value);
4487 
4488     // bool arithmetic promotes to int, and the conversion back to bool
4489     // doesn't reduce mod 2^n, so special-case it.
4490     if (SubobjType->isBooleanType()) {
4491       if (AccessKind == AK_Increment)
4492         Value = 1;
4493       else
4494         Value = !Value;
4495       return true;
4496     }
4497 
4498     bool WasNegative = Value.isNegative();
4499     if (AccessKind == AK_Increment) {
4500       ++Value;
4501 
4502       if (!WasNegative && Value.isNegative() && E->canOverflow()) {
4503         APSInt ActualValue(Value, /*IsUnsigned*/true);
4504         return HandleOverflow(Info, E, ActualValue, SubobjType);
4505       }
4506     } else {
4507       --Value;
4508 
4509       if (WasNegative && !Value.isNegative() && E->canOverflow()) {
4510         unsigned BitWidth = Value.getBitWidth();
4511         APSInt ActualValue(Value.sext(BitWidth + 1), /*IsUnsigned*/false);
4512         ActualValue.setBit(BitWidth);
4513         return HandleOverflow(Info, E, ActualValue, SubobjType);
4514       }
4515     }
4516     return true;
4517   }
found__anonb66d72d20d11::IncDecSubobjectHandler4518   bool found(APFloat &Value, QualType SubobjType) {
4519     if (!checkConst(SubobjType))
4520       return false;
4521 
4522     if (Old) *Old = APValue(Value);
4523 
4524     APFloat One(Value.getSemantics(), 1);
4525     if (AccessKind == AK_Increment)
4526       Value.add(One, APFloat::rmNearestTiesToEven);
4527     else
4528       Value.subtract(One, APFloat::rmNearestTiesToEven);
4529     return true;
4530   }
foundPointer__anonb66d72d20d11::IncDecSubobjectHandler4531   bool foundPointer(APValue &Subobj, QualType SubobjType) {
4532     if (!checkConst(SubobjType))
4533       return false;
4534 
4535     QualType PointeeType;
4536     if (const PointerType *PT = SubobjType->getAs<PointerType>())
4537       PointeeType = PT->getPointeeType();
4538     else {
4539       Info.FFDiag(E);
4540       return false;
4541     }
4542 
4543     LValue LVal;
4544     LVal.setFrom(Info.Ctx, Subobj);
4545     if (!HandleLValueArrayAdjustment(Info, E, LVal, PointeeType,
4546                                      AccessKind == AK_Increment ? 1 : -1))
4547       return false;
4548     LVal.moveInto(Subobj);
4549     return true;
4550   }
4551 };
4552 } // end anonymous namespace
4553 
4554 /// Perform an increment or decrement on LVal.
handleIncDec(EvalInfo & Info,const Expr * E,const LValue & LVal,QualType LValType,bool IsIncrement,APValue * Old)4555 static bool handleIncDec(EvalInfo &Info, const Expr *E, const LValue &LVal,
4556                          QualType LValType, bool IsIncrement, APValue *Old) {
4557   if (LVal.Designator.Invalid)
4558     return false;
4559 
4560   if (!Info.getLangOpts().CPlusPlus14) {
4561     Info.FFDiag(E);
4562     return false;
4563   }
4564 
4565   AccessKinds AK = IsIncrement ? AK_Increment : AK_Decrement;
4566   CompleteObject Obj = findCompleteObject(Info, E, AK, LVal, LValType);
4567   IncDecSubobjectHandler Handler = {Info, cast<UnaryOperator>(E), AK, Old};
4568   return Obj && findSubobject(Info, E, Obj, LVal.Designator, Handler);
4569 }
4570 
4571 /// Build an lvalue for the object argument of a member function call.
EvaluateObjectArgument(EvalInfo & Info,const Expr * Object,LValue & This)4572 static bool EvaluateObjectArgument(EvalInfo &Info, const Expr *Object,
4573                                    LValue &This) {
4574   if (Object->getType()->isPointerType() && Object->isPRValue())
4575     return EvaluatePointer(Object, This, Info);
4576 
4577   if (Object->isGLValue())
4578     return EvaluateLValue(Object, This, Info);
4579 
4580   if (Object->getType()->isLiteralType(Info.Ctx))
4581     return EvaluateTemporary(Object, This, Info);
4582 
4583   Info.FFDiag(Object, diag::note_constexpr_nonliteral) << Object->getType();
4584   return false;
4585 }
4586 
4587 /// HandleMemberPointerAccess - Evaluate a member access operation and build an
4588 /// lvalue referring to the result.
4589 ///
4590 /// \param Info - Information about the ongoing evaluation.
4591 /// \param LV - An lvalue referring to the base of the member pointer.
4592 /// \param RHS - The member pointer expression.
4593 /// \param IncludeMember - Specifies whether the member itself is included in
4594 ///        the resulting LValue subobject designator. This is not possible when
4595 ///        creating a bound member function.
4596 /// \return The field or method declaration to which the member pointer refers,
4597 ///         or 0 if evaluation fails.
HandleMemberPointerAccess(EvalInfo & Info,QualType LVType,LValue & LV,const Expr * RHS,bool IncludeMember=true)4598 static const ValueDecl *HandleMemberPointerAccess(EvalInfo &Info,
4599                                                   QualType LVType,
4600                                                   LValue &LV,
4601                                                   const Expr *RHS,
4602                                                   bool IncludeMember = true) {
4603   MemberPtr MemPtr;
4604   if (!EvaluateMemberPointer(RHS, MemPtr, Info))
4605     return nullptr;
4606 
4607   // C++11 [expr.mptr.oper]p6: If the second operand is the null pointer to
4608   // member value, the behavior is undefined.
4609   if (!MemPtr.getDecl()) {
4610     // FIXME: Specific diagnostic.
4611     Info.FFDiag(RHS);
4612     return nullptr;
4613   }
4614 
4615   if (MemPtr.isDerivedMember()) {
4616     // This is a member of some derived class. Truncate LV appropriately.
4617     // The end of the derived-to-base path for the base object must match the
4618     // derived-to-base path for the member pointer.
4619     if (LV.Designator.MostDerivedPathLength + MemPtr.Path.size() >
4620         LV.Designator.Entries.size()) {
4621       Info.FFDiag(RHS);
4622       return nullptr;
4623     }
4624     unsigned PathLengthToMember =
4625         LV.Designator.Entries.size() - MemPtr.Path.size();
4626     for (unsigned I = 0, N = MemPtr.Path.size(); I != N; ++I) {
4627       const CXXRecordDecl *LVDecl = getAsBaseClass(
4628           LV.Designator.Entries[PathLengthToMember + I]);
4629       const CXXRecordDecl *MPDecl = MemPtr.Path[I];
4630       if (LVDecl->getCanonicalDecl() != MPDecl->getCanonicalDecl()) {
4631         Info.FFDiag(RHS);
4632         return nullptr;
4633       }
4634     }
4635 
4636     // Truncate the lvalue to the appropriate derived class.
4637     if (!CastToDerivedClass(Info, RHS, LV, MemPtr.getContainingRecord(),
4638                             PathLengthToMember))
4639       return nullptr;
4640   } else if (!MemPtr.Path.empty()) {
4641     // Extend the LValue path with the member pointer's path.
4642     LV.Designator.Entries.reserve(LV.Designator.Entries.size() +
4643                                   MemPtr.Path.size() + IncludeMember);
4644 
4645     // Walk down to the appropriate base class.
4646     if (const PointerType *PT = LVType->getAs<PointerType>())
4647       LVType = PT->getPointeeType();
4648     const CXXRecordDecl *RD = LVType->getAsCXXRecordDecl();
4649     assert(RD && "member pointer access on non-class-type expression");
4650     // The first class in the path is that of the lvalue.
4651     for (unsigned I = 1, N = MemPtr.Path.size(); I != N; ++I) {
4652       const CXXRecordDecl *Base = MemPtr.Path[N - I - 1];
4653       if (!HandleLValueDirectBase(Info, RHS, LV, RD, Base))
4654         return nullptr;
4655       RD = Base;
4656     }
4657     // Finally cast to the class containing the member.
4658     if (!HandleLValueDirectBase(Info, RHS, LV, RD,
4659                                 MemPtr.getContainingRecord()))
4660       return nullptr;
4661   }
4662 
4663   // Add the member. Note that we cannot build bound member functions here.
4664   if (IncludeMember) {
4665     if (const FieldDecl *FD = dyn_cast<FieldDecl>(MemPtr.getDecl())) {
4666       if (!HandleLValueMember(Info, RHS, LV, FD))
4667         return nullptr;
4668     } else if (const IndirectFieldDecl *IFD =
4669                  dyn_cast<IndirectFieldDecl>(MemPtr.getDecl())) {
4670       if (!HandleLValueIndirectMember(Info, RHS, LV, IFD))
4671         return nullptr;
4672     } else {
4673       llvm_unreachable("can't construct reference to bound member function");
4674     }
4675   }
4676 
4677   return MemPtr.getDecl();
4678 }
4679 
HandleMemberPointerAccess(EvalInfo & Info,const BinaryOperator * BO,LValue & LV,bool IncludeMember=true)4680 static const ValueDecl *HandleMemberPointerAccess(EvalInfo &Info,
4681                                                   const BinaryOperator *BO,
4682                                                   LValue &LV,
4683                                                   bool IncludeMember = true) {
4684   assert(BO->getOpcode() == BO_PtrMemD || BO->getOpcode() == BO_PtrMemI);
4685 
4686   if (!EvaluateObjectArgument(Info, BO->getLHS(), LV)) {
4687     if (Info.noteFailure()) {
4688       MemberPtr MemPtr;
4689       EvaluateMemberPointer(BO->getRHS(), MemPtr, Info);
4690     }
4691     return nullptr;
4692   }
4693 
4694   return HandleMemberPointerAccess(Info, BO->getLHS()->getType(), LV,
4695                                    BO->getRHS(), IncludeMember);
4696 }
4697 
4698 /// HandleBaseToDerivedCast - Apply the given base-to-derived cast operation on
4699 /// the provided lvalue, which currently refers to the base object.
HandleBaseToDerivedCast(EvalInfo & Info,const CastExpr * E,LValue & Result)4700 static bool HandleBaseToDerivedCast(EvalInfo &Info, const CastExpr *E,
4701                                     LValue &Result) {
4702   SubobjectDesignator &D = Result.Designator;
4703   if (D.Invalid || !Result.checkNullPointer(Info, E, CSK_Derived))
4704     return false;
4705 
4706   QualType TargetQT = E->getType();
4707   if (const PointerType *PT = TargetQT->getAs<PointerType>())
4708     TargetQT = PT->getPointeeType();
4709 
4710   // Check this cast lands within the final derived-to-base subobject path.
4711   if (D.MostDerivedPathLength + E->path_size() > D.Entries.size()) {
4712     Info.CCEDiag(E, diag::note_constexpr_invalid_downcast)
4713       << D.MostDerivedType << TargetQT;
4714     return false;
4715   }
4716 
4717   // Check the type of the final cast. We don't need to check the path,
4718   // since a cast can only be formed if the path is unique.
4719   unsigned NewEntriesSize = D.Entries.size() - E->path_size();
4720   const CXXRecordDecl *TargetType = TargetQT->getAsCXXRecordDecl();
4721   const CXXRecordDecl *FinalType;
4722   if (NewEntriesSize == D.MostDerivedPathLength)
4723     FinalType = D.MostDerivedType->getAsCXXRecordDecl();
4724   else
4725     FinalType = getAsBaseClass(D.Entries[NewEntriesSize - 1]);
4726   if (FinalType->getCanonicalDecl() != TargetType->getCanonicalDecl()) {
4727     Info.CCEDiag(E, diag::note_constexpr_invalid_downcast)
4728       << D.MostDerivedType << TargetQT;
4729     return false;
4730   }
4731 
4732   // Truncate the lvalue to the appropriate derived class.
4733   return CastToDerivedClass(Info, E, Result, TargetType, NewEntriesSize);
4734 }
4735 
4736 /// Get the value to use for a default-initialized object of type T.
4737 /// Return false if it encounters something invalid.
getDefaultInitValue(QualType T,APValue & Result)4738 static bool getDefaultInitValue(QualType T, APValue &Result) {
4739   bool Success = true;
4740   if (auto *RD = T->getAsCXXRecordDecl()) {
4741     if (RD->isInvalidDecl()) {
4742       Result = APValue();
4743       return false;
4744     }
4745     if (RD->isUnion()) {
4746       Result = APValue((const FieldDecl *)nullptr);
4747       return true;
4748     }
4749     Result = APValue(APValue::UninitStruct(), RD->getNumBases(),
4750                      std::distance(RD->field_begin(), RD->field_end()));
4751 
4752     unsigned Index = 0;
4753     for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
4754                                                   End = RD->bases_end();
4755          I != End; ++I, ++Index)
4756       Success &= getDefaultInitValue(I->getType(), Result.getStructBase(Index));
4757 
4758     for (const auto *I : RD->fields()) {
4759       if (I->isUnnamedBitfield())
4760         continue;
4761       Success &= getDefaultInitValue(I->getType(),
4762                                      Result.getStructField(I->getFieldIndex()));
4763     }
4764     return Success;
4765   }
4766 
4767   if (auto *AT =
4768           dyn_cast_or_null<ConstantArrayType>(T->getAsArrayTypeUnsafe())) {
4769     Result = APValue(APValue::UninitArray(), 0, AT->getSize().getZExtValue());
4770     if (Result.hasArrayFiller())
4771       Success &=
4772           getDefaultInitValue(AT->getElementType(), Result.getArrayFiller());
4773 
4774     return Success;
4775   }
4776 
4777   Result = APValue::IndeterminateValue();
4778   return true;
4779 }
4780 
4781 namespace {
4782 enum EvalStmtResult {
4783   /// Evaluation failed.
4784   ESR_Failed,
4785   /// Hit a 'return' statement.
4786   ESR_Returned,
4787   /// Evaluation succeeded.
4788   ESR_Succeeded,
4789   /// Hit a 'continue' statement.
4790   ESR_Continue,
4791   /// Hit a 'break' statement.
4792   ESR_Break,
4793   /// Still scanning for 'case' or 'default' statement.
4794   ESR_CaseNotFound
4795 };
4796 }
4797 
EvaluateVarDecl(EvalInfo & Info,const VarDecl * VD)4798 static bool EvaluateVarDecl(EvalInfo &Info, const VarDecl *VD) {
4799   // We don't need to evaluate the initializer for a static local.
4800   if (!VD->hasLocalStorage())
4801     return true;
4802 
4803   LValue Result;
4804   APValue &Val = Info.CurrentCall->createTemporary(VD, VD->getType(),
4805                                                    ScopeKind::Block, Result);
4806 
4807   const Expr *InitE = VD->getInit();
4808   if (!InitE) {
4809     if (VD->getType()->isDependentType())
4810       return Info.noteSideEffect();
4811     return getDefaultInitValue(VD->getType(), Val);
4812   }
4813   if (InitE->isValueDependent())
4814     return false;
4815 
4816   if (!EvaluateInPlace(Val, Info, Result, InitE)) {
4817     // Wipe out any partially-computed value, to allow tracking that this
4818     // evaluation failed.
4819     Val = APValue();
4820     return false;
4821   }
4822 
4823   return true;
4824 }
4825 
EvaluateDecl(EvalInfo & Info,const Decl * D)4826 static bool EvaluateDecl(EvalInfo &Info, const Decl *D) {
4827   bool OK = true;
4828 
4829   if (const VarDecl *VD = dyn_cast<VarDecl>(D))
4830     OK &= EvaluateVarDecl(Info, VD);
4831 
4832   if (const DecompositionDecl *DD = dyn_cast<DecompositionDecl>(D))
4833     for (auto *BD : DD->bindings())
4834       if (auto *VD = BD->getHoldingVar())
4835         OK &= EvaluateDecl(Info, VD);
4836 
4837   return OK;
4838 }
4839 
EvaluateDependentExpr(const Expr * E,EvalInfo & Info)4840 static bool EvaluateDependentExpr(const Expr *E, EvalInfo &Info) {
4841   assert(E->isValueDependent());
4842   if (Info.noteSideEffect())
4843     return true;
4844   assert(E->containsErrors() && "valid value-dependent expression should never "
4845                                 "reach invalid code path.");
4846   return false;
4847 }
4848 
4849 /// Evaluate a condition (either a variable declaration or an expression).
EvaluateCond(EvalInfo & Info,const VarDecl * CondDecl,const Expr * Cond,bool & Result)4850 static bool EvaluateCond(EvalInfo &Info, const VarDecl *CondDecl,
4851                          const Expr *Cond, bool &Result) {
4852   if (Cond->isValueDependent())
4853     return false;
4854   FullExpressionRAII Scope(Info);
4855   if (CondDecl && !EvaluateDecl(Info, CondDecl))
4856     return false;
4857   if (!EvaluateAsBooleanCondition(Cond, Result, Info))
4858     return false;
4859   return Scope.destroy();
4860 }
4861 
4862 namespace {
4863 /// A location where the result (returned value) of evaluating a
4864 /// statement should be stored.
4865 struct StmtResult {
4866   /// The APValue that should be filled in with the returned value.
4867   APValue &Value;
4868   /// The location containing the result, if any (used to support RVO).
4869   const LValue *Slot;
4870 };
4871 
4872 struct TempVersionRAII {
4873   CallStackFrame &Frame;
4874 
TempVersionRAII__anonb66d72d20f11::TempVersionRAII4875   TempVersionRAII(CallStackFrame &Frame) : Frame(Frame) {
4876     Frame.pushTempVersion();
4877   }
4878 
~TempVersionRAII__anonb66d72d20f11::TempVersionRAII4879   ~TempVersionRAII() {
4880     Frame.popTempVersion();
4881   }
4882 };
4883 
4884 }
4885 
4886 static EvalStmtResult EvaluateStmt(StmtResult &Result, EvalInfo &Info,
4887                                    const Stmt *S,
4888                                    const SwitchCase *SC = nullptr);
4889 
4890 /// Evaluate the body of a loop, and translate the result as appropriate.
EvaluateLoopBody(StmtResult & Result,EvalInfo & Info,const Stmt * Body,const SwitchCase * Case=nullptr)4891 static EvalStmtResult EvaluateLoopBody(StmtResult &Result, EvalInfo &Info,
4892                                        const Stmt *Body,
4893                                        const SwitchCase *Case = nullptr) {
4894   BlockScopeRAII Scope(Info);
4895 
4896   EvalStmtResult ESR = EvaluateStmt(Result, Info, Body, Case);
4897   if (ESR != ESR_Failed && ESR != ESR_CaseNotFound && !Scope.destroy())
4898     ESR = ESR_Failed;
4899 
4900   switch (ESR) {
4901   case ESR_Break:
4902     return ESR_Succeeded;
4903   case ESR_Succeeded:
4904   case ESR_Continue:
4905     return ESR_Continue;
4906   case ESR_Failed:
4907   case ESR_Returned:
4908   case ESR_CaseNotFound:
4909     return ESR;
4910   }
4911   llvm_unreachable("Invalid EvalStmtResult!");
4912 }
4913 
4914 /// Evaluate a switch statement.
EvaluateSwitch(StmtResult & Result,EvalInfo & Info,const SwitchStmt * SS)4915 static EvalStmtResult EvaluateSwitch(StmtResult &Result, EvalInfo &Info,
4916                                      const SwitchStmt *SS) {
4917   BlockScopeRAII Scope(Info);
4918 
4919   // Evaluate the switch condition.
4920   APSInt Value;
4921   {
4922     if (const Stmt *Init = SS->getInit()) {
4923       EvalStmtResult ESR = EvaluateStmt(Result, Info, Init);
4924       if (ESR != ESR_Succeeded) {
4925         if (ESR != ESR_Failed && !Scope.destroy())
4926           ESR = ESR_Failed;
4927         return ESR;
4928       }
4929     }
4930 
4931     FullExpressionRAII CondScope(Info);
4932     if (SS->getConditionVariable() &&
4933         !EvaluateDecl(Info, SS->getConditionVariable()))
4934       return ESR_Failed;
4935     if (!EvaluateInteger(SS->getCond(), Value, Info))
4936       return ESR_Failed;
4937     if (!CondScope.destroy())
4938       return ESR_Failed;
4939   }
4940 
4941   // Find the switch case corresponding to the value of the condition.
4942   // FIXME: Cache this lookup.
4943   const SwitchCase *Found = nullptr;
4944   for (const SwitchCase *SC = SS->getSwitchCaseList(); SC;
4945        SC = SC->getNextSwitchCase()) {
4946     if (isa<DefaultStmt>(SC)) {
4947       Found = SC;
4948       continue;
4949     }
4950 
4951     const CaseStmt *CS = cast<CaseStmt>(SC);
4952     APSInt LHS = CS->getLHS()->EvaluateKnownConstInt(Info.Ctx);
4953     APSInt RHS = CS->getRHS() ? CS->getRHS()->EvaluateKnownConstInt(Info.Ctx)
4954                               : LHS;
4955     if (LHS <= Value && Value <= RHS) {
4956       Found = SC;
4957       break;
4958     }
4959   }
4960 
4961   if (!Found)
4962     return Scope.destroy() ? ESR_Succeeded : ESR_Failed;
4963 
4964   // Search the switch body for the switch case and evaluate it from there.
4965   EvalStmtResult ESR = EvaluateStmt(Result, Info, SS->getBody(), Found);
4966   if (ESR != ESR_Failed && ESR != ESR_CaseNotFound && !Scope.destroy())
4967     return ESR_Failed;
4968 
4969   switch (ESR) {
4970   case ESR_Break:
4971     return ESR_Succeeded;
4972   case ESR_Succeeded:
4973   case ESR_Continue:
4974   case ESR_Failed:
4975   case ESR_Returned:
4976     return ESR;
4977   case ESR_CaseNotFound:
4978     // This can only happen if the switch case is nested within a statement
4979     // expression. We have no intention of supporting that.
4980     Info.FFDiag(Found->getBeginLoc(),
4981                 diag::note_constexpr_stmt_expr_unsupported);
4982     return ESR_Failed;
4983   }
4984   llvm_unreachable("Invalid EvalStmtResult!");
4985 }
4986 
4987 // Evaluate a statement.
EvaluateStmt(StmtResult & Result,EvalInfo & Info,const Stmt * S,const SwitchCase * Case)4988 static EvalStmtResult EvaluateStmt(StmtResult &Result, EvalInfo &Info,
4989                                    const Stmt *S, const SwitchCase *Case) {
4990   if (!Info.nextStep(S))
4991     return ESR_Failed;
4992 
4993   // If we're hunting down a 'case' or 'default' label, recurse through
4994   // substatements until we hit the label.
4995   if (Case) {
4996     switch (S->getStmtClass()) {
4997     case Stmt::CompoundStmtClass:
4998       // FIXME: Precompute which substatement of a compound statement we
4999       // would jump to, and go straight there rather than performing a
5000       // linear scan each time.
5001     case Stmt::LabelStmtClass:
5002     case Stmt::AttributedStmtClass:
5003     case Stmt::DoStmtClass:
5004       break;
5005 
5006     case Stmt::CaseStmtClass:
5007     case Stmt::DefaultStmtClass:
5008       if (Case == S)
5009         Case = nullptr;
5010       break;
5011 
5012     case Stmt::IfStmtClass: {
5013       // FIXME: Precompute which side of an 'if' we would jump to, and go
5014       // straight there rather than scanning both sides.
5015       const IfStmt *IS = cast<IfStmt>(S);
5016 
5017       // Wrap the evaluation in a block scope, in case it's a DeclStmt
5018       // preceded by our switch label.
5019       BlockScopeRAII Scope(Info);
5020 
5021       // Step into the init statement in case it brings an (uninitialized)
5022       // variable into scope.
5023       if (const Stmt *Init = IS->getInit()) {
5024         EvalStmtResult ESR = EvaluateStmt(Result, Info, Init, Case);
5025         if (ESR != ESR_CaseNotFound) {
5026           assert(ESR != ESR_Succeeded);
5027           return ESR;
5028         }
5029       }
5030 
5031       // Condition variable must be initialized if it exists.
5032       // FIXME: We can skip evaluating the body if there's a condition
5033       // variable, as there can't be any case labels within it.
5034       // (The same is true for 'for' statements.)
5035 
5036       EvalStmtResult ESR = EvaluateStmt(Result, Info, IS->getThen(), Case);
5037       if (ESR == ESR_Failed)
5038         return ESR;
5039       if (ESR != ESR_CaseNotFound)
5040         return Scope.destroy() ? ESR : ESR_Failed;
5041       if (!IS->getElse())
5042         return ESR_CaseNotFound;
5043 
5044       ESR = EvaluateStmt(Result, Info, IS->getElse(), Case);
5045       if (ESR == ESR_Failed)
5046         return ESR;
5047       if (ESR != ESR_CaseNotFound)
5048         return Scope.destroy() ? ESR : ESR_Failed;
5049       return ESR_CaseNotFound;
5050     }
5051 
5052     case Stmt::WhileStmtClass: {
5053       EvalStmtResult ESR =
5054           EvaluateLoopBody(Result, Info, cast<WhileStmt>(S)->getBody(), Case);
5055       if (ESR != ESR_Continue)
5056         return ESR;
5057       break;
5058     }
5059 
5060     case Stmt::ForStmtClass: {
5061       const ForStmt *FS = cast<ForStmt>(S);
5062       BlockScopeRAII Scope(Info);
5063 
5064       // Step into the init statement in case it brings an (uninitialized)
5065       // variable into scope.
5066       if (const Stmt *Init = FS->getInit()) {
5067         EvalStmtResult ESR = EvaluateStmt(Result, Info, Init, Case);
5068         if (ESR != ESR_CaseNotFound) {
5069           assert(ESR != ESR_Succeeded);
5070           return ESR;
5071         }
5072       }
5073 
5074       EvalStmtResult ESR =
5075           EvaluateLoopBody(Result, Info, FS->getBody(), Case);
5076       if (ESR != ESR_Continue)
5077         return ESR;
5078       if (const auto *Inc = FS->getInc()) {
5079         if (Inc->isValueDependent()) {
5080           if (!EvaluateDependentExpr(Inc, Info))
5081             return ESR_Failed;
5082         } else {
5083           FullExpressionRAII IncScope(Info);
5084           if (!EvaluateIgnoredValue(Info, Inc) || !IncScope.destroy())
5085             return ESR_Failed;
5086         }
5087       }
5088       break;
5089     }
5090 
5091     case Stmt::DeclStmtClass: {
5092       // Start the lifetime of any uninitialized variables we encounter. They
5093       // might be used by the selected branch of the switch.
5094       const DeclStmt *DS = cast<DeclStmt>(S);
5095       for (const auto *D : DS->decls()) {
5096         if (const auto *VD = dyn_cast<VarDecl>(D)) {
5097           if (VD->hasLocalStorage() && !VD->getInit())
5098             if (!EvaluateVarDecl(Info, VD))
5099               return ESR_Failed;
5100           // FIXME: If the variable has initialization that can't be jumped
5101           // over, bail out of any immediately-surrounding compound-statement
5102           // too. There can't be any case labels here.
5103         }
5104       }
5105       return ESR_CaseNotFound;
5106     }
5107 
5108     default:
5109       return ESR_CaseNotFound;
5110     }
5111   }
5112 
5113   switch (S->getStmtClass()) {
5114   default:
5115     if (const Expr *E = dyn_cast<Expr>(S)) {
5116       if (E->isValueDependent()) {
5117         if (!EvaluateDependentExpr(E, Info))
5118           return ESR_Failed;
5119       } else {
5120         // Don't bother evaluating beyond an expression-statement which couldn't
5121         // be evaluated.
5122         // FIXME: Do we need the FullExpressionRAII object here?
5123         // VisitExprWithCleanups should create one when necessary.
5124         FullExpressionRAII Scope(Info);
5125         if (!EvaluateIgnoredValue(Info, E) || !Scope.destroy())
5126           return ESR_Failed;
5127       }
5128       return ESR_Succeeded;
5129     }
5130 
5131     Info.FFDiag(S->getBeginLoc());
5132     return ESR_Failed;
5133 
5134   case Stmt::NullStmtClass:
5135     return ESR_Succeeded;
5136 
5137   case Stmt::DeclStmtClass: {
5138     const DeclStmt *DS = cast<DeclStmt>(S);
5139     for (const auto *D : DS->decls()) {
5140       // Each declaration initialization is its own full-expression.
5141       FullExpressionRAII Scope(Info);
5142       if (!EvaluateDecl(Info, D) && !Info.noteFailure())
5143         return ESR_Failed;
5144       if (!Scope.destroy())
5145         return ESR_Failed;
5146     }
5147     return ESR_Succeeded;
5148   }
5149 
5150   case Stmt::ReturnStmtClass: {
5151     const Expr *RetExpr = cast<ReturnStmt>(S)->getRetValue();
5152     FullExpressionRAII Scope(Info);
5153     if (RetExpr && RetExpr->isValueDependent()) {
5154       EvaluateDependentExpr(RetExpr, Info);
5155       // We know we returned, but we don't know what the value is.
5156       return ESR_Failed;
5157     }
5158     if (RetExpr &&
5159         !(Result.Slot
5160               ? EvaluateInPlace(Result.Value, Info, *Result.Slot, RetExpr)
5161               : Evaluate(Result.Value, Info, RetExpr)))
5162       return ESR_Failed;
5163     return Scope.destroy() ? ESR_Returned : ESR_Failed;
5164   }
5165 
5166   case Stmt::CompoundStmtClass: {
5167     BlockScopeRAII Scope(Info);
5168 
5169     const CompoundStmt *CS = cast<CompoundStmt>(S);
5170     for (const auto *BI : CS->body()) {
5171       EvalStmtResult ESR = EvaluateStmt(Result, Info, BI, Case);
5172       if (ESR == ESR_Succeeded)
5173         Case = nullptr;
5174       else if (ESR != ESR_CaseNotFound) {
5175         if (ESR != ESR_Failed && !Scope.destroy())
5176           return ESR_Failed;
5177         return ESR;
5178       }
5179     }
5180     if (Case)
5181       return ESR_CaseNotFound;
5182     return Scope.destroy() ? ESR_Succeeded : ESR_Failed;
5183   }
5184 
5185   case Stmt::IfStmtClass: {
5186     const IfStmt *IS = cast<IfStmt>(S);
5187 
5188     // Evaluate the condition, as either a var decl or as an expression.
5189     BlockScopeRAII Scope(Info);
5190     if (const Stmt *Init = IS->getInit()) {
5191       EvalStmtResult ESR = EvaluateStmt(Result, Info, Init);
5192       if (ESR != ESR_Succeeded) {
5193         if (ESR != ESR_Failed && !Scope.destroy())
5194           return ESR_Failed;
5195         return ESR;
5196       }
5197     }
5198     bool Cond;
5199     if (!EvaluateCond(Info, IS->getConditionVariable(), IS->getCond(), Cond))
5200       return ESR_Failed;
5201 
5202     if (const Stmt *SubStmt = Cond ? IS->getThen() : IS->getElse()) {
5203       EvalStmtResult ESR = EvaluateStmt(Result, Info, SubStmt);
5204       if (ESR != ESR_Succeeded) {
5205         if (ESR != ESR_Failed && !Scope.destroy())
5206           return ESR_Failed;
5207         return ESR;
5208       }
5209     }
5210     return Scope.destroy() ? ESR_Succeeded : ESR_Failed;
5211   }
5212 
5213   case Stmt::WhileStmtClass: {
5214     const WhileStmt *WS = cast<WhileStmt>(S);
5215     while (true) {
5216       BlockScopeRAII Scope(Info);
5217       bool Continue;
5218       if (!EvaluateCond(Info, WS->getConditionVariable(), WS->getCond(),
5219                         Continue))
5220         return ESR_Failed;
5221       if (!Continue)
5222         break;
5223 
5224       EvalStmtResult ESR = EvaluateLoopBody(Result, Info, WS->getBody());
5225       if (ESR != ESR_Continue) {
5226         if (ESR != ESR_Failed && !Scope.destroy())
5227           return ESR_Failed;
5228         return ESR;
5229       }
5230       if (!Scope.destroy())
5231         return ESR_Failed;
5232     }
5233     return ESR_Succeeded;
5234   }
5235 
5236   case Stmt::DoStmtClass: {
5237     const DoStmt *DS = cast<DoStmt>(S);
5238     bool Continue;
5239     do {
5240       EvalStmtResult ESR = EvaluateLoopBody(Result, Info, DS->getBody(), Case);
5241       if (ESR != ESR_Continue)
5242         return ESR;
5243       Case = nullptr;
5244 
5245       if (DS->getCond()->isValueDependent()) {
5246         EvaluateDependentExpr(DS->getCond(), Info);
5247         // Bailout as we don't know whether to keep going or terminate the loop.
5248         return ESR_Failed;
5249       }
5250       FullExpressionRAII CondScope(Info);
5251       if (!EvaluateAsBooleanCondition(DS->getCond(), Continue, Info) ||
5252           !CondScope.destroy())
5253         return ESR_Failed;
5254     } while (Continue);
5255     return ESR_Succeeded;
5256   }
5257 
5258   case Stmt::ForStmtClass: {
5259     const ForStmt *FS = cast<ForStmt>(S);
5260     BlockScopeRAII ForScope(Info);
5261     if (FS->getInit()) {
5262       EvalStmtResult ESR = EvaluateStmt(Result, Info, FS->getInit());
5263       if (ESR != ESR_Succeeded) {
5264         if (ESR != ESR_Failed && !ForScope.destroy())
5265           return ESR_Failed;
5266         return ESR;
5267       }
5268     }
5269     while (true) {
5270       BlockScopeRAII IterScope(Info);
5271       bool Continue = true;
5272       if (FS->getCond() && !EvaluateCond(Info, FS->getConditionVariable(),
5273                                          FS->getCond(), Continue))
5274         return ESR_Failed;
5275       if (!Continue)
5276         break;
5277 
5278       EvalStmtResult ESR = EvaluateLoopBody(Result, Info, FS->getBody());
5279       if (ESR != ESR_Continue) {
5280         if (ESR != ESR_Failed && (!IterScope.destroy() || !ForScope.destroy()))
5281           return ESR_Failed;
5282         return ESR;
5283       }
5284 
5285       if (const auto *Inc = FS->getInc()) {
5286         if (Inc->isValueDependent()) {
5287           if (!EvaluateDependentExpr(Inc, Info))
5288             return ESR_Failed;
5289         } else {
5290           FullExpressionRAII IncScope(Info);
5291           if (!EvaluateIgnoredValue(Info, Inc) || !IncScope.destroy())
5292             return ESR_Failed;
5293         }
5294       }
5295 
5296       if (!IterScope.destroy())
5297         return ESR_Failed;
5298     }
5299     return ForScope.destroy() ? ESR_Succeeded : ESR_Failed;
5300   }
5301 
5302   case Stmt::CXXForRangeStmtClass: {
5303     const CXXForRangeStmt *FS = cast<CXXForRangeStmt>(S);
5304     BlockScopeRAII Scope(Info);
5305 
5306     // Evaluate the init-statement if present.
5307     if (FS->getInit()) {
5308       EvalStmtResult ESR = EvaluateStmt(Result, Info, FS->getInit());
5309       if (ESR != ESR_Succeeded) {
5310         if (ESR != ESR_Failed && !Scope.destroy())
5311           return ESR_Failed;
5312         return ESR;
5313       }
5314     }
5315 
5316     // Initialize the __range variable.
5317     EvalStmtResult ESR = EvaluateStmt(Result, Info, FS->getRangeStmt());
5318     if (ESR != ESR_Succeeded) {
5319       if (ESR != ESR_Failed && !Scope.destroy())
5320         return ESR_Failed;
5321       return ESR;
5322     }
5323 
5324     // Create the __begin and __end iterators.
5325     ESR = EvaluateStmt(Result, Info, FS->getBeginStmt());
5326     if (ESR != ESR_Succeeded) {
5327       if (ESR != ESR_Failed && !Scope.destroy())
5328         return ESR_Failed;
5329       return ESR;
5330     }
5331     ESR = EvaluateStmt(Result, Info, FS->getEndStmt());
5332     if (ESR != ESR_Succeeded) {
5333       if (ESR != ESR_Failed && !Scope.destroy())
5334         return ESR_Failed;
5335       return ESR;
5336     }
5337 
5338     while (true) {
5339       // Condition: __begin != __end.
5340       {
5341         if (FS->getCond()->isValueDependent()) {
5342           EvaluateDependentExpr(FS->getCond(), Info);
5343           // We don't know whether to keep going or terminate the loop.
5344           return ESR_Failed;
5345         }
5346         bool Continue = true;
5347         FullExpressionRAII CondExpr(Info);
5348         if (!EvaluateAsBooleanCondition(FS->getCond(), Continue, Info))
5349           return ESR_Failed;
5350         if (!Continue)
5351           break;
5352       }
5353 
5354       // User's variable declaration, initialized by *__begin.
5355       BlockScopeRAII InnerScope(Info);
5356       ESR = EvaluateStmt(Result, Info, FS->getLoopVarStmt());
5357       if (ESR != ESR_Succeeded) {
5358         if (ESR != ESR_Failed && (!InnerScope.destroy() || !Scope.destroy()))
5359           return ESR_Failed;
5360         return ESR;
5361       }
5362 
5363       // Loop body.
5364       ESR = EvaluateLoopBody(Result, Info, FS->getBody());
5365       if (ESR != ESR_Continue) {
5366         if (ESR != ESR_Failed && (!InnerScope.destroy() || !Scope.destroy()))
5367           return ESR_Failed;
5368         return ESR;
5369       }
5370       if (FS->getInc()->isValueDependent()) {
5371         if (!EvaluateDependentExpr(FS->getInc(), Info))
5372           return ESR_Failed;
5373       } else {
5374         // Increment: ++__begin
5375         if (!EvaluateIgnoredValue(Info, FS->getInc()))
5376           return ESR_Failed;
5377       }
5378 
5379       if (!InnerScope.destroy())
5380         return ESR_Failed;
5381     }
5382 
5383     return Scope.destroy() ? ESR_Succeeded : ESR_Failed;
5384   }
5385 
5386   case Stmt::SwitchStmtClass:
5387     return EvaluateSwitch(Result, Info, cast<SwitchStmt>(S));
5388 
5389   case Stmt::ContinueStmtClass:
5390     return ESR_Continue;
5391 
5392   case Stmt::BreakStmtClass:
5393     return ESR_Break;
5394 
5395   case Stmt::LabelStmtClass:
5396     return EvaluateStmt(Result, Info, cast<LabelStmt>(S)->getSubStmt(), Case);
5397 
5398   case Stmt::AttributedStmtClass:
5399     // As a general principle, C++11 attributes can be ignored without
5400     // any semantic impact.
5401     return EvaluateStmt(Result, Info, cast<AttributedStmt>(S)->getSubStmt(),
5402                         Case);
5403 
5404   case Stmt::CaseStmtClass:
5405   case Stmt::DefaultStmtClass:
5406     return EvaluateStmt(Result, Info, cast<SwitchCase>(S)->getSubStmt(), Case);
5407   case Stmt::CXXTryStmtClass:
5408     // Evaluate try blocks by evaluating all sub statements.
5409     return EvaluateStmt(Result, Info, cast<CXXTryStmt>(S)->getTryBlock(), Case);
5410   }
5411 }
5412 
5413 /// CheckTrivialDefaultConstructor - Check whether a constructor is a trivial
5414 /// default constructor. If so, we'll fold it whether or not it's marked as
5415 /// constexpr. If it is marked as constexpr, we will never implicitly define it,
5416 /// so we need special handling.
CheckTrivialDefaultConstructor(EvalInfo & Info,SourceLocation Loc,const CXXConstructorDecl * CD,bool IsValueInitialization)5417 static bool CheckTrivialDefaultConstructor(EvalInfo &Info, SourceLocation Loc,
5418                                            const CXXConstructorDecl *CD,
5419                                            bool IsValueInitialization) {
5420   if (!CD->isTrivial() || !CD->isDefaultConstructor())
5421     return false;
5422 
5423   // Value-initialization does not call a trivial default constructor, so such a
5424   // call is a core constant expression whether or not the constructor is
5425   // constexpr.
5426   if (!CD->isConstexpr() && !IsValueInitialization) {
5427     if (Info.getLangOpts().CPlusPlus11) {
5428       // FIXME: If DiagDecl is an implicitly-declared special member function,
5429       // we should be much more explicit about why it's not constexpr.
5430       Info.CCEDiag(Loc, diag::note_constexpr_invalid_function, 1)
5431         << /*IsConstexpr*/0 << /*IsConstructor*/1 << CD;
5432       Info.Note(CD->getLocation(), diag::note_declared_at);
5433     } else {
5434       Info.CCEDiag(Loc, diag::note_invalid_subexpr_in_const_expr);
5435     }
5436   }
5437   return true;
5438 }
5439 
5440 /// CheckConstexprFunction - Check that a function can be called in a constant
5441 /// expression.
CheckConstexprFunction(EvalInfo & Info,SourceLocation CallLoc,const FunctionDecl * Declaration,const FunctionDecl * Definition,const Stmt * Body)5442 static bool CheckConstexprFunction(EvalInfo &Info, SourceLocation CallLoc,
5443                                    const FunctionDecl *Declaration,
5444                                    const FunctionDecl *Definition,
5445                                    const Stmt *Body) {
5446   // Potential constant expressions can contain calls to declared, but not yet
5447   // defined, constexpr functions.
5448   if (Info.checkingPotentialConstantExpression() && !Definition &&
5449       Declaration->isConstexpr())
5450     return false;
5451 
5452   // Bail out if the function declaration itself is invalid.  We will
5453   // have produced a relevant diagnostic while parsing it, so just
5454   // note the problematic sub-expression.
5455   if (Declaration->isInvalidDecl()) {
5456     Info.FFDiag(CallLoc, diag::note_invalid_subexpr_in_const_expr);
5457     return false;
5458   }
5459 
5460   // DR1872: An instantiated virtual constexpr function can't be called in a
5461   // constant expression (prior to C++20). We can still constant-fold such a
5462   // call.
5463   if (!Info.Ctx.getLangOpts().CPlusPlus20 && isa<CXXMethodDecl>(Declaration) &&
5464       cast<CXXMethodDecl>(Declaration)->isVirtual())
5465     Info.CCEDiag(CallLoc, diag::note_constexpr_virtual_call);
5466 
5467   if (Definition && Definition->isInvalidDecl()) {
5468     Info.FFDiag(CallLoc, diag::note_invalid_subexpr_in_const_expr);
5469     return false;
5470   }
5471 
5472   // Can we evaluate this function call?
5473   if (Definition && Definition->isConstexpr() && Body)
5474     return true;
5475 
5476   if (Info.getLangOpts().CPlusPlus11) {
5477     const FunctionDecl *DiagDecl = Definition ? Definition : Declaration;
5478 
5479     // If this function is not constexpr because it is an inherited
5480     // non-constexpr constructor, diagnose that directly.
5481     auto *CD = dyn_cast<CXXConstructorDecl>(DiagDecl);
5482     if (CD && CD->isInheritingConstructor()) {
5483       auto *Inherited = CD->getInheritedConstructor().getConstructor();
5484       if (!Inherited->isConstexpr())
5485         DiagDecl = CD = Inherited;
5486     }
5487 
5488     // FIXME: If DiagDecl is an implicitly-declared special member function
5489     // or an inheriting constructor, we should be much more explicit about why
5490     // it's not constexpr.
5491     if (CD && CD->isInheritingConstructor())
5492       Info.FFDiag(CallLoc, diag::note_constexpr_invalid_inhctor, 1)
5493         << CD->getInheritedConstructor().getConstructor()->getParent();
5494     else
5495       Info.FFDiag(CallLoc, diag::note_constexpr_invalid_function, 1)
5496         << DiagDecl->isConstexpr() << (bool)CD << DiagDecl;
5497     Info.Note(DiagDecl->getLocation(), diag::note_declared_at);
5498   } else {
5499     Info.FFDiag(CallLoc, diag::note_invalid_subexpr_in_const_expr);
5500   }
5501   return false;
5502 }
5503 
5504 namespace {
5505 struct CheckDynamicTypeHandler {
5506   AccessKinds AccessKind;
5507   typedef bool result_type;
failed__anonb66d72d21011::CheckDynamicTypeHandler5508   bool failed() { return false; }
found__anonb66d72d21011::CheckDynamicTypeHandler5509   bool found(APValue &Subobj, QualType SubobjType) { return true; }
found__anonb66d72d21011::CheckDynamicTypeHandler5510   bool found(APSInt &Value, QualType SubobjType) { return true; }
found__anonb66d72d21011::CheckDynamicTypeHandler5511   bool found(APFloat &Value, QualType SubobjType) { return true; }
5512 };
5513 } // end anonymous namespace
5514 
5515 /// Check that we can access the notional vptr of an object / determine its
5516 /// dynamic type.
checkDynamicType(EvalInfo & Info,const Expr * E,const LValue & This,AccessKinds AK,bool Polymorphic)5517 static bool checkDynamicType(EvalInfo &Info, const Expr *E, const LValue &This,
5518                              AccessKinds AK, bool Polymorphic) {
5519   if (This.Designator.Invalid)
5520     return false;
5521 
5522   CompleteObject Obj = findCompleteObject(Info, E, AK, This, QualType());
5523 
5524   if (!Obj)
5525     return false;
5526 
5527   if (!Obj.Value) {
5528     // The object is not usable in constant expressions, so we can't inspect
5529     // its value to see if it's in-lifetime or what the active union members
5530     // are. We can still check for a one-past-the-end lvalue.
5531     if (This.Designator.isOnePastTheEnd() ||
5532         This.Designator.isMostDerivedAnUnsizedArray()) {
5533       Info.FFDiag(E, This.Designator.isOnePastTheEnd()
5534                          ? diag::note_constexpr_access_past_end
5535                          : diag::note_constexpr_access_unsized_array)
5536           << AK;
5537       return false;
5538     } else if (Polymorphic) {
5539       // Conservatively refuse to perform a polymorphic operation if we would
5540       // not be able to read a notional 'vptr' value.
5541       APValue Val;
5542       This.moveInto(Val);
5543       QualType StarThisType =
5544           Info.Ctx.getLValueReferenceType(This.Designator.getType(Info.Ctx));
5545       Info.FFDiag(E, diag::note_constexpr_polymorphic_unknown_dynamic_type)
5546           << AK << Val.getAsString(Info.Ctx, StarThisType);
5547       return false;
5548     }
5549     return true;
5550   }
5551 
5552   CheckDynamicTypeHandler Handler{AK};
5553   return Obj && findSubobject(Info, E, Obj, This.Designator, Handler);
5554 }
5555 
5556 /// Check that the pointee of the 'this' pointer in a member function call is
5557 /// either within its lifetime or in its period of construction or destruction.
5558 static bool
checkNonVirtualMemberCallThisPointer(EvalInfo & Info,const Expr * E,const LValue & This,const CXXMethodDecl * NamedMember)5559 checkNonVirtualMemberCallThisPointer(EvalInfo &Info, const Expr *E,
5560                                      const LValue &This,
5561                                      const CXXMethodDecl *NamedMember) {
5562   return checkDynamicType(
5563       Info, E, This,
5564       isa<CXXDestructorDecl>(NamedMember) ? AK_Destroy : AK_MemberCall, false);
5565 }
5566 
5567 struct DynamicType {
5568   /// The dynamic class type of the object.
5569   const CXXRecordDecl *Type;
5570   /// The corresponding path length in the lvalue.
5571   unsigned PathLength;
5572 };
5573 
getBaseClassType(SubobjectDesignator & Designator,unsigned PathLength)5574 static const CXXRecordDecl *getBaseClassType(SubobjectDesignator &Designator,
5575                                              unsigned PathLength) {
5576   assert(PathLength >= Designator.MostDerivedPathLength && PathLength <=
5577       Designator.Entries.size() && "invalid path length");
5578   return (PathLength == Designator.MostDerivedPathLength)
5579              ? Designator.MostDerivedType->getAsCXXRecordDecl()
5580              : getAsBaseClass(Designator.Entries[PathLength - 1]);
5581 }
5582 
5583 /// Determine the dynamic type of an object.
ComputeDynamicType(EvalInfo & Info,const Expr * E,LValue & This,AccessKinds AK)5584 static Optional<DynamicType> ComputeDynamicType(EvalInfo &Info, const Expr *E,
5585                                                 LValue &This, AccessKinds AK) {
5586   // If we don't have an lvalue denoting an object of class type, there is no
5587   // meaningful dynamic type. (We consider objects of non-class type to have no
5588   // dynamic type.)
5589   if (!checkDynamicType(Info, E, This, AK, true))
5590     return None;
5591 
5592   // Refuse to compute a dynamic type in the presence of virtual bases. This
5593   // shouldn't happen other than in constant-folding situations, since literal
5594   // types can't have virtual bases.
5595   //
5596   // Note that consumers of DynamicType assume that the type has no virtual
5597   // bases, and will need modifications if this restriction is relaxed.
5598   const CXXRecordDecl *Class =
5599       This.Designator.MostDerivedType->getAsCXXRecordDecl();
5600   if (!Class || Class->getNumVBases()) {
5601     Info.FFDiag(E);
5602     return None;
5603   }
5604 
5605   // FIXME: For very deep class hierarchies, it might be beneficial to use a
5606   // binary search here instead. But the overwhelmingly common case is that
5607   // we're not in the middle of a constructor, so it probably doesn't matter
5608   // in practice.
5609   ArrayRef<APValue::LValuePathEntry> Path = This.Designator.Entries;
5610   for (unsigned PathLength = This.Designator.MostDerivedPathLength;
5611        PathLength <= Path.size(); ++PathLength) {
5612     switch (Info.isEvaluatingCtorDtor(This.getLValueBase(),
5613                                       Path.slice(0, PathLength))) {
5614     case ConstructionPhase::Bases:
5615     case ConstructionPhase::DestroyingBases:
5616       // We're constructing or destroying a base class. This is not the dynamic
5617       // type.
5618       break;
5619 
5620     case ConstructionPhase::None:
5621     case ConstructionPhase::AfterBases:
5622     case ConstructionPhase::AfterFields:
5623     case ConstructionPhase::Destroying:
5624       // We've finished constructing the base classes and not yet started
5625       // destroying them again, so this is the dynamic type.
5626       return DynamicType{getBaseClassType(This.Designator, PathLength),
5627                          PathLength};
5628     }
5629   }
5630 
5631   // CWG issue 1517: we're constructing a base class of the object described by
5632   // 'This', so that object has not yet begun its period of construction and
5633   // any polymorphic operation on it results in undefined behavior.
5634   Info.FFDiag(E);
5635   return None;
5636 }
5637 
5638 /// Perform virtual dispatch.
HandleVirtualDispatch(EvalInfo & Info,const Expr * E,LValue & This,const CXXMethodDecl * Found,llvm::SmallVectorImpl<QualType> & CovariantAdjustmentPath)5639 static const CXXMethodDecl *HandleVirtualDispatch(
5640     EvalInfo &Info, const Expr *E, LValue &This, const CXXMethodDecl *Found,
5641     llvm::SmallVectorImpl<QualType> &CovariantAdjustmentPath) {
5642   Optional<DynamicType> DynType = ComputeDynamicType(
5643       Info, E, This,
5644       isa<CXXDestructorDecl>(Found) ? AK_Destroy : AK_MemberCall);
5645   if (!DynType)
5646     return nullptr;
5647 
5648   // Find the final overrider. It must be declared in one of the classes on the
5649   // path from the dynamic type to the static type.
5650   // FIXME: If we ever allow literal types to have virtual base classes, that
5651   // won't be true.
5652   const CXXMethodDecl *Callee = Found;
5653   unsigned PathLength = DynType->PathLength;
5654   for (/**/; PathLength <= This.Designator.Entries.size(); ++PathLength) {
5655     const CXXRecordDecl *Class = getBaseClassType(This.Designator, PathLength);
5656     const CXXMethodDecl *Overrider =
5657         Found->getCorrespondingMethodDeclaredInClass(Class, false);
5658     if (Overrider) {
5659       Callee = Overrider;
5660       break;
5661     }
5662   }
5663 
5664   // C++2a [class.abstract]p6:
5665   //   the effect of making a virtual call to a pure virtual function [...] is
5666   //   undefined
5667   if (Callee->isPure()) {
5668     Info.FFDiag(E, diag::note_constexpr_pure_virtual_call, 1) << Callee;
5669     Info.Note(Callee->getLocation(), diag::note_declared_at);
5670     return nullptr;
5671   }
5672 
5673   // If necessary, walk the rest of the path to determine the sequence of
5674   // covariant adjustment steps to apply.
5675   if (!Info.Ctx.hasSameUnqualifiedType(Callee->getReturnType(),
5676                                        Found->getReturnType())) {
5677     CovariantAdjustmentPath.push_back(Callee->getReturnType());
5678     for (unsigned CovariantPathLength = PathLength + 1;
5679          CovariantPathLength != This.Designator.Entries.size();
5680          ++CovariantPathLength) {
5681       const CXXRecordDecl *NextClass =
5682           getBaseClassType(This.Designator, CovariantPathLength);
5683       const CXXMethodDecl *Next =
5684           Found->getCorrespondingMethodDeclaredInClass(NextClass, false);
5685       if (Next && !Info.Ctx.hasSameUnqualifiedType(
5686                       Next->getReturnType(), CovariantAdjustmentPath.back()))
5687         CovariantAdjustmentPath.push_back(Next->getReturnType());
5688     }
5689     if (!Info.Ctx.hasSameUnqualifiedType(Found->getReturnType(),
5690                                          CovariantAdjustmentPath.back()))
5691       CovariantAdjustmentPath.push_back(Found->getReturnType());
5692   }
5693 
5694   // Perform 'this' adjustment.
5695   if (!CastToDerivedClass(Info, E, This, Callee->getParent(), PathLength))
5696     return nullptr;
5697 
5698   return Callee;
5699 }
5700 
5701 /// Perform the adjustment from a value returned by a virtual function to
5702 /// a value of the statically expected type, which may be a pointer or
5703 /// reference to a base class of the returned type.
HandleCovariantReturnAdjustment(EvalInfo & Info,const Expr * E,APValue & Result,ArrayRef<QualType> Path)5704 static bool HandleCovariantReturnAdjustment(EvalInfo &Info, const Expr *E,
5705                                             APValue &Result,
5706                                             ArrayRef<QualType> Path) {
5707   assert(Result.isLValue() &&
5708          "unexpected kind of APValue for covariant return");
5709   if (Result.isNullPointer())
5710     return true;
5711 
5712   LValue LVal;
5713   LVal.setFrom(Info.Ctx, Result);
5714 
5715   const CXXRecordDecl *OldClass = Path[0]->getPointeeCXXRecordDecl();
5716   for (unsigned I = 1; I != Path.size(); ++I) {
5717     const CXXRecordDecl *NewClass = Path[I]->getPointeeCXXRecordDecl();
5718     assert(OldClass && NewClass && "unexpected kind of covariant return");
5719     if (OldClass != NewClass &&
5720         !CastToBaseClass(Info, E, LVal, OldClass, NewClass))
5721       return false;
5722     OldClass = NewClass;
5723   }
5724 
5725   LVal.moveInto(Result);
5726   return true;
5727 }
5728 
5729 /// Determine whether \p Base, which is known to be a direct base class of
5730 /// \p Derived, is a public base class.
isBaseClassPublic(const CXXRecordDecl * Derived,const CXXRecordDecl * Base)5731 static bool isBaseClassPublic(const CXXRecordDecl *Derived,
5732                               const CXXRecordDecl *Base) {
5733   for (const CXXBaseSpecifier &BaseSpec : Derived->bases()) {
5734     auto *BaseClass = BaseSpec.getType()->getAsCXXRecordDecl();
5735     if (BaseClass && declaresSameEntity(BaseClass, Base))
5736       return BaseSpec.getAccessSpecifier() == AS_public;
5737   }
5738   llvm_unreachable("Base is not a direct base of Derived");
5739 }
5740 
5741 /// Apply the given dynamic cast operation on the provided lvalue.
5742 ///
5743 /// This implements the hard case of dynamic_cast, requiring a "runtime check"
5744 /// to find a suitable target subobject.
HandleDynamicCast(EvalInfo & Info,const ExplicitCastExpr * E,LValue & Ptr)5745 static bool HandleDynamicCast(EvalInfo &Info, const ExplicitCastExpr *E,
5746                               LValue &Ptr) {
5747   // We can't do anything with a non-symbolic pointer value.
5748   SubobjectDesignator &D = Ptr.Designator;
5749   if (D.Invalid)
5750     return false;
5751 
5752   // C++ [expr.dynamic.cast]p6:
5753   //   If v is a null pointer value, the result is a null pointer value.
5754   if (Ptr.isNullPointer() && !E->isGLValue())
5755     return true;
5756 
5757   // For all the other cases, we need the pointer to point to an object within
5758   // its lifetime / period of construction / destruction, and we need to know
5759   // its dynamic type.
5760   Optional<DynamicType> DynType =
5761       ComputeDynamicType(Info, E, Ptr, AK_DynamicCast);
5762   if (!DynType)
5763     return false;
5764 
5765   // C++ [expr.dynamic.cast]p7:
5766   //   If T is "pointer to cv void", then the result is a pointer to the most
5767   //   derived object
5768   if (E->getType()->isVoidPointerType())
5769     return CastToDerivedClass(Info, E, Ptr, DynType->Type, DynType->PathLength);
5770 
5771   const CXXRecordDecl *C = E->getTypeAsWritten()->getPointeeCXXRecordDecl();
5772   assert(C && "dynamic_cast target is not void pointer nor class");
5773   CanQualType CQT = Info.Ctx.getCanonicalType(Info.Ctx.getRecordType(C));
5774 
5775   auto RuntimeCheckFailed = [&] (CXXBasePaths *Paths) {
5776     // C++ [expr.dynamic.cast]p9:
5777     if (!E->isGLValue()) {
5778       //   The value of a failed cast to pointer type is the null pointer value
5779       //   of the required result type.
5780       Ptr.setNull(Info.Ctx, E->getType());
5781       return true;
5782     }
5783 
5784     //   A failed cast to reference type throws [...] std::bad_cast.
5785     unsigned DiagKind;
5786     if (!Paths && (declaresSameEntity(DynType->Type, C) ||
5787                    DynType->Type->isDerivedFrom(C)))
5788       DiagKind = 0;
5789     else if (!Paths || Paths->begin() == Paths->end())
5790       DiagKind = 1;
5791     else if (Paths->isAmbiguous(CQT))
5792       DiagKind = 2;
5793     else {
5794       assert(Paths->front().Access != AS_public && "why did the cast fail?");
5795       DiagKind = 3;
5796     }
5797     Info.FFDiag(E, diag::note_constexpr_dynamic_cast_to_reference_failed)
5798         << DiagKind << Ptr.Designator.getType(Info.Ctx)
5799         << Info.Ctx.getRecordType(DynType->Type)
5800         << E->getType().getUnqualifiedType();
5801     return false;
5802   };
5803 
5804   // Runtime check, phase 1:
5805   //   Walk from the base subobject towards the derived object looking for the
5806   //   target type.
5807   for (int PathLength = Ptr.Designator.Entries.size();
5808        PathLength >= (int)DynType->PathLength; --PathLength) {
5809     const CXXRecordDecl *Class = getBaseClassType(Ptr.Designator, PathLength);
5810     if (declaresSameEntity(Class, C))
5811       return CastToDerivedClass(Info, E, Ptr, Class, PathLength);
5812     // We can only walk across public inheritance edges.
5813     if (PathLength > (int)DynType->PathLength &&
5814         !isBaseClassPublic(getBaseClassType(Ptr.Designator, PathLength - 1),
5815                            Class))
5816       return RuntimeCheckFailed(nullptr);
5817   }
5818 
5819   // Runtime check, phase 2:
5820   //   Search the dynamic type for an unambiguous public base of type C.
5821   CXXBasePaths Paths(/*FindAmbiguities=*/true,
5822                      /*RecordPaths=*/true, /*DetectVirtual=*/false);
5823   if (DynType->Type->isDerivedFrom(C, Paths) && !Paths.isAmbiguous(CQT) &&
5824       Paths.front().Access == AS_public) {
5825     // Downcast to the dynamic type...
5826     if (!CastToDerivedClass(Info, E, Ptr, DynType->Type, DynType->PathLength))
5827       return false;
5828     // ... then upcast to the chosen base class subobject.
5829     for (CXXBasePathElement &Elem : Paths.front())
5830       if (!HandleLValueBase(Info, E, Ptr, Elem.Class, Elem.Base))
5831         return false;
5832     return true;
5833   }
5834 
5835   // Otherwise, the runtime check fails.
5836   return RuntimeCheckFailed(&Paths);
5837 }
5838 
5839 namespace {
5840 struct StartLifetimeOfUnionMemberHandler {
5841   EvalInfo &Info;
5842   const Expr *LHSExpr;
5843   const FieldDecl *Field;
5844   bool DuringInit;
5845   bool Failed = false;
5846   static const AccessKinds AccessKind = AK_Assign;
5847 
5848   typedef bool result_type;
failed__anonb66d72d21211::StartLifetimeOfUnionMemberHandler5849   bool failed() { return Failed; }
found__anonb66d72d21211::StartLifetimeOfUnionMemberHandler5850   bool found(APValue &Subobj, QualType SubobjType) {
5851     // We are supposed to perform no initialization but begin the lifetime of
5852     // the object. We interpret that as meaning to do what default
5853     // initialization of the object would do if all constructors involved were
5854     // trivial:
5855     //  * All base, non-variant member, and array element subobjects' lifetimes
5856     //    begin
5857     //  * No variant members' lifetimes begin
5858     //  * All scalar subobjects whose lifetimes begin have indeterminate values
5859     assert(SubobjType->isUnionType());
5860     if (declaresSameEntity(Subobj.getUnionField(), Field)) {
5861       // This union member is already active. If it's also in-lifetime, there's
5862       // nothing to do.
5863       if (Subobj.getUnionValue().hasValue())
5864         return true;
5865     } else if (DuringInit) {
5866       // We're currently in the process of initializing a different union
5867       // member.  If we carried on, that initialization would attempt to
5868       // store to an inactive union member, resulting in undefined behavior.
5869       Info.FFDiag(LHSExpr,
5870                   diag::note_constexpr_union_member_change_during_init);
5871       return false;
5872     }
5873     APValue Result;
5874     Failed = !getDefaultInitValue(Field->getType(), Result);
5875     Subobj.setUnion(Field, Result);
5876     return true;
5877   }
found__anonb66d72d21211::StartLifetimeOfUnionMemberHandler5878   bool found(APSInt &Value, QualType SubobjType) {
5879     llvm_unreachable("wrong value kind for union object");
5880   }
found__anonb66d72d21211::StartLifetimeOfUnionMemberHandler5881   bool found(APFloat &Value, QualType SubobjType) {
5882     llvm_unreachable("wrong value kind for union object");
5883   }
5884 };
5885 } // end anonymous namespace
5886 
5887 const AccessKinds StartLifetimeOfUnionMemberHandler::AccessKind;
5888 
5889 /// Handle a builtin simple-assignment or a call to a trivial assignment
5890 /// operator whose left-hand side might involve a union member access. If it
5891 /// does, implicitly start the lifetime of any accessed union elements per
5892 /// C++20 [class.union]5.
HandleUnionActiveMemberChange(EvalInfo & Info,const Expr * LHSExpr,const LValue & LHS)5893 static bool HandleUnionActiveMemberChange(EvalInfo &Info, const Expr *LHSExpr,
5894                                           const LValue &LHS) {
5895   if (LHS.InvalidBase || LHS.Designator.Invalid)
5896     return false;
5897 
5898   llvm::SmallVector<std::pair<unsigned, const FieldDecl*>, 4> UnionPathLengths;
5899   // C++ [class.union]p5:
5900   //   define the set S(E) of subexpressions of E as follows:
5901   unsigned PathLength = LHS.Designator.Entries.size();
5902   for (const Expr *E = LHSExpr; E != nullptr;) {
5903     //   -- If E is of the form A.B, S(E) contains the elements of S(A)...
5904     if (auto *ME = dyn_cast<MemberExpr>(E)) {
5905       auto *FD = dyn_cast<FieldDecl>(ME->getMemberDecl());
5906       // Note that we can't implicitly start the lifetime of a reference,
5907       // so we don't need to proceed any further if we reach one.
5908       if (!FD || FD->getType()->isReferenceType())
5909         break;
5910 
5911       //    ... and also contains A.B if B names a union member ...
5912       if (FD->getParent()->isUnion()) {
5913         //    ... of a non-class, non-array type, or of a class type with a
5914         //    trivial default constructor that is not deleted, or an array of
5915         //    such types.
5916         auto *RD =
5917             FD->getType()->getBaseElementTypeUnsafe()->getAsCXXRecordDecl();
5918         if (!RD || RD->hasTrivialDefaultConstructor())
5919           UnionPathLengths.push_back({PathLength - 1, FD});
5920       }
5921 
5922       E = ME->getBase();
5923       --PathLength;
5924       assert(declaresSameEntity(FD,
5925                                 LHS.Designator.Entries[PathLength]
5926                                     .getAsBaseOrMember().getPointer()));
5927 
5928       //   -- If E is of the form A[B] and is interpreted as a built-in array
5929       //      subscripting operator, S(E) is [S(the array operand, if any)].
5930     } else if (auto *ASE = dyn_cast<ArraySubscriptExpr>(E)) {
5931       // Step over an ArrayToPointerDecay implicit cast.
5932       auto *Base = ASE->getBase()->IgnoreImplicit();
5933       if (!Base->getType()->isArrayType())
5934         break;
5935 
5936       E = Base;
5937       --PathLength;
5938 
5939     } else if (auto *ICE = dyn_cast<ImplicitCastExpr>(E)) {
5940       // Step over a derived-to-base conversion.
5941       E = ICE->getSubExpr();
5942       if (ICE->getCastKind() == CK_NoOp)
5943         continue;
5944       if (ICE->getCastKind() != CK_DerivedToBase &&
5945           ICE->getCastKind() != CK_UncheckedDerivedToBase)
5946         break;
5947       // Walk path backwards as we walk up from the base to the derived class.
5948       for (const CXXBaseSpecifier *Elt : llvm::reverse(ICE->path())) {
5949         --PathLength;
5950         (void)Elt;
5951         assert(declaresSameEntity(Elt->getType()->getAsCXXRecordDecl(),
5952                                   LHS.Designator.Entries[PathLength]
5953                                       .getAsBaseOrMember().getPointer()));
5954       }
5955 
5956     //   -- Otherwise, S(E) is empty.
5957     } else {
5958       break;
5959     }
5960   }
5961 
5962   // Common case: no unions' lifetimes are started.
5963   if (UnionPathLengths.empty())
5964     return true;
5965 
5966   //   if modification of X [would access an inactive union member], an object
5967   //   of the type of X is implicitly created
5968   CompleteObject Obj =
5969       findCompleteObject(Info, LHSExpr, AK_Assign, LHS, LHSExpr->getType());
5970   if (!Obj)
5971     return false;
5972   for (std::pair<unsigned, const FieldDecl *> LengthAndField :
5973            llvm::reverse(UnionPathLengths)) {
5974     // Form a designator for the union object.
5975     SubobjectDesignator D = LHS.Designator;
5976     D.truncate(Info.Ctx, LHS.Base, LengthAndField.first);
5977 
5978     bool DuringInit = Info.isEvaluatingCtorDtor(LHS.Base, D.Entries) ==
5979                       ConstructionPhase::AfterBases;
5980     StartLifetimeOfUnionMemberHandler StartLifetime{
5981         Info, LHSExpr, LengthAndField.second, DuringInit};
5982     if (!findSubobject(Info, LHSExpr, Obj, D, StartLifetime))
5983       return false;
5984   }
5985 
5986   return true;
5987 }
5988 
EvaluateCallArg(const ParmVarDecl * PVD,const Expr * Arg,CallRef Call,EvalInfo & Info,bool NonNull=false)5989 static bool EvaluateCallArg(const ParmVarDecl *PVD, const Expr *Arg,
5990                             CallRef Call, EvalInfo &Info,
5991                             bool NonNull = false) {
5992   LValue LV;
5993   // Create the parameter slot and register its destruction. For a vararg
5994   // argument, create a temporary.
5995   // FIXME: For calling conventions that destroy parameters in the callee,
5996   // should we consider performing destruction when the function returns
5997   // instead?
5998   APValue &V = PVD ? Info.CurrentCall->createParam(Call, PVD, LV)
5999                    : Info.CurrentCall->createTemporary(Arg, Arg->getType(),
6000                                                        ScopeKind::Call, LV);
6001   if (!EvaluateInPlace(V, Info, LV, Arg))
6002     return false;
6003 
6004   // Passing a null pointer to an __attribute__((nonnull)) parameter results in
6005   // undefined behavior, so is non-constant.
6006   if (NonNull && V.isLValue() && V.isNullPointer()) {
6007     Info.CCEDiag(Arg, diag::note_non_null_attribute_failed);
6008     return false;
6009   }
6010 
6011   return true;
6012 }
6013 
6014 /// Evaluate the arguments to a function call.
EvaluateArgs(ArrayRef<const Expr * > Args,CallRef Call,EvalInfo & Info,const FunctionDecl * Callee,bool RightToLeft=false)6015 static bool EvaluateArgs(ArrayRef<const Expr *> Args, CallRef Call,
6016                          EvalInfo &Info, const FunctionDecl *Callee,
6017                          bool RightToLeft = false) {
6018   bool Success = true;
6019   llvm::SmallBitVector ForbiddenNullArgs;
6020   if (Callee->hasAttr<NonNullAttr>()) {
6021     ForbiddenNullArgs.resize(Args.size());
6022     for (const auto *Attr : Callee->specific_attrs<NonNullAttr>()) {
6023       if (!Attr->args_size()) {
6024         ForbiddenNullArgs.set();
6025         break;
6026       } else
6027         for (auto Idx : Attr->args()) {
6028           unsigned ASTIdx = Idx.getASTIndex();
6029           if (ASTIdx >= Args.size())
6030             continue;
6031           ForbiddenNullArgs[ASTIdx] = 1;
6032         }
6033     }
6034   }
6035   for (unsigned I = 0; I < Args.size(); I++) {
6036     unsigned Idx = RightToLeft ? Args.size() - I - 1 : I;
6037     const ParmVarDecl *PVD =
6038         Idx < Callee->getNumParams() ? Callee->getParamDecl(Idx) : nullptr;
6039     bool NonNull = !ForbiddenNullArgs.empty() && ForbiddenNullArgs[Idx];
6040     if (!EvaluateCallArg(PVD, Args[Idx], Call, Info, NonNull)) {
6041       // If we're checking for a potential constant expression, evaluate all
6042       // initializers even if some of them fail.
6043       if (!Info.noteFailure())
6044         return false;
6045       Success = false;
6046     }
6047   }
6048   return Success;
6049 }
6050 
6051 /// Perform a trivial copy from Param, which is the parameter of a copy or move
6052 /// constructor or assignment operator.
handleTrivialCopy(EvalInfo & Info,const ParmVarDecl * Param,const Expr * E,APValue & Result,bool CopyObjectRepresentation)6053 static bool handleTrivialCopy(EvalInfo &Info, const ParmVarDecl *Param,
6054                               const Expr *E, APValue &Result,
6055                               bool CopyObjectRepresentation) {
6056   // Find the reference argument.
6057   CallStackFrame *Frame = Info.CurrentCall;
6058   APValue *RefValue = Info.getParamSlot(Frame->Arguments, Param);
6059   if (!RefValue) {
6060     Info.FFDiag(E);
6061     return false;
6062   }
6063 
6064   // Copy out the contents of the RHS object.
6065   LValue RefLValue;
6066   RefLValue.setFrom(Info.Ctx, *RefValue);
6067   return handleLValueToRValueConversion(
6068       Info, E, Param->getType().getNonReferenceType(), RefLValue, Result,
6069       CopyObjectRepresentation);
6070 }
6071 
6072 /// Evaluate a function call.
HandleFunctionCall(SourceLocation CallLoc,const FunctionDecl * Callee,const LValue * This,ArrayRef<const Expr * > Args,CallRef Call,const Stmt * Body,EvalInfo & Info,APValue & Result,const LValue * ResultSlot)6073 static bool HandleFunctionCall(SourceLocation CallLoc,
6074                                const FunctionDecl *Callee, const LValue *This,
6075                                ArrayRef<const Expr *> Args, CallRef Call,
6076                                const Stmt *Body, EvalInfo &Info,
6077                                APValue &Result, const LValue *ResultSlot) {
6078   if (!Info.CheckCallLimit(CallLoc))
6079     return false;
6080 
6081   CallStackFrame Frame(Info, CallLoc, Callee, This, Call);
6082 
6083   // For a trivial copy or move assignment, perform an APValue copy. This is
6084   // essential for unions, where the operations performed by the assignment
6085   // operator cannot be represented as statements.
6086   //
6087   // Skip this for non-union classes with no fields; in that case, the defaulted
6088   // copy/move does not actually read the object.
6089   const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(Callee);
6090   if (MD && MD->isDefaulted() &&
6091       (MD->getParent()->isUnion() ||
6092        (MD->isTrivial() &&
6093         isReadByLvalueToRvalueConversion(MD->getParent())))) {
6094     assert(This &&
6095            (MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()));
6096     APValue RHSValue;
6097     if (!handleTrivialCopy(Info, MD->getParamDecl(0), Args[0], RHSValue,
6098                            MD->getParent()->isUnion()))
6099       return false;
6100     if (!handleAssignment(Info, Args[0], *This, MD->getThisType(),
6101                           RHSValue))
6102       return false;
6103     This->moveInto(Result);
6104     return true;
6105   } else if (MD && isLambdaCallOperator(MD)) {
6106     // We're in a lambda; determine the lambda capture field maps unless we're
6107     // just constexpr checking a lambda's call operator. constexpr checking is
6108     // done before the captures have been added to the closure object (unless
6109     // we're inferring constexpr-ness), so we don't have access to them in this
6110     // case. But since we don't need the captures to constexpr check, we can
6111     // just ignore them.
6112     if (!Info.checkingPotentialConstantExpression())
6113       MD->getParent()->getCaptureFields(Frame.LambdaCaptureFields,
6114                                         Frame.LambdaThisCaptureField);
6115   }
6116 
6117   StmtResult Ret = {Result, ResultSlot};
6118   EvalStmtResult ESR = EvaluateStmt(Ret, Info, Body);
6119   if (ESR == ESR_Succeeded) {
6120     if (Callee->getReturnType()->isVoidType())
6121       return true;
6122     Info.FFDiag(Callee->getEndLoc(), diag::note_constexpr_no_return);
6123   }
6124   return ESR == ESR_Returned;
6125 }
6126 
6127 /// Evaluate a constructor call.
HandleConstructorCall(const Expr * E,const LValue & This,CallRef Call,const CXXConstructorDecl * Definition,EvalInfo & Info,APValue & Result)6128 static bool HandleConstructorCall(const Expr *E, const LValue &This,
6129                                   CallRef Call,
6130                                   const CXXConstructorDecl *Definition,
6131                                   EvalInfo &Info, APValue &Result) {
6132   SourceLocation CallLoc = E->getExprLoc();
6133   if (!Info.CheckCallLimit(CallLoc))
6134     return false;
6135 
6136   const CXXRecordDecl *RD = Definition->getParent();
6137   if (RD->getNumVBases()) {
6138     Info.FFDiag(CallLoc, diag::note_constexpr_virtual_base) << RD;
6139     return false;
6140   }
6141 
6142   EvalInfo::EvaluatingConstructorRAII EvalObj(
6143       Info,
6144       ObjectUnderConstruction{This.getLValueBase(), This.Designator.Entries},
6145       RD->getNumBases());
6146   CallStackFrame Frame(Info, CallLoc, Definition, &This, Call);
6147 
6148   // FIXME: Creating an APValue just to hold a nonexistent return value is
6149   // wasteful.
6150   APValue RetVal;
6151   StmtResult Ret = {RetVal, nullptr};
6152 
6153   // If it's a delegating constructor, delegate.
6154   if (Definition->isDelegatingConstructor()) {
6155     CXXConstructorDecl::init_const_iterator I = Definition->init_begin();
6156     if ((*I)->getInit()->isValueDependent()) {
6157       if (!EvaluateDependentExpr((*I)->getInit(), Info))
6158         return false;
6159     } else {
6160       FullExpressionRAII InitScope(Info);
6161       if (!EvaluateInPlace(Result, Info, This, (*I)->getInit()) ||
6162           !InitScope.destroy())
6163         return false;
6164     }
6165     return EvaluateStmt(Ret, Info, Definition->getBody()) != ESR_Failed;
6166   }
6167 
6168   // For a trivial copy or move constructor, perform an APValue copy. This is
6169   // essential for unions (or classes with anonymous union members), where the
6170   // operations performed by the constructor cannot be represented by
6171   // ctor-initializers.
6172   //
6173   // Skip this for empty non-union classes; we should not perform an
6174   // lvalue-to-rvalue conversion on them because their copy constructor does not
6175   // actually read them.
6176   if (Definition->isDefaulted() && Definition->isCopyOrMoveConstructor() &&
6177       (Definition->getParent()->isUnion() ||
6178        (Definition->isTrivial() &&
6179         isReadByLvalueToRvalueConversion(Definition->getParent())))) {
6180     return handleTrivialCopy(Info, Definition->getParamDecl(0), E, Result,
6181                              Definition->getParent()->isUnion());
6182   }
6183 
6184   // Reserve space for the struct members.
6185   if (!Result.hasValue()) {
6186     if (!RD->isUnion())
6187       Result = APValue(APValue::UninitStruct(), RD->getNumBases(),
6188                        std::distance(RD->field_begin(), RD->field_end()));
6189     else
6190       // A union starts with no active member.
6191       Result = APValue((const FieldDecl*)nullptr);
6192   }
6193 
6194   if (RD->isInvalidDecl()) return false;
6195   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
6196 
6197   // A scope for temporaries lifetime-extended by reference members.
6198   BlockScopeRAII LifetimeExtendedScope(Info);
6199 
6200   bool Success = true;
6201   unsigned BasesSeen = 0;
6202 #ifndef NDEBUG
6203   CXXRecordDecl::base_class_const_iterator BaseIt = RD->bases_begin();
6204 #endif
6205   CXXRecordDecl::field_iterator FieldIt = RD->field_begin();
6206   auto SkipToField = [&](FieldDecl *FD, bool Indirect) {
6207     // We might be initializing the same field again if this is an indirect
6208     // field initialization.
6209     if (FieldIt == RD->field_end() ||
6210         FieldIt->getFieldIndex() > FD->getFieldIndex()) {
6211       assert(Indirect && "fields out of order?");
6212       return;
6213     }
6214 
6215     // Default-initialize any fields with no explicit initializer.
6216     for (; !declaresSameEntity(*FieldIt, FD); ++FieldIt) {
6217       assert(FieldIt != RD->field_end() && "missing field?");
6218       if (!FieldIt->isUnnamedBitfield())
6219         Success &= getDefaultInitValue(
6220             FieldIt->getType(),
6221             Result.getStructField(FieldIt->getFieldIndex()));
6222     }
6223     ++FieldIt;
6224   };
6225   for (const auto *I : Definition->inits()) {
6226     LValue Subobject = This;
6227     LValue SubobjectParent = This;
6228     APValue *Value = &Result;
6229 
6230     // Determine the subobject to initialize.
6231     FieldDecl *FD = nullptr;
6232     if (I->isBaseInitializer()) {
6233       QualType BaseType(I->getBaseClass(), 0);
6234 #ifndef NDEBUG
6235       // Non-virtual base classes are initialized in the order in the class
6236       // definition. We have already checked for virtual base classes.
6237       assert(!BaseIt->isVirtual() && "virtual base for literal type");
6238       assert(Info.Ctx.hasSameType(BaseIt->getType(), BaseType) &&
6239              "base class initializers not in expected order");
6240       ++BaseIt;
6241 #endif
6242       if (!HandleLValueDirectBase(Info, I->getInit(), Subobject, RD,
6243                                   BaseType->getAsCXXRecordDecl(), &Layout))
6244         return false;
6245       Value = &Result.getStructBase(BasesSeen++);
6246     } else if ((FD = I->getMember())) {
6247       if (!HandleLValueMember(Info, I->getInit(), Subobject, FD, &Layout))
6248         return false;
6249       if (RD->isUnion()) {
6250         Result = APValue(FD);
6251         Value = &Result.getUnionValue();
6252       } else {
6253         SkipToField(FD, false);
6254         Value = &Result.getStructField(FD->getFieldIndex());
6255       }
6256     } else if (IndirectFieldDecl *IFD = I->getIndirectMember()) {
6257       // Walk the indirect field decl's chain to find the object to initialize,
6258       // and make sure we've initialized every step along it.
6259       auto IndirectFieldChain = IFD->chain();
6260       for (auto *C : IndirectFieldChain) {
6261         FD = cast<FieldDecl>(C);
6262         CXXRecordDecl *CD = cast<CXXRecordDecl>(FD->getParent());
6263         // Switch the union field if it differs. This happens if we had
6264         // preceding zero-initialization, and we're now initializing a union
6265         // subobject other than the first.
6266         // FIXME: In this case, the values of the other subobjects are
6267         // specified, since zero-initialization sets all padding bits to zero.
6268         if (!Value->hasValue() ||
6269             (Value->isUnion() && Value->getUnionField() != FD)) {
6270           if (CD->isUnion())
6271             *Value = APValue(FD);
6272           else
6273             // FIXME: This immediately starts the lifetime of all members of
6274             // an anonymous struct. It would be preferable to strictly start
6275             // member lifetime in initialization order.
6276             Success &= getDefaultInitValue(Info.Ctx.getRecordType(CD), *Value);
6277         }
6278         // Store Subobject as its parent before updating it for the last element
6279         // in the chain.
6280         if (C == IndirectFieldChain.back())
6281           SubobjectParent = Subobject;
6282         if (!HandleLValueMember(Info, I->getInit(), Subobject, FD))
6283           return false;
6284         if (CD->isUnion())
6285           Value = &Value->getUnionValue();
6286         else {
6287           if (C == IndirectFieldChain.front() && !RD->isUnion())
6288             SkipToField(FD, true);
6289           Value = &Value->getStructField(FD->getFieldIndex());
6290         }
6291       }
6292     } else {
6293       llvm_unreachable("unknown base initializer kind");
6294     }
6295 
6296     // Need to override This for implicit field initializers as in this case
6297     // This refers to innermost anonymous struct/union containing initializer,
6298     // not to currently constructed class.
6299     const Expr *Init = I->getInit();
6300     if (Init->isValueDependent()) {
6301       if (!EvaluateDependentExpr(Init, Info))
6302         return false;
6303     } else {
6304       ThisOverrideRAII ThisOverride(*Info.CurrentCall, &SubobjectParent,
6305                                     isa<CXXDefaultInitExpr>(Init));
6306       FullExpressionRAII InitScope(Info);
6307       if (!EvaluateInPlace(*Value, Info, Subobject, Init) ||
6308           (FD && FD->isBitField() &&
6309            !truncateBitfieldValue(Info, Init, *Value, FD))) {
6310         // If we're checking for a potential constant expression, evaluate all
6311         // initializers even if some of them fail.
6312         if (!Info.noteFailure())
6313           return false;
6314         Success = false;
6315       }
6316     }
6317 
6318     // This is the point at which the dynamic type of the object becomes this
6319     // class type.
6320     if (I->isBaseInitializer() && BasesSeen == RD->getNumBases())
6321       EvalObj.finishedConstructingBases();
6322   }
6323 
6324   // Default-initialize any remaining fields.
6325   if (!RD->isUnion()) {
6326     for (; FieldIt != RD->field_end(); ++FieldIt) {
6327       if (!FieldIt->isUnnamedBitfield())
6328         Success &= getDefaultInitValue(
6329             FieldIt->getType(),
6330             Result.getStructField(FieldIt->getFieldIndex()));
6331     }
6332   }
6333 
6334   EvalObj.finishedConstructingFields();
6335 
6336   return Success &&
6337          EvaluateStmt(Ret, Info, Definition->getBody()) != ESR_Failed &&
6338          LifetimeExtendedScope.destroy();
6339 }
6340 
HandleConstructorCall(const Expr * E,const LValue & This,ArrayRef<const Expr * > Args,const CXXConstructorDecl * Definition,EvalInfo & Info,APValue & Result)6341 static bool HandleConstructorCall(const Expr *E, const LValue &This,
6342                                   ArrayRef<const Expr*> Args,
6343                                   const CXXConstructorDecl *Definition,
6344                                   EvalInfo &Info, APValue &Result) {
6345   CallScopeRAII CallScope(Info);
6346   CallRef Call = Info.CurrentCall->createCall(Definition);
6347   if (!EvaluateArgs(Args, Call, Info, Definition))
6348     return false;
6349 
6350   return HandleConstructorCall(E, This, Call, Definition, Info, Result) &&
6351          CallScope.destroy();
6352 }
6353 
HandleDestructionImpl(EvalInfo & Info,SourceLocation CallLoc,const LValue & This,APValue & Value,QualType T)6354 static bool HandleDestructionImpl(EvalInfo &Info, SourceLocation CallLoc,
6355                                   const LValue &This, APValue &Value,
6356                                   QualType T) {
6357   // Objects can only be destroyed while they're within their lifetimes.
6358   // FIXME: We have no representation for whether an object of type nullptr_t
6359   // is in its lifetime; it usually doesn't matter. Perhaps we should model it
6360   // as indeterminate instead?
6361   if (Value.isAbsent() && !T->isNullPtrType()) {
6362     APValue Printable;
6363     This.moveInto(Printable);
6364     Info.FFDiag(CallLoc, diag::note_constexpr_destroy_out_of_lifetime)
6365       << Printable.getAsString(Info.Ctx, Info.Ctx.getLValueReferenceType(T));
6366     return false;
6367   }
6368 
6369   // Invent an expression for location purposes.
6370   // FIXME: We shouldn't need to do this.
6371   OpaqueValueExpr LocE(CallLoc, Info.Ctx.IntTy, VK_PRValue);
6372 
6373   // For arrays, destroy elements right-to-left.
6374   if (const ConstantArrayType *CAT = Info.Ctx.getAsConstantArrayType(T)) {
6375     uint64_t Size = CAT->getSize().getZExtValue();
6376     QualType ElemT = CAT->getElementType();
6377 
6378     LValue ElemLV = This;
6379     ElemLV.addArray(Info, &LocE, CAT);
6380     if (!HandleLValueArrayAdjustment(Info, &LocE, ElemLV, ElemT, Size))
6381       return false;
6382 
6383     // Ensure that we have actual array elements available to destroy; the
6384     // destructors might mutate the value, so we can't run them on the array
6385     // filler.
6386     if (Size && Size > Value.getArrayInitializedElts())
6387       expandArray(Value, Value.getArraySize() - 1);
6388 
6389     for (; Size != 0; --Size) {
6390       APValue &Elem = Value.getArrayInitializedElt(Size - 1);
6391       if (!HandleLValueArrayAdjustment(Info, &LocE, ElemLV, ElemT, -1) ||
6392           !HandleDestructionImpl(Info, CallLoc, ElemLV, Elem, ElemT))
6393         return false;
6394     }
6395 
6396     // End the lifetime of this array now.
6397     Value = APValue();
6398     return true;
6399   }
6400 
6401   const CXXRecordDecl *RD = T->getAsCXXRecordDecl();
6402   if (!RD) {
6403     if (T.isDestructedType()) {
6404       Info.FFDiag(CallLoc, diag::note_constexpr_unsupported_destruction) << T;
6405       return false;
6406     }
6407 
6408     Value = APValue();
6409     return true;
6410   }
6411 
6412   if (RD->getNumVBases()) {
6413     Info.FFDiag(CallLoc, diag::note_constexpr_virtual_base) << RD;
6414     return false;
6415   }
6416 
6417   const CXXDestructorDecl *DD = RD->getDestructor();
6418   if (!DD && !RD->hasTrivialDestructor()) {
6419     Info.FFDiag(CallLoc);
6420     return false;
6421   }
6422 
6423   if (!DD || DD->isTrivial() ||
6424       (RD->isAnonymousStructOrUnion() && RD->isUnion())) {
6425     // A trivial destructor just ends the lifetime of the object. Check for
6426     // this case before checking for a body, because we might not bother
6427     // building a body for a trivial destructor. Note that it doesn't matter
6428     // whether the destructor is constexpr in this case; all trivial
6429     // destructors are constexpr.
6430     //
6431     // If an anonymous union would be destroyed, some enclosing destructor must
6432     // have been explicitly defined, and the anonymous union destruction should
6433     // have no effect.
6434     Value = APValue();
6435     return true;
6436   }
6437 
6438   if (!Info.CheckCallLimit(CallLoc))
6439     return false;
6440 
6441   const FunctionDecl *Definition = nullptr;
6442   const Stmt *Body = DD->getBody(Definition);
6443 
6444   if (!CheckConstexprFunction(Info, CallLoc, DD, Definition, Body))
6445     return false;
6446 
6447   CallStackFrame Frame(Info, CallLoc, Definition, &This, CallRef());
6448 
6449   // We're now in the period of destruction of this object.
6450   unsigned BasesLeft = RD->getNumBases();
6451   EvalInfo::EvaluatingDestructorRAII EvalObj(
6452       Info,
6453       ObjectUnderConstruction{This.getLValueBase(), This.Designator.Entries});
6454   if (!EvalObj.DidInsert) {
6455     // C++2a [class.dtor]p19:
6456     //   the behavior is undefined if the destructor is invoked for an object
6457     //   whose lifetime has ended
6458     // (Note that formally the lifetime ends when the period of destruction
6459     // begins, even though certain uses of the object remain valid until the
6460     // period of destruction ends.)
6461     Info.FFDiag(CallLoc, diag::note_constexpr_double_destroy);
6462     return false;
6463   }
6464 
6465   // FIXME: Creating an APValue just to hold a nonexistent return value is
6466   // wasteful.
6467   APValue RetVal;
6468   StmtResult Ret = {RetVal, nullptr};
6469   if (EvaluateStmt(Ret, Info, Definition->getBody()) == ESR_Failed)
6470     return false;
6471 
6472   // A union destructor does not implicitly destroy its members.
6473   if (RD->isUnion())
6474     return true;
6475 
6476   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
6477 
6478   // We don't have a good way to iterate fields in reverse, so collect all the
6479   // fields first and then walk them backwards.
6480   SmallVector<FieldDecl*, 16> Fields(RD->field_begin(), RD->field_end());
6481   for (const FieldDecl *FD : llvm::reverse(Fields)) {
6482     if (FD->isUnnamedBitfield())
6483       continue;
6484 
6485     LValue Subobject = This;
6486     if (!HandleLValueMember(Info, &LocE, Subobject, FD, &Layout))
6487       return false;
6488 
6489     APValue *SubobjectValue = &Value.getStructField(FD->getFieldIndex());
6490     if (!HandleDestructionImpl(Info, CallLoc, Subobject, *SubobjectValue,
6491                                FD->getType()))
6492       return false;
6493   }
6494 
6495   if (BasesLeft != 0)
6496     EvalObj.startedDestroyingBases();
6497 
6498   // Destroy base classes in reverse order.
6499   for (const CXXBaseSpecifier &Base : llvm::reverse(RD->bases())) {
6500     --BasesLeft;
6501 
6502     QualType BaseType = Base.getType();
6503     LValue Subobject = This;
6504     if (!HandleLValueDirectBase(Info, &LocE, Subobject, RD,
6505                                 BaseType->getAsCXXRecordDecl(), &Layout))
6506       return false;
6507 
6508     APValue *SubobjectValue = &Value.getStructBase(BasesLeft);
6509     if (!HandleDestructionImpl(Info, CallLoc, Subobject, *SubobjectValue,
6510                                BaseType))
6511       return false;
6512   }
6513   assert(BasesLeft == 0 && "NumBases was wrong?");
6514 
6515   // The period of destruction ends now. The object is gone.
6516   Value = APValue();
6517   return true;
6518 }
6519 
6520 namespace {
6521 struct DestroyObjectHandler {
6522   EvalInfo &Info;
6523   const Expr *E;
6524   const LValue &This;
6525   const AccessKinds AccessKind;
6526 
6527   typedef bool result_type;
failed__anonb66d72d21411::DestroyObjectHandler6528   bool failed() { return false; }
found__anonb66d72d21411::DestroyObjectHandler6529   bool found(APValue &Subobj, QualType SubobjType) {
6530     return HandleDestructionImpl(Info, E->getExprLoc(), This, Subobj,
6531                                  SubobjType);
6532   }
found__anonb66d72d21411::DestroyObjectHandler6533   bool found(APSInt &Value, QualType SubobjType) {
6534     Info.FFDiag(E, diag::note_constexpr_destroy_complex_elem);
6535     return false;
6536   }
found__anonb66d72d21411::DestroyObjectHandler6537   bool found(APFloat &Value, QualType SubobjType) {
6538     Info.FFDiag(E, diag::note_constexpr_destroy_complex_elem);
6539     return false;
6540   }
6541 };
6542 }
6543 
6544 /// Perform a destructor or pseudo-destructor call on the given object, which
6545 /// might in general not be a complete object.
HandleDestruction(EvalInfo & Info,const Expr * E,const LValue & This,QualType ThisType)6546 static bool HandleDestruction(EvalInfo &Info, const Expr *E,
6547                               const LValue &This, QualType ThisType) {
6548   CompleteObject Obj = findCompleteObject(Info, E, AK_Destroy, This, ThisType);
6549   DestroyObjectHandler Handler = {Info, E, This, AK_Destroy};
6550   return Obj && findSubobject(Info, E, Obj, This.Designator, Handler);
6551 }
6552 
6553 /// Destroy and end the lifetime of the given complete object.
HandleDestruction(EvalInfo & Info,SourceLocation Loc,APValue::LValueBase LVBase,APValue & Value,QualType T)6554 static bool HandleDestruction(EvalInfo &Info, SourceLocation Loc,
6555                               APValue::LValueBase LVBase, APValue &Value,
6556                               QualType T) {
6557   // If we've had an unmodeled side-effect, we can't rely on mutable state
6558   // (such as the object we're about to destroy) being correct.
6559   if (Info.EvalStatus.HasSideEffects)
6560     return false;
6561 
6562   LValue LV;
6563   LV.set({LVBase});
6564   return HandleDestructionImpl(Info, Loc, LV, Value, T);
6565 }
6566 
6567 /// Perform a call to 'perator new' or to `__builtin_operator_new'.
HandleOperatorNewCall(EvalInfo & Info,const CallExpr * E,LValue & Result)6568 static bool HandleOperatorNewCall(EvalInfo &Info, const CallExpr *E,
6569                                   LValue &Result) {
6570   if (Info.checkingPotentialConstantExpression() ||
6571       Info.SpeculativeEvaluationDepth)
6572     return false;
6573 
6574   // This is permitted only within a call to std::allocator<T>::allocate.
6575   auto Caller = Info.getStdAllocatorCaller("allocate");
6576   if (!Caller) {
6577     Info.FFDiag(E->getExprLoc(), Info.getLangOpts().CPlusPlus20
6578                                      ? diag::note_constexpr_new_untyped
6579                                      : diag::note_constexpr_new);
6580     return false;
6581   }
6582 
6583   QualType ElemType = Caller.ElemType;
6584   if (ElemType->isIncompleteType() || ElemType->isFunctionType()) {
6585     Info.FFDiag(E->getExprLoc(),
6586                 diag::note_constexpr_new_not_complete_object_type)
6587         << (ElemType->isIncompleteType() ? 0 : 1) << ElemType;
6588     return false;
6589   }
6590 
6591   APSInt ByteSize;
6592   if (!EvaluateInteger(E->getArg(0), ByteSize, Info))
6593     return false;
6594   bool IsNothrow = false;
6595   for (unsigned I = 1, N = E->getNumArgs(); I != N; ++I) {
6596     EvaluateIgnoredValue(Info, E->getArg(I));
6597     IsNothrow |= E->getType()->isNothrowT();
6598   }
6599 
6600   CharUnits ElemSize;
6601   if (!HandleSizeof(Info, E->getExprLoc(), ElemType, ElemSize))
6602     return false;
6603   APInt Size, Remainder;
6604   APInt ElemSizeAP(ByteSize.getBitWidth(), ElemSize.getQuantity());
6605   APInt::udivrem(ByteSize, ElemSizeAP, Size, Remainder);
6606   if (Remainder != 0) {
6607     // This likely indicates a bug in the implementation of 'std::allocator'.
6608     Info.FFDiag(E->getExprLoc(), diag::note_constexpr_operator_new_bad_size)
6609         << ByteSize << APSInt(ElemSizeAP, true) << ElemType;
6610     return false;
6611   }
6612 
6613   if (ByteSize.getActiveBits() > ConstantArrayType::getMaxSizeBits(Info.Ctx)) {
6614     if (IsNothrow) {
6615       Result.setNull(Info.Ctx, E->getType());
6616       return true;
6617     }
6618 
6619     Info.FFDiag(E, diag::note_constexpr_new_too_large) << APSInt(Size, true);
6620     return false;
6621   }
6622 
6623   QualType AllocType = Info.Ctx.getConstantArrayType(ElemType, Size, nullptr,
6624                                                      ArrayType::Normal, 0);
6625   APValue *Val = Info.createHeapAlloc(E, AllocType, Result);
6626   *Val = APValue(APValue::UninitArray(), 0, Size.getZExtValue());
6627   Result.addArray(Info, E, cast<ConstantArrayType>(AllocType));
6628   return true;
6629 }
6630 
hasVirtualDestructor(QualType T)6631 static bool hasVirtualDestructor(QualType T) {
6632   if (CXXRecordDecl *RD = T->getAsCXXRecordDecl())
6633     if (CXXDestructorDecl *DD = RD->getDestructor())
6634       return DD->isVirtual();
6635   return false;
6636 }
6637 
getVirtualOperatorDelete(QualType T)6638 static const FunctionDecl *getVirtualOperatorDelete(QualType T) {
6639   if (CXXRecordDecl *RD = T->getAsCXXRecordDecl())
6640     if (CXXDestructorDecl *DD = RD->getDestructor())
6641       return DD->isVirtual() ? DD->getOperatorDelete() : nullptr;
6642   return nullptr;
6643 }
6644 
6645 /// Check that the given object is a suitable pointer to a heap allocation that
6646 /// still exists and is of the right kind for the purpose of a deletion.
6647 ///
6648 /// On success, returns the heap allocation to deallocate. On failure, produces
6649 /// a diagnostic and returns None.
CheckDeleteKind(EvalInfo & Info,const Expr * E,const LValue & Pointer,DynAlloc::Kind DeallocKind)6650 static Optional<DynAlloc *> CheckDeleteKind(EvalInfo &Info, const Expr *E,
6651                                             const LValue &Pointer,
6652                                             DynAlloc::Kind DeallocKind) {
6653   auto PointerAsString = [&] {
6654     return Pointer.toString(Info.Ctx, Info.Ctx.VoidPtrTy);
6655   };
6656 
6657   DynamicAllocLValue DA = Pointer.Base.dyn_cast<DynamicAllocLValue>();
6658   if (!DA) {
6659     Info.FFDiag(E, diag::note_constexpr_delete_not_heap_alloc)
6660         << PointerAsString();
6661     if (Pointer.Base)
6662       NoteLValueLocation(Info, Pointer.Base);
6663     return None;
6664   }
6665 
6666   Optional<DynAlloc *> Alloc = Info.lookupDynamicAlloc(DA);
6667   if (!Alloc) {
6668     Info.FFDiag(E, diag::note_constexpr_double_delete);
6669     return None;
6670   }
6671 
6672   QualType AllocType = Pointer.Base.getDynamicAllocType();
6673   if (DeallocKind != (*Alloc)->getKind()) {
6674     Info.FFDiag(E, diag::note_constexpr_new_delete_mismatch)
6675         << DeallocKind << (*Alloc)->getKind() << AllocType;
6676     NoteLValueLocation(Info, Pointer.Base);
6677     return None;
6678   }
6679 
6680   bool Subobject = false;
6681   if (DeallocKind == DynAlloc::New) {
6682     Subobject = Pointer.Designator.MostDerivedPathLength != 0 ||
6683                 Pointer.Designator.isOnePastTheEnd();
6684   } else {
6685     Subobject = Pointer.Designator.Entries.size() != 1 ||
6686                 Pointer.Designator.Entries[0].getAsArrayIndex() != 0;
6687   }
6688   if (Subobject) {
6689     Info.FFDiag(E, diag::note_constexpr_delete_subobject)
6690         << PointerAsString() << Pointer.Designator.isOnePastTheEnd();
6691     return None;
6692   }
6693 
6694   return Alloc;
6695 }
6696 
6697 // Perform a call to 'operator delete' or '__builtin_operator_delete'.
HandleOperatorDeleteCall(EvalInfo & Info,const CallExpr * E)6698 bool HandleOperatorDeleteCall(EvalInfo &Info, const CallExpr *E) {
6699   if (Info.checkingPotentialConstantExpression() ||
6700       Info.SpeculativeEvaluationDepth)
6701     return false;
6702 
6703   // This is permitted only within a call to std::allocator<T>::deallocate.
6704   if (!Info.getStdAllocatorCaller("deallocate")) {
6705     Info.FFDiag(E->getExprLoc());
6706     return true;
6707   }
6708 
6709   LValue Pointer;
6710   if (!EvaluatePointer(E->getArg(0), Pointer, Info))
6711     return false;
6712   for (unsigned I = 1, N = E->getNumArgs(); I != N; ++I)
6713     EvaluateIgnoredValue(Info, E->getArg(I));
6714 
6715   if (Pointer.Designator.Invalid)
6716     return false;
6717 
6718   // Deleting a null pointer would have no effect, but it's not permitted by
6719   // std::allocator<T>::deallocate's contract.
6720   if (Pointer.isNullPointer()) {
6721     Info.CCEDiag(E->getExprLoc(), diag::note_constexpr_deallocate_null);
6722     return true;
6723   }
6724 
6725   if (!CheckDeleteKind(Info, E, Pointer, DynAlloc::StdAllocator))
6726     return false;
6727 
6728   Info.HeapAllocs.erase(Pointer.Base.get<DynamicAllocLValue>());
6729   return true;
6730 }
6731 
6732 //===----------------------------------------------------------------------===//
6733 // Generic Evaluation
6734 //===----------------------------------------------------------------------===//
6735 namespace {
6736 
6737 class BitCastBuffer {
6738   // FIXME: We're going to need bit-level granularity when we support
6739   // bit-fields.
6740   // FIXME: Its possible under the C++ standard for 'char' to not be 8 bits, but
6741   // we don't support a host or target where that is the case. Still, we should
6742   // use a more generic type in case we ever do.
6743   SmallVector<Optional<unsigned char>, 32> Bytes;
6744 
6745   static_assert(std::numeric_limits<unsigned char>::digits >= 8,
6746                 "Need at least 8 bit unsigned char");
6747 
6748   bool TargetIsLittleEndian;
6749 
6750 public:
BitCastBuffer(CharUnits Width,bool TargetIsLittleEndian)6751   BitCastBuffer(CharUnits Width, bool TargetIsLittleEndian)
6752       : Bytes(Width.getQuantity()),
6753         TargetIsLittleEndian(TargetIsLittleEndian) {}
6754 
6755   LLVM_NODISCARD
readObject(CharUnits Offset,CharUnits Width,SmallVectorImpl<unsigned char> & Output) const6756   bool readObject(CharUnits Offset, CharUnits Width,
6757                   SmallVectorImpl<unsigned char> &Output) const {
6758     for (CharUnits I = Offset, E = Offset + Width; I != E; ++I) {
6759       // If a byte of an integer is uninitialized, then the whole integer is
6760       // uninitalized.
6761       if (!Bytes[I.getQuantity()])
6762         return false;
6763       Output.push_back(*Bytes[I.getQuantity()]);
6764     }
6765     if (llvm::sys::IsLittleEndianHost != TargetIsLittleEndian)
6766       std::reverse(Output.begin(), Output.end());
6767     return true;
6768   }
6769 
writeObject(CharUnits Offset,SmallVectorImpl<unsigned char> & Input)6770   void writeObject(CharUnits Offset, SmallVectorImpl<unsigned char> &Input) {
6771     if (llvm::sys::IsLittleEndianHost != TargetIsLittleEndian)
6772       std::reverse(Input.begin(), Input.end());
6773 
6774     size_t Index = 0;
6775     for (unsigned char Byte : Input) {
6776       assert(!Bytes[Offset.getQuantity() + Index] && "overwriting a byte?");
6777       Bytes[Offset.getQuantity() + Index] = Byte;
6778       ++Index;
6779     }
6780   }
6781 
size()6782   size_t size() { return Bytes.size(); }
6783 };
6784 
6785 /// Traverse an APValue to produce an BitCastBuffer, emulating how the current
6786 /// target would represent the value at runtime.
6787 class APValueToBufferConverter {
6788   EvalInfo &Info;
6789   BitCastBuffer Buffer;
6790   const CastExpr *BCE;
6791 
APValueToBufferConverter(EvalInfo & Info,CharUnits ObjectWidth,const CastExpr * BCE)6792   APValueToBufferConverter(EvalInfo &Info, CharUnits ObjectWidth,
6793                            const CastExpr *BCE)
6794       : Info(Info),
6795         Buffer(ObjectWidth, Info.Ctx.getTargetInfo().isLittleEndian()),
6796         BCE(BCE) {}
6797 
visit(const APValue & Val,QualType Ty)6798   bool visit(const APValue &Val, QualType Ty) {
6799     return visit(Val, Ty, CharUnits::fromQuantity(0));
6800   }
6801 
6802   // Write out Val with type Ty into Buffer starting at Offset.
visit(const APValue & Val,QualType Ty,CharUnits Offset)6803   bool visit(const APValue &Val, QualType Ty, CharUnits Offset) {
6804     assert((size_t)Offset.getQuantity() <= Buffer.size());
6805 
6806     // As a special case, nullptr_t has an indeterminate value.
6807     if (Ty->isNullPtrType())
6808       return true;
6809 
6810     // Dig through Src to find the byte at SrcOffset.
6811     switch (Val.getKind()) {
6812     case APValue::Indeterminate:
6813     case APValue::None:
6814       return true;
6815 
6816     case APValue::Int:
6817       return visitInt(Val.getInt(), Ty, Offset);
6818     case APValue::Float:
6819       return visitFloat(Val.getFloat(), Ty, Offset);
6820     case APValue::Array:
6821       return visitArray(Val, Ty, Offset);
6822     case APValue::Struct:
6823       return visitRecord(Val, Ty, Offset);
6824 
6825     case APValue::ComplexInt:
6826     case APValue::ComplexFloat:
6827     case APValue::Vector:
6828     case APValue::FixedPoint:
6829       // FIXME: We should support these.
6830 
6831     case APValue::Union:
6832     case APValue::MemberPointer:
6833     case APValue::AddrLabelDiff: {
6834       Info.FFDiag(BCE->getBeginLoc(),
6835                   diag::note_constexpr_bit_cast_unsupported_type)
6836           << Ty;
6837       return false;
6838     }
6839 
6840     case APValue::LValue:
6841       llvm_unreachable("LValue subobject in bit_cast?");
6842     }
6843     llvm_unreachable("Unhandled APValue::ValueKind");
6844   }
6845 
visitRecord(const APValue & Val,QualType Ty,CharUnits Offset)6846   bool visitRecord(const APValue &Val, QualType Ty, CharUnits Offset) {
6847     const RecordDecl *RD = Ty->getAsRecordDecl();
6848     const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
6849 
6850     // Visit the base classes.
6851     if (auto *CXXRD = dyn_cast<CXXRecordDecl>(RD)) {
6852       for (size_t I = 0, E = CXXRD->getNumBases(); I != E; ++I) {
6853         const CXXBaseSpecifier &BS = CXXRD->bases_begin()[I];
6854         CXXRecordDecl *BaseDecl = BS.getType()->getAsCXXRecordDecl();
6855 
6856         if (!visitRecord(Val.getStructBase(I), BS.getType(),
6857                          Layout.getBaseClassOffset(BaseDecl) + Offset))
6858           return false;
6859       }
6860     }
6861 
6862     // Visit the fields.
6863     unsigned FieldIdx = 0;
6864     for (FieldDecl *FD : RD->fields()) {
6865       if (FD->isBitField()) {
6866         Info.FFDiag(BCE->getBeginLoc(),
6867                     diag::note_constexpr_bit_cast_unsupported_bitfield);
6868         return false;
6869       }
6870 
6871       uint64_t FieldOffsetBits = Layout.getFieldOffset(FieldIdx);
6872 
6873       assert(FieldOffsetBits % Info.Ctx.getCharWidth() == 0 &&
6874              "only bit-fields can have sub-char alignment");
6875       CharUnits FieldOffset =
6876           Info.Ctx.toCharUnitsFromBits(FieldOffsetBits) + Offset;
6877       QualType FieldTy = FD->getType();
6878       if (!visit(Val.getStructField(FieldIdx), FieldTy, FieldOffset))
6879         return false;
6880       ++FieldIdx;
6881     }
6882 
6883     return true;
6884   }
6885 
visitArray(const APValue & Val,QualType Ty,CharUnits Offset)6886   bool visitArray(const APValue &Val, QualType Ty, CharUnits Offset) {
6887     const auto *CAT =
6888         dyn_cast_or_null<ConstantArrayType>(Ty->getAsArrayTypeUnsafe());
6889     if (!CAT)
6890       return false;
6891 
6892     CharUnits ElemWidth = Info.Ctx.getTypeSizeInChars(CAT->getElementType());
6893     unsigned NumInitializedElts = Val.getArrayInitializedElts();
6894     unsigned ArraySize = Val.getArraySize();
6895     // First, initialize the initialized elements.
6896     for (unsigned I = 0; I != NumInitializedElts; ++I) {
6897       const APValue &SubObj = Val.getArrayInitializedElt(I);
6898       if (!visit(SubObj, CAT->getElementType(), Offset + I * ElemWidth))
6899         return false;
6900     }
6901 
6902     // Next, initialize the rest of the array using the filler.
6903     if (Val.hasArrayFiller()) {
6904       const APValue &Filler = Val.getArrayFiller();
6905       for (unsigned I = NumInitializedElts; I != ArraySize; ++I) {
6906         if (!visit(Filler, CAT->getElementType(), Offset + I * ElemWidth))
6907           return false;
6908       }
6909     }
6910 
6911     return true;
6912   }
6913 
visitInt(const APSInt & Val,QualType Ty,CharUnits Offset)6914   bool visitInt(const APSInt &Val, QualType Ty, CharUnits Offset) {
6915     APSInt AdjustedVal = Val;
6916     unsigned Width = AdjustedVal.getBitWidth();
6917     if (Ty->isBooleanType()) {
6918       Width = Info.Ctx.getTypeSize(Ty);
6919       AdjustedVal = AdjustedVal.extend(Width);
6920     }
6921 
6922     SmallVector<unsigned char, 8> Bytes(Width / 8);
6923     llvm::StoreIntToMemory(AdjustedVal, &*Bytes.begin(), Width / 8);
6924     Buffer.writeObject(Offset, Bytes);
6925     return true;
6926   }
6927 
visitFloat(const APFloat & Val,QualType Ty,CharUnits Offset)6928   bool visitFloat(const APFloat &Val, QualType Ty, CharUnits Offset) {
6929     APSInt AsInt(Val.bitcastToAPInt());
6930     return visitInt(AsInt, Ty, Offset);
6931   }
6932 
6933 public:
convert(EvalInfo & Info,const APValue & Src,const CastExpr * BCE)6934   static Optional<BitCastBuffer> convert(EvalInfo &Info, const APValue &Src,
6935                                          const CastExpr *BCE) {
6936     CharUnits DstSize = Info.Ctx.getTypeSizeInChars(BCE->getType());
6937     APValueToBufferConverter Converter(Info, DstSize, BCE);
6938     if (!Converter.visit(Src, BCE->getSubExpr()->getType()))
6939       return None;
6940     return Converter.Buffer;
6941   }
6942 };
6943 
6944 /// Write an BitCastBuffer into an APValue.
6945 class BufferToAPValueConverter {
6946   EvalInfo &Info;
6947   const BitCastBuffer &Buffer;
6948   const CastExpr *BCE;
6949 
BufferToAPValueConverter(EvalInfo & Info,const BitCastBuffer & Buffer,const CastExpr * BCE)6950   BufferToAPValueConverter(EvalInfo &Info, const BitCastBuffer &Buffer,
6951                            const CastExpr *BCE)
6952       : Info(Info), Buffer(Buffer), BCE(BCE) {}
6953 
6954   // Emit an unsupported bit_cast type error. Sema refuses to build a bit_cast
6955   // with an invalid type, so anything left is a deficiency on our part (FIXME).
6956   // Ideally this will be unreachable.
unsupportedType(QualType Ty)6957   llvm::NoneType unsupportedType(QualType Ty) {
6958     Info.FFDiag(BCE->getBeginLoc(),
6959                 diag::note_constexpr_bit_cast_unsupported_type)
6960         << Ty;
6961     return None;
6962   }
6963 
unrepresentableValue(QualType Ty,const APSInt & Val)6964   llvm::NoneType unrepresentableValue(QualType Ty, const APSInt &Val) {
6965     Info.FFDiag(BCE->getBeginLoc(),
6966                 diag::note_constexpr_bit_cast_unrepresentable_value)
6967         << Ty << toString(Val, /*Radix=*/10);
6968     return None;
6969   }
6970 
visit(const BuiltinType * T,CharUnits Offset,const EnumType * EnumSugar=nullptr)6971   Optional<APValue> visit(const BuiltinType *T, CharUnits Offset,
6972                           const EnumType *EnumSugar = nullptr) {
6973     if (T->isNullPtrType()) {
6974       uint64_t NullValue = Info.Ctx.getTargetNullPointerValue(QualType(T, 0));
6975       return APValue((Expr *)nullptr,
6976                      /*Offset=*/CharUnits::fromQuantity(NullValue),
6977                      APValue::NoLValuePath{}, /*IsNullPtr=*/true);
6978     }
6979 
6980     CharUnits SizeOf = Info.Ctx.getTypeSizeInChars(T);
6981 
6982     // Work around floating point types that contain unused padding bytes. This
6983     // is really just `long double` on x86, which is the only fundamental type
6984     // with padding bytes.
6985     if (T->isRealFloatingType()) {
6986       const llvm::fltSemantics &Semantics =
6987           Info.Ctx.getFloatTypeSemantics(QualType(T, 0));
6988       unsigned NumBits = llvm::APFloatBase::getSizeInBits(Semantics);
6989       assert(NumBits % 8 == 0);
6990       CharUnits NumBytes = CharUnits::fromQuantity(NumBits / 8);
6991       if (NumBytes != SizeOf)
6992         SizeOf = NumBytes;
6993     }
6994 
6995     SmallVector<uint8_t, 8> Bytes;
6996     if (!Buffer.readObject(Offset, SizeOf, Bytes)) {
6997       // If this is std::byte or unsigned char, then its okay to store an
6998       // indeterminate value.
6999       bool IsStdByte = EnumSugar && EnumSugar->isStdByteType();
7000       bool IsUChar =
7001           !EnumSugar && (T->isSpecificBuiltinType(BuiltinType::UChar) ||
7002                          T->isSpecificBuiltinType(BuiltinType::Char_U));
7003       if (!IsStdByte && !IsUChar) {
7004         QualType DisplayType(EnumSugar ? (const Type *)EnumSugar : T, 0);
7005         Info.FFDiag(BCE->getExprLoc(),
7006                     diag::note_constexpr_bit_cast_indet_dest)
7007             << DisplayType << Info.Ctx.getLangOpts().CharIsSigned;
7008         return None;
7009       }
7010 
7011       return APValue::IndeterminateValue();
7012     }
7013 
7014     APSInt Val(SizeOf.getQuantity() * Info.Ctx.getCharWidth(), true);
7015     llvm::LoadIntFromMemory(Val, &*Bytes.begin(), Bytes.size());
7016 
7017     if (T->isIntegralOrEnumerationType()) {
7018       Val.setIsSigned(T->isSignedIntegerOrEnumerationType());
7019 
7020       unsigned IntWidth = Info.Ctx.getIntWidth(QualType(T, 0));
7021       if (IntWidth != Val.getBitWidth()) {
7022         APSInt Truncated = Val.trunc(IntWidth);
7023         if (Truncated.extend(Val.getBitWidth()) != Val)
7024           return unrepresentableValue(QualType(T, 0), Val);
7025         Val = Truncated;
7026       }
7027 
7028       return APValue(Val);
7029     }
7030 
7031     if (T->isRealFloatingType()) {
7032       const llvm::fltSemantics &Semantics =
7033           Info.Ctx.getFloatTypeSemantics(QualType(T, 0));
7034       return APValue(APFloat(Semantics, Val));
7035     }
7036 
7037     return unsupportedType(QualType(T, 0));
7038   }
7039 
visit(const RecordType * RTy,CharUnits Offset)7040   Optional<APValue> visit(const RecordType *RTy, CharUnits Offset) {
7041     const RecordDecl *RD = RTy->getAsRecordDecl();
7042     const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
7043 
7044     unsigned NumBases = 0;
7045     if (auto *CXXRD = dyn_cast<CXXRecordDecl>(RD))
7046       NumBases = CXXRD->getNumBases();
7047 
7048     APValue ResultVal(APValue::UninitStruct(), NumBases,
7049                       std::distance(RD->field_begin(), RD->field_end()));
7050 
7051     // Visit the base classes.
7052     if (auto *CXXRD = dyn_cast<CXXRecordDecl>(RD)) {
7053       for (size_t I = 0, E = CXXRD->getNumBases(); I != E; ++I) {
7054         const CXXBaseSpecifier &BS = CXXRD->bases_begin()[I];
7055         CXXRecordDecl *BaseDecl = BS.getType()->getAsCXXRecordDecl();
7056         if (BaseDecl->isEmpty() ||
7057             Info.Ctx.getASTRecordLayout(BaseDecl).getNonVirtualSize().isZero())
7058           continue;
7059 
7060         Optional<APValue> SubObj = visitType(
7061             BS.getType(), Layout.getBaseClassOffset(BaseDecl) + Offset);
7062         if (!SubObj)
7063           return None;
7064         ResultVal.getStructBase(I) = *SubObj;
7065       }
7066     }
7067 
7068     // Visit the fields.
7069     unsigned FieldIdx = 0;
7070     for (FieldDecl *FD : RD->fields()) {
7071       // FIXME: We don't currently support bit-fields. A lot of the logic for
7072       // this is in CodeGen, so we need to factor it around.
7073       if (FD->isBitField()) {
7074         Info.FFDiag(BCE->getBeginLoc(),
7075                     diag::note_constexpr_bit_cast_unsupported_bitfield);
7076         return None;
7077       }
7078 
7079       uint64_t FieldOffsetBits = Layout.getFieldOffset(FieldIdx);
7080       assert(FieldOffsetBits % Info.Ctx.getCharWidth() == 0);
7081 
7082       CharUnits FieldOffset =
7083           CharUnits::fromQuantity(FieldOffsetBits / Info.Ctx.getCharWidth()) +
7084           Offset;
7085       QualType FieldTy = FD->getType();
7086       Optional<APValue> SubObj = visitType(FieldTy, FieldOffset);
7087       if (!SubObj)
7088         return None;
7089       ResultVal.getStructField(FieldIdx) = *SubObj;
7090       ++FieldIdx;
7091     }
7092 
7093     return ResultVal;
7094   }
7095 
visit(const EnumType * Ty,CharUnits Offset)7096   Optional<APValue> visit(const EnumType *Ty, CharUnits Offset) {
7097     QualType RepresentationType = Ty->getDecl()->getIntegerType();
7098     assert(!RepresentationType.isNull() &&
7099            "enum forward decl should be caught by Sema");
7100     const auto *AsBuiltin =
7101         RepresentationType.getCanonicalType()->castAs<BuiltinType>();
7102     // Recurse into the underlying type. Treat std::byte transparently as
7103     // unsigned char.
7104     return visit(AsBuiltin, Offset, /*EnumTy=*/Ty);
7105   }
7106 
visit(const ConstantArrayType * Ty,CharUnits Offset)7107   Optional<APValue> visit(const ConstantArrayType *Ty, CharUnits Offset) {
7108     size_t Size = Ty->getSize().getLimitedValue();
7109     CharUnits ElementWidth = Info.Ctx.getTypeSizeInChars(Ty->getElementType());
7110 
7111     APValue ArrayValue(APValue::UninitArray(), Size, Size);
7112     for (size_t I = 0; I != Size; ++I) {
7113       Optional<APValue> ElementValue =
7114           visitType(Ty->getElementType(), Offset + I * ElementWidth);
7115       if (!ElementValue)
7116         return None;
7117       ArrayValue.getArrayInitializedElt(I) = std::move(*ElementValue);
7118     }
7119 
7120     return ArrayValue;
7121   }
7122 
visit(const Type * Ty,CharUnits Offset)7123   Optional<APValue> visit(const Type *Ty, CharUnits Offset) {
7124     return unsupportedType(QualType(Ty, 0));
7125   }
7126 
visitType(QualType Ty,CharUnits Offset)7127   Optional<APValue> visitType(QualType Ty, CharUnits Offset) {
7128     QualType Can = Ty.getCanonicalType();
7129 
7130     switch (Can->getTypeClass()) {
7131 #define TYPE(Class, Base)                                                      \
7132   case Type::Class:                                                            \
7133     return visit(cast<Class##Type>(Can.getTypePtr()), Offset);
7134 #define ABSTRACT_TYPE(Class, Base)
7135 #define NON_CANONICAL_TYPE(Class, Base)                                        \
7136   case Type::Class:                                                            \
7137     llvm_unreachable("non-canonical type should be impossible!");
7138 #define DEPENDENT_TYPE(Class, Base)                                            \
7139   case Type::Class:                                                            \
7140     llvm_unreachable(                                                          \
7141         "dependent types aren't supported in the constant evaluator!");
7142 #define NON_CANONICAL_UNLESS_DEPENDENT(Class, Base)                            \
7143   case Type::Class:                                                            \
7144     llvm_unreachable("either dependent or not canonical!");
7145 #include "clang/AST/TypeNodes.inc"
7146     }
7147     llvm_unreachable("Unhandled Type::TypeClass");
7148   }
7149 
7150 public:
7151   // Pull out a full value of type DstType.
convert(EvalInfo & Info,BitCastBuffer & Buffer,const CastExpr * BCE)7152   static Optional<APValue> convert(EvalInfo &Info, BitCastBuffer &Buffer,
7153                                    const CastExpr *BCE) {
7154     BufferToAPValueConverter Converter(Info, Buffer, BCE);
7155     return Converter.visitType(BCE->getType(), CharUnits::fromQuantity(0));
7156   }
7157 };
7158 
checkBitCastConstexprEligibilityType(SourceLocation Loc,QualType Ty,EvalInfo * Info,const ASTContext & Ctx,bool CheckingDest)7159 static bool checkBitCastConstexprEligibilityType(SourceLocation Loc,
7160                                                  QualType Ty, EvalInfo *Info,
7161                                                  const ASTContext &Ctx,
7162                                                  bool CheckingDest) {
7163   Ty = Ty.getCanonicalType();
7164 
7165   auto diag = [&](int Reason) {
7166     if (Info)
7167       Info->FFDiag(Loc, diag::note_constexpr_bit_cast_invalid_type)
7168           << CheckingDest << (Reason == 4) << Reason;
7169     return false;
7170   };
7171   auto note = [&](int Construct, QualType NoteTy, SourceLocation NoteLoc) {
7172     if (Info)
7173       Info->Note(NoteLoc, diag::note_constexpr_bit_cast_invalid_subtype)
7174           << NoteTy << Construct << Ty;
7175     return false;
7176   };
7177 
7178   if (Ty->isUnionType())
7179     return diag(0);
7180   if (Ty->isPointerType())
7181     return diag(1);
7182   if (Ty->isMemberPointerType())
7183     return diag(2);
7184   if (Ty.isVolatileQualified())
7185     return diag(3);
7186 
7187   if (RecordDecl *Record = Ty->getAsRecordDecl()) {
7188     if (auto *CXXRD = dyn_cast<CXXRecordDecl>(Record)) {
7189       for (CXXBaseSpecifier &BS : CXXRD->bases())
7190         if (!checkBitCastConstexprEligibilityType(Loc, BS.getType(), Info, Ctx,
7191                                                   CheckingDest))
7192           return note(1, BS.getType(), BS.getBeginLoc());
7193     }
7194     for (FieldDecl *FD : Record->fields()) {
7195       if (FD->getType()->isReferenceType())
7196         return diag(4);
7197       if (!checkBitCastConstexprEligibilityType(Loc, FD->getType(), Info, Ctx,
7198                                                 CheckingDest))
7199         return note(0, FD->getType(), FD->getBeginLoc());
7200     }
7201   }
7202 
7203   if (Ty->isArrayType() &&
7204       !checkBitCastConstexprEligibilityType(Loc, Ctx.getBaseElementType(Ty),
7205                                             Info, Ctx, CheckingDest))
7206     return false;
7207 
7208   return true;
7209 }
7210 
checkBitCastConstexprEligibility(EvalInfo * Info,const ASTContext & Ctx,const CastExpr * BCE)7211 static bool checkBitCastConstexprEligibility(EvalInfo *Info,
7212                                              const ASTContext &Ctx,
7213                                              const CastExpr *BCE) {
7214   bool DestOK = checkBitCastConstexprEligibilityType(
7215       BCE->getBeginLoc(), BCE->getType(), Info, Ctx, true);
7216   bool SourceOK = DestOK && checkBitCastConstexprEligibilityType(
7217                                 BCE->getBeginLoc(),
7218                                 BCE->getSubExpr()->getType(), Info, Ctx, false);
7219   return SourceOK;
7220 }
7221 
handleLValueToRValueBitCast(EvalInfo & Info,APValue & DestValue,APValue & SourceValue,const CastExpr * BCE)7222 static bool handleLValueToRValueBitCast(EvalInfo &Info, APValue &DestValue,
7223                                         APValue &SourceValue,
7224                                         const CastExpr *BCE) {
7225   assert(CHAR_BIT == 8 && Info.Ctx.getTargetInfo().getCharWidth() == 8 &&
7226          "no host or target supports non 8-bit chars");
7227   assert(SourceValue.isLValue() &&
7228          "LValueToRValueBitcast requires an lvalue operand!");
7229 
7230   if (!checkBitCastConstexprEligibility(&Info, Info.Ctx, BCE))
7231     return false;
7232 
7233   LValue SourceLValue;
7234   APValue SourceRValue;
7235   SourceLValue.setFrom(Info.Ctx, SourceValue);
7236   if (!handleLValueToRValueConversion(
7237           Info, BCE, BCE->getSubExpr()->getType().withConst(), SourceLValue,
7238           SourceRValue, /*WantObjectRepresentation=*/true))
7239     return false;
7240 
7241   // Read out SourceValue into a char buffer.
7242   Optional<BitCastBuffer> Buffer =
7243       APValueToBufferConverter::convert(Info, SourceRValue, BCE);
7244   if (!Buffer)
7245     return false;
7246 
7247   // Write out the buffer into a new APValue.
7248   Optional<APValue> MaybeDestValue =
7249       BufferToAPValueConverter::convert(Info, *Buffer, BCE);
7250   if (!MaybeDestValue)
7251     return false;
7252 
7253   DestValue = std::move(*MaybeDestValue);
7254   return true;
7255 }
7256 
7257 template <class Derived>
7258 class ExprEvaluatorBase
7259   : public ConstStmtVisitor<Derived, bool> {
7260 private:
getDerived()7261   Derived &getDerived() { return static_cast<Derived&>(*this); }
DerivedSuccess(const APValue & V,const Expr * E)7262   bool DerivedSuccess(const APValue &V, const Expr *E) {
7263     return getDerived().Success(V, E);
7264   }
DerivedZeroInitialization(const Expr * E)7265   bool DerivedZeroInitialization(const Expr *E) {
7266     return getDerived().ZeroInitialization(E);
7267   }
7268 
7269   // Check whether a conditional operator with a non-constant condition is a
7270   // potential constant expression. If neither arm is a potential constant
7271   // expression, then the conditional operator is not either.
7272   template<typename ConditionalOperator>
CheckPotentialConstantConditional(const ConditionalOperator * E)7273   void CheckPotentialConstantConditional(const ConditionalOperator *E) {
7274     assert(Info.checkingPotentialConstantExpression());
7275 
7276     // Speculatively evaluate both arms.
7277     SmallVector<PartialDiagnosticAt, 8> Diag;
7278     {
7279       SpeculativeEvaluationRAII Speculate(Info, &Diag);
7280       StmtVisitorTy::Visit(E->getFalseExpr());
7281       if (Diag.empty())
7282         return;
7283     }
7284 
7285     {
7286       SpeculativeEvaluationRAII Speculate(Info, &Diag);
7287       Diag.clear();
7288       StmtVisitorTy::Visit(E->getTrueExpr());
7289       if (Diag.empty())
7290         return;
7291     }
7292 
7293     Error(E, diag::note_constexpr_conditional_never_const);
7294   }
7295 
7296 
7297   template<typename ConditionalOperator>
HandleConditionalOperator(const ConditionalOperator * E)7298   bool HandleConditionalOperator(const ConditionalOperator *E) {
7299     bool BoolResult;
7300     if (!EvaluateAsBooleanCondition(E->getCond(), BoolResult, Info)) {
7301       if (Info.checkingPotentialConstantExpression() && Info.noteFailure()) {
7302         CheckPotentialConstantConditional(E);
7303         return false;
7304       }
7305       if (Info.noteFailure()) {
7306         StmtVisitorTy::Visit(E->getTrueExpr());
7307         StmtVisitorTy::Visit(E->getFalseExpr());
7308       }
7309       return false;
7310     }
7311 
7312     Expr *EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
7313     return StmtVisitorTy::Visit(EvalExpr);
7314   }
7315 
7316 protected:
7317   EvalInfo &Info;
7318   typedef ConstStmtVisitor<Derived, bool> StmtVisitorTy;
7319   typedef ExprEvaluatorBase ExprEvaluatorBaseTy;
7320 
CCEDiag(const Expr * E,diag::kind D)7321   OptionalDiagnostic CCEDiag(const Expr *E, diag::kind D) {
7322     return Info.CCEDiag(E, D);
7323   }
7324 
ZeroInitialization(const Expr * E)7325   bool ZeroInitialization(const Expr *E) { return Error(E); }
7326 
7327 public:
ExprEvaluatorBase(EvalInfo & Info)7328   ExprEvaluatorBase(EvalInfo &Info) : Info(Info) {}
7329 
getEvalInfo()7330   EvalInfo &getEvalInfo() { return Info; }
7331 
7332   /// Report an evaluation error. This should only be called when an error is
7333   /// first discovered. When propagating an error, just return false.
Error(const Expr * E,diag::kind D)7334   bool Error(const Expr *E, diag::kind D) {
7335     Info.FFDiag(E, D);
7336     return false;
7337   }
Error(const Expr * E)7338   bool Error(const Expr *E) {
7339     return Error(E, diag::note_invalid_subexpr_in_const_expr);
7340   }
7341 
VisitStmt(const Stmt *)7342   bool VisitStmt(const Stmt *) {
7343     llvm_unreachable("Expression evaluator should not be called on stmts");
7344   }
VisitExpr(const Expr * E)7345   bool VisitExpr(const Expr *E) {
7346     return Error(E);
7347   }
7348 
VisitConstantExpr(const ConstantExpr * E)7349   bool VisitConstantExpr(const ConstantExpr *E) {
7350     if (E->hasAPValueResult())
7351       return DerivedSuccess(E->getAPValueResult(), E);
7352 
7353     return StmtVisitorTy::Visit(E->getSubExpr());
7354   }
7355 
VisitParenExpr(const ParenExpr * E)7356   bool VisitParenExpr(const ParenExpr *E)
7357     { return StmtVisitorTy::Visit(E->getSubExpr()); }
VisitUnaryExtension(const UnaryOperator * E)7358   bool VisitUnaryExtension(const UnaryOperator *E)
7359     { return StmtVisitorTy::Visit(E->getSubExpr()); }
VisitUnaryPlus(const UnaryOperator * E)7360   bool VisitUnaryPlus(const UnaryOperator *E)
7361     { return StmtVisitorTy::Visit(E->getSubExpr()); }
VisitChooseExpr(const ChooseExpr * E)7362   bool VisitChooseExpr(const ChooseExpr *E)
7363     { return StmtVisitorTy::Visit(E->getChosenSubExpr()); }
VisitGenericSelectionExpr(const GenericSelectionExpr * E)7364   bool VisitGenericSelectionExpr(const GenericSelectionExpr *E)
7365     { return StmtVisitorTy::Visit(E->getResultExpr()); }
VisitSubstNonTypeTemplateParmExpr(const SubstNonTypeTemplateParmExpr * E)7366   bool VisitSubstNonTypeTemplateParmExpr(const SubstNonTypeTemplateParmExpr *E)
7367     { return StmtVisitorTy::Visit(E->getReplacement()); }
VisitCXXDefaultArgExpr(const CXXDefaultArgExpr * E)7368   bool VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *E) {
7369     TempVersionRAII RAII(*Info.CurrentCall);
7370     SourceLocExprScopeGuard Guard(E, Info.CurrentCall->CurSourceLocExprScope);
7371     return StmtVisitorTy::Visit(E->getExpr());
7372   }
VisitCXXDefaultInitExpr(const CXXDefaultInitExpr * E)7373   bool VisitCXXDefaultInitExpr(const CXXDefaultInitExpr *E) {
7374     TempVersionRAII RAII(*Info.CurrentCall);
7375     // The initializer may not have been parsed yet, or might be erroneous.
7376     if (!E->getExpr())
7377       return Error(E);
7378     SourceLocExprScopeGuard Guard(E, Info.CurrentCall->CurSourceLocExprScope);
7379     return StmtVisitorTy::Visit(E->getExpr());
7380   }
7381 
VisitExprWithCleanups(const ExprWithCleanups * E)7382   bool VisitExprWithCleanups(const ExprWithCleanups *E) {
7383     FullExpressionRAII Scope(Info);
7384     return StmtVisitorTy::Visit(E->getSubExpr()) && Scope.destroy();
7385   }
7386 
7387   // Temporaries are registered when created, so we don't care about
7388   // CXXBindTemporaryExpr.
VisitCXXBindTemporaryExpr(const CXXBindTemporaryExpr * E)7389   bool VisitCXXBindTemporaryExpr(const CXXBindTemporaryExpr *E) {
7390     return StmtVisitorTy::Visit(E->getSubExpr());
7391   }
7392 
VisitCXXReinterpretCastExpr(const CXXReinterpretCastExpr * E)7393   bool VisitCXXReinterpretCastExpr(const CXXReinterpretCastExpr *E) {
7394     CCEDiag(E, diag::note_constexpr_invalid_cast) << 0;
7395     return static_cast<Derived*>(this)->VisitCastExpr(E);
7396   }
VisitCXXDynamicCastExpr(const CXXDynamicCastExpr * E)7397   bool VisitCXXDynamicCastExpr(const CXXDynamicCastExpr *E) {
7398     if (!Info.Ctx.getLangOpts().CPlusPlus20)
7399       CCEDiag(E, diag::note_constexpr_invalid_cast) << 1;
7400     return static_cast<Derived*>(this)->VisitCastExpr(E);
7401   }
VisitBuiltinBitCastExpr(const BuiltinBitCastExpr * E)7402   bool VisitBuiltinBitCastExpr(const BuiltinBitCastExpr *E) {
7403     return static_cast<Derived*>(this)->VisitCastExpr(E);
7404   }
7405 
VisitBinaryOperator(const BinaryOperator * E)7406   bool VisitBinaryOperator(const BinaryOperator *E) {
7407     switch (E->getOpcode()) {
7408     default:
7409       return Error(E);
7410 
7411     case BO_Comma:
7412       VisitIgnoredValue(E->getLHS());
7413       return StmtVisitorTy::Visit(E->getRHS());
7414 
7415     case BO_PtrMemD:
7416     case BO_PtrMemI: {
7417       LValue Obj;
7418       if (!HandleMemberPointerAccess(Info, E, Obj))
7419         return false;
7420       APValue Result;
7421       if (!handleLValueToRValueConversion(Info, E, E->getType(), Obj, Result))
7422         return false;
7423       return DerivedSuccess(Result, E);
7424     }
7425     }
7426   }
7427 
VisitCXXRewrittenBinaryOperator(const CXXRewrittenBinaryOperator * E)7428   bool VisitCXXRewrittenBinaryOperator(const CXXRewrittenBinaryOperator *E) {
7429     return StmtVisitorTy::Visit(E->getSemanticForm());
7430   }
7431 
VisitBinaryConditionalOperator(const BinaryConditionalOperator * E)7432   bool VisitBinaryConditionalOperator(const BinaryConditionalOperator *E) {
7433     // Evaluate and cache the common expression. We treat it as a temporary,
7434     // even though it's not quite the same thing.
7435     LValue CommonLV;
7436     if (!Evaluate(Info.CurrentCall->createTemporary(
7437                       E->getOpaqueValue(),
7438                       getStorageType(Info.Ctx, E->getOpaqueValue()),
7439                       ScopeKind::FullExpression, CommonLV),
7440                   Info, E->getCommon()))
7441       return false;
7442 
7443     return HandleConditionalOperator(E);
7444   }
7445 
VisitConditionalOperator(const ConditionalOperator * E)7446   bool VisitConditionalOperator(const ConditionalOperator *E) {
7447     bool IsBcpCall = false;
7448     // If the condition (ignoring parens) is a __builtin_constant_p call,
7449     // the result is a constant expression if it can be folded without
7450     // side-effects. This is an important GNU extension. See GCC PR38377
7451     // for discussion.
7452     if (const CallExpr *CallCE =
7453           dyn_cast<CallExpr>(E->getCond()->IgnoreParenCasts()))
7454       if (CallCE->getBuiltinCallee() == Builtin::BI__builtin_constant_p)
7455         IsBcpCall = true;
7456 
7457     // Always assume __builtin_constant_p(...) ? ... : ... is a potential
7458     // constant expression; we can't check whether it's potentially foldable.
7459     // FIXME: We should instead treat __builtin_constant_p as non-constant if
7460     // it would return 'false' in this mode.
7461     if (Info.checkingPotentialConstantExpression() && IsBcpCall)
7462       return false;
7463 
7464     FoldConstant Fold(Info, IsBcpCall);
7465     if (!HandleConditionalOperator(E)) {
7466       Fold.keepDiagnostics();
7467       return false;
7468     }
7469 
7470     return true;
7471   }
7472 
VisitOpaqueValueExpr(const OpaqueValueExpr * E)7473   bool VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
7474     if (APValue *Value = Info.CurrentCall->getCurrentTemporary(E))
7475       return DerivedSuccess(*Value, E);
7476 
7477     const Expr *Source = E->getSourceExpr();
7478     if (!Source)
7479       return Error(E);
7480     if (Source == E) { // sanity checking.
7481       assert(0 && "OpaqueValueExpr recursively refers to itself");
7482       return Error(E);
7483     }
7484     return StmtVisitorTy::Visit(Source);
7485   }
7486 
VisitPseudoObjectExpr(const PseudoObjectExpr * E)7487   bool VisitPseudoObjectExpr(const PseudoObjectExpr *E) {
7488     for (const Expr *SemE : E->semantics()) {
7489       if (auto *OVE = dyn_cast<OpaqueValueExpr>(SemE)) {
7490         // FIXME: We can't handle the case where an OpaqueValueExpr is also the
7491         // result expression: there could be two different LValues that would
7492         // refer to the same object in that case, and we can't model that.
7493         if (SemE == E->getResultExpr())
7494           return Error(E);
7495 
7496         // Unique OVEs get evaluated if and when we encounter them when
7497         // emitting the rest of the semantic form, rather than eagerly.
7498         if (OVE->isUnique())
7499           continue;
7500 
7501         LValue LV;
7502         if (!Evaluate(Info.CurrentCall->createTemporary(
7503                           OVE, getStorageType(Info.Ctx, OVE),
7504                           ScopeKind::FullExpression, LV),
7505                       Info, OVE->getSourceExpr()))
7506           return false;
7507       } else if (SemE == E->getResultExpr()) {
7508         if (!StmtVisitorTy::Visit(SemE))
7509           return false;
7510       } else {
7511         if (!EvaluateIgnoredValue(Info, SemE))
7512           return false;
7513       }
7514     }
7515     return true;
7516   }
7517 
VisitCallExpr(const CallExpr * E)7518   bool VisitCallExpr(const CallExpr *E) {
7519     APValue Result;
7520     if (!handleCallExpr(E, Result, nullptr))
7521       return false;
7522     return DerivedSuccess(Result, E);
7523   }
7524 
handleCallExpr(const CallExpr * E,APValue & Result,const LValue * ResultSlot)7525   bool handleCallExpr(const CallExpr *E, APValue &Result,
7526                      const LValue *ResultSlot) {
7527     CallScopeRAII CallScope(Info);
7528 
7529     const Expr *Callee = E->getCallee()->IgnoreParens();
7530     QualType CalleeType = Callee->getType();
7531 
7532     const FunctionDecl *FD = nullptr;
7533     LValue *This = nullptr, ThisVal;
7534     auto Args = llvm::makeArrayRef(E->getArgs(), E->getNumArgs());
7535     bool HasQualifier = false;
7536 
7537     CallRef Call;
7538 
7539     // Extract function decl and 'this' pointer from the callee.
7540     if (CalleeType->isSpecificBuiltinType(BuiltinType::BoundMember)) {
7541       const CXXMethodDecl *Member = nullptr;
7542       if (const MemberExpr *ME = dyn_cast<MemberExpr>(Callee)) {
7543         // Explicit bound member calls, such as x.f() or p->g();
7544         if (!EvaluateObjectArgument(Info, ME->getBase(), ThisVal))
7545           return false;
7546         Member = dyn_cast<CXXMethodDecl>(ME->getMemberDecl());
7547         if (!Member)
7548           return Error(Callee);
7549         This = &ThisVal;
7550         HasQualifier = ME->hasQualifier();
7551       } else if (const BinaryOperator *BE = dyn_cast<BinaryOperator>(Callee)) {
7552         // Indirect bound member calls ('.*' or '->*').
7553         const ValueDecl *D =
7554             HandleMemberPointerAccess(Info, BE, ThisVal, false);
7555         if (!D)
7556           return false;
7557         Member = dyn_cast<CXXMethodDecl>(D);
7558         if (!Member)
7559           return Error(Callee);
7560         This = &ThisVal;
7561       } else if (const auto *PDE = dyn_cast<CXXPseudoDestructorExpr>(Callee)) {
7562         if (!Info.getLangOpts().CPlusPlus20)
7563           Info.CCEDiag(PDE, diag::note_constexpr_pseudo_destructor);
7564         return EvaluateObjectArgument(Info, PDE->getBase(), ThisVal) &&
7565                HandleDestruction(Info, PDE, ThisVal, PDE->getDestroyedType());
7566       } else
7567         return Error(Callee);
7568       FD = Member;
7569     } else if (CalleeType->isFunctionPointerType()) {
7570       LValue CalleeLV;
7571       if (!EvaluatePointer(Callee, CalleeLV, Info))
7572         return false;
7573 
7574       if (!CalleeLV.getLValueOffset().isZero())
7575         return Error(Callee);
7576       FD = dyn_cast_or_null<FunctionDecl>(
7577           CalleeLV.getLValueBase().dyn_cast<const ValueDecl *>());
7578       if (!FD)
7579         return Error(Callee);
7580       // Don't call function pointers which have been cast to some other type.
7581       // Per DR (no number yet), the caller and callee can differ in noexcept.
7582       if (!Info.Ctx.hasSameFunctionTypeIgnoringExceptionSpec(
7583         CalleeType->getPointeeType(), FD->getType())) {
7584         return Error(E);
7585       }
7586 
7587       // For an (overloaded) assignment expression, evaluate the RHS before the
7588       // LHS.
7589       auto *OCE = dyn_cast<CXXOperatorCallExpr>(E);
7590       if (OCE && OCE->isAssignmentOp()) {
7591         assert(Args.size() == 2 && "wrong number of arguments in assignment");
7592         Call = Info.CurrentCall->createCall(FD);
7593         if (!EvaluateArgs(isa<CXXMethodDecl>(FD) ? Args.slice(1) : Args, Call,
7594                           Info, FD, /*RightToLeft=*/true))
7595           return false;
7596       }
7597 
7598       // Overloaded operator calls to member functions are represented as normal
7599       // calls with '*this' as the first argument.
7600       const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD);
7601       if (MD && !MD->isStatic()) {
7602         // FIXME: When selecting an implicit conversion for an overloaded
7603         // operator delete, we sometimes try to evaluate calls to conversion
7604         // operators without a 'this' parameter!
7605         if (Args.empty())
7606           return Error(E);
7607 
7608         if (!EvaluateObjectArgument(Info, Args[0], ThisVal))
7609           return false;
7610         This = &ThisVal;
7611 
7612         // If this is syntactically a simple assignment using a trivial
7613         // assignment operator, start the lifetimes of union members as needed,
7614         // per C++20 [class.union]5.
7615         if (Info.getLangOpts().CPlusPlus20 && OCE &&
7616             OCE->getOperator() == OO_Equal && MD->isTrivial() &&
7617             !HandleUnionActiveMemberChange(Info, Args[0], ThisVal))
7618           return false;
7619 
7620         Args = Args.slice(1);
7621       } else if (MD && MD->isLambdaStaticInvoker()) {
7622         // Map the static invoker for the lambda back to the call operator.
7623         // Conveniently, we don't have to slice out the 'this' argument (as is
7624         // being done for the non-static case), since a static member function
7625         // doesn't have an implicit argument passed in.
7626         const CXXRecordDecl *ClosureClass = MD->getParent();
7627         assert(
7628             ClosureClass->captures_begin() == ClosureClass->captures_end() &&
7629             "Number of captures must be zero for conversion to function-ptr");
7630 
7631         const CXXMethodDecl *LambdaCallOp =
7632             ClosureClass->getLambdaCallOperator();
7633 
7634         // Set 'FD', the function that will be called below, to the call
7635         // operator.  If the closure object represents a generic lambda, find
7636         // the corresponding specialization of the call operator.
7637 
7638         if (ClosureClass->isGenericLambda()) {
7639           assert(MD->isFunctionTemplateSpecialization() &&
7640                  "A generic lambda's static-invoker function must be a "
7641                  "template specialization");
7642           const TemplateArgumentList *TAL = MD->getTemplateSpecializationArgs();
7643           FunctionTemplateDecl *CallOpTemplate =
7644               LambdaCallOp->getDescribedFunctionTemplate();
7645           void *InsertPos = nullptr;
7646           FunctionDecl *CorrespondingCallOpSpecialization =
7647               CallOpTemplate->findSpecialization(TAL->asArray(), InsertPos);
7648           assert(CorrespondingCallOpSpecialization &&
7649                  "We must always have a function call operator specialization "
7650                  "that corresponds to our static invoker specialization");
7651           FD = cast<CXXMethodDecl>(CorrespondingCallOpSpecialization);
7652         } else
7653           FD = LambdaCallOp;
7654       } else if (FD->isReplaceableGlobalAllocationFunction()) {
7655         if (FD->getDeclName().getCXXOverloadedOperator() == OO_New ||
7656             FD->getDeclName().getCXXOverloadedOperator() == OO_Array_New) {
7657           LValue Ptr;
7658           if (!HandleOperatorNewCall(Info, E, Ptr))
7659             return false;
7660           Ptr.moveInto(Result);
7661           return CallScope.destroy();
7662         } else {
7663           return HandleOperatorDeleteCall(Info, E) && CallScope.destroy();
7664         }
7665       }
7666     } else
7667       return Error(E);
7668 
7669     // Evaluate the arguments now if we've not already done so.
7670     if (!Call) {
7671       Call = Info.CurrentCall->createCall(FD);
7672       if (!EvaluateArgs(Args, Call, Info, FD))
7673         return false;
7674     }
7675 
7676     SmallVector<QualType, 4> CovariantAdjustmentPath;
7677     if (This) {
7678       auto *NamedMember = dyn_cast<CXXMethodDecl>(FD);
7679       if (NamedMember && NamedMember->isVirtual() && !HasQualifier) {
7680         // Perform virtual dispatch, if necessary.
7681         FD = HandleVirtualDispatch(Info, E, *This, NamedMember,
7682                                    CovariantAdjustmentPath);
7683         if (!FD)
7684           return false;
7685       } else {
7686         // Check that the 'this' pointer points to an object of the right type.
7687         // FIXME: If this is an assignment operator call, we may need to change
7688         // the active union member before we check this.
7689         if (!checkNonVirtualMemberCallThisPointer(Info, E, *This, NamedMember))
7690           return false;
7691       }
7692     }
7693 
7694     // Destructor calls are different enough that they have their own codepath.
7695     if (auto *DD = dyn_cast<CXXDestructorDecl>(FD)) {
7696       assert(This && "no 'this' pointer for destructor call");
7697       return HandleDestruction(Info, E, *This,
7698                                Info.Ctx.getRecordType(DD->getParent())) &&
7699              CallScope.destroy();
7700     }
7701 
7702     const FunctionDecl *Definition = nullptr;
7703     Stmt *Body = FD->getBody(Definition);
7704 
7705     if (!CheckConstexprFunction(Info, E->getExprLoc(), FD, Definition, Body) ||
7706         !HandleFunctionCall(E->getExprLoc(), Definition, This, Args, Call,
7707                             Body, Info, Result, ResultSlot))
7708       return false;
7709 
7710     if (!CovariantAdjustmentPath.empty() &&
7711         !HandleCovariantReturnAdjustment(Info, E, Result,
7712                                          CovariantAdjustmentPath))
7713       return false;
7714 
7715     return CallScope.destroy();
7716   }
7717 
VisitCompoundLiteralExpr(const CompoundLiteralExpr * E)7718   bool VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
7719     return StmtVisitorTy::Visit(E->getInitializer());
7720   }
VisitInitListExpr(const InitListExpr * E)7721   bool VisitInitListExpr(const InitListExpr *E) {
7722     if (E->getNumInits() == 0)
7723       return DerivedZeroInitialization(E);
7724     if (E->getNumInits() == 1)
7725       return StmtVisitorTy::Visit(E->getInit(0));
7726     return Error(E);
7727   }
VisitImplicitValueInitExpr(const ImplicitValueInitExpr * E)7728   bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
7729     return DerivedZeroInitialization(E);
7730   }
VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr * E)7731   bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
7732     return DerivedZeroInitialization(E);
7733   }
VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr * E)7734   bool VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *E) {
7735     return DerivedZeroInitialization(E);
7736   }
7737 
7738   /// A member expression where the object is a prvalue is itself a prvalue.
VisitMemberExpr(const MemberExpr * E)7739   bool VisitMemberExpr(const MemberExpr *E) {
7740     assert(!Info.Ctx.getLangOpts().CPlusPlus11 &&
7741            "missing temporary materialization conversion");
7742     assert(!E->isArrow() && "missing call to bound member function?");
7743 
7744     APValue Val;
7745     if (!Evaluate(Val, Info, E->getBase()))
7746       return false;
7747 
7748     QualType BaseTy = E->getBase()->getType();
7749 
7750     const FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
7751     if (!FD) return Error(E);
7752     assert(!FD->getType()->isReferenceType() && "prvalue reference?");
7753     assert(BaseTy->castAs<RecordType>()->getDecl()->getCanonicalDecl() ==
7754            FD->getParent()->getCanonicalDecl() && "record / field mismatch");
7755 
7756     // Note: there is no lvalue base here. But this case should only ever
7757     // happen in C or in C++98, where we cannot be evaluating a constexpr
7758     // constructor, which is the only case the base matters.
7759     CompleteObject Obj(APValue::LValueBase(), &Val, BaseTy);
7760     SubobjectDesignator Designator(BaseTy);
7761     Designator.addDeclUnchecked(FD);
7762 
7763     APValue Result;
7764     return extractSubobject(Info, E, Obj, Designator, Result) &&
7765            DerivedSuccess(Result, E);
7766   }
7767 
VisitExtVectorElementExpr(const ExtVectorElementExpr * E)7768   bool VisitExtVectorElementExpr(const ExtVectorElementExpr *E) {
7769     APValue Val;
7770     if (!Evaluate(Val, Info, E->getBase()))
7771       return false;
7772 
7773     if (Val.isVector()) {
7774       SmallVector<uint32_t, 4> Indices;
7775       E->getEncodedElementAccess(Indices);
7776       if (Indices.size() == 1) {
7777         // Return scalar.
7778         return DerivedSuccess(Val.getVectorElt(Indices[0]), E);
7779       } else {
7780         // Construct new APValue vector.
7781         SmallVector<APValue, 4> Elts;
7782         for (unsigned I = 0; I < Indices.size(); ++I) {
7783           Elts.push_back(Val.getVectorElt(Indices[I]));
7784         }
7785         APValue VecResult(Elts.data(), Indices.size());
7786         return DerivedSuccess(VecResult, E);
7787       }
7788     }
7789 
7790     return false;
7791   }
7792 
VisitCastExpr(const CastExpr * E)7793   bool VisitCastExpr(const CastExpr *E) {
7794     switch (E->getCastKind()) {
7795     default:
7796       break;
7797 
7798     case CK_AtomicToNonAtomic: {
7799       APValue AtomicVal;
7800       // This does not need to be done in place even for class/array types:
7801       // atomic-to-non-atomic conversion implies copying the object
7802       // representation.
7803       if (!Evaluate(AtomicVal, Info, E->getSubExpr()))
7804         return false;
7805       return DerivedSuccess(AtomicVal, E);
7806     }
7807 
7808     case CK_NoOp:
7809     case CK_UserDefinedConversion:
7810       return StmtVisitorTy::Visit(E->getSubExpr());
7811 
7812     case CK_LValueToRValue: {
7813       LValue LVal;
7814       if (!EvaluateLValue(E->getSubExpr(), LVal, Info))
7815         return false;
7816       APValue RVal;
7817       // Note, we use the subexpression's type in order to retain cv-qualifiers.
7818       if (!handleLValueToRValueConversion(Info, E, E->getSubExpr()->getType(),
7819                                           LVal, RVal))
7820         return false;
7821       return DerivedSuccess(RVal, E);
7822     }
7823     case CK_LValueToRValueBitCast: {
7824       APValue DestValue, SourceValue;
7825       if (!Evaluate(SourceValue, Info, E->getSubExpr()))
7826         return false;
7827       if (!handleLValueToRValueBitCast(Info, DestValue, SourceValue, E))
7828         return false;
7829       return DerivedSuccess(DestValue, E);
7830     }
7831 
7832     case CK_AddressSpaceConversion: {
7833       APValue Value;
7834       if (!Evaluate(Value, Info, E->getSubExpr()))
7835         return false;
7836       return DerivedSuccess(Value, E);
7837     }
7838     }
7839 
7840     return Error(E);
7841   }
7842 
VisitUnaryPostInc(const UnaryOperator * UO)7843   bool VisitUnaryPostInc(const UnaryOperator *UO) {
7844     return VisitUnaryPostIncDec(UO);
7845   }
VisitUnaryPostDec(const UnaryOperator * UO)7846   bool VisitUnaryPostDec(const UnaryOperator *UO) {
7847     return VisitUnaryPostIncDec(UO);
7848   }
VisitUnaryPostIncDec(const UnaryOperator * UO)7849   bool VisitUnaryPostIncDec(const UnaryOperator *UO) {
7850     if (!Info.getLangOpts().CPlusPlus14 && !Info.keepEvaluatingAfterFailure())
7851       return Error(UO);
7852 
7853     LValue LVal;
7854     if (!EvaluateLValue(UO->getSubExpr(), LVal, Info))
7855       return false;
7856     APValue RVal;
7857     if (!handleIncDec(this->Info, UO, LVal, UO->getSubExpr()->getType(),
7858                       UO->isIncrementOp(), &RVal))
7859       return false;
7860     return DerivedSuccess(RVal, UO);
7861   }
7862 
VisitStmtExpr(const StmtExpr * E)7863   bool VisitStmtExpr(const StmtExpr *E) {
7864     // We will have checked the full-expressions inside the statement expression
7865     // when they were completed, and don't need to check them again now.
7866     llvm::SaveAndRestore<bool> NotCheckingForUB(
7867         Info.CheckingForUndefinedBehavior, false);
7868 
7869     const CompoundStmt *CS = E->getSubStmt();
7870     if (CS->body_empty())
7871       return true;
7872 
7873     BlockScopeRAII Scope(Info);
7874     for (CompoundStmt::const_body_iterator BI = CS->body_begin(),
7875                                            BE = CS->body_end();
7876          /**/; ++BI) {
7877       if (BI + 1 == BE) {
7878         const Expr *FinalExpr = dyn_cast<Expr>(*BI);
7879         if (!FinalExpr) {
7880           Info.FFDiag((*BI)->getBeginLoc(),
7881                       diag::note_constexpr_stmt_expr_unsupported);
7882           return false;
7883         }
7884         return this->Visit(FinalExpr) && Scope.destroy();
7885       }
7886 
7887       APValue ReturnValue;
7888       StmtResult Result = { ReturnValue, nullptr };
7889       EvalStmtResult ESR = EvaluateStmt(Result, Info, *BI);
7890       if (ESR != ESR_Succeeded) {
7891         // FIXME: If the statement-expression terminated due to 'return',
7892         // 'break', or 'continue', it would be nice to propagate that to
7893         // the outer statement evaluation rather than bailing out.
7894         if (ESR != ESR_Failed)
7895           Info.FFDiag((*BI)->getBeginLoc(),
7896                       diag::note_constexpr_stmt_expr_unsupported);
7897         return false;
7898       }
7899     }
7900 
7901     llvm_unreachable("Return from function from the loop above.");
7902   }
7903 
7904   /// Visit a value which is evaluated, but whose value is ignored.
VisitIgnoredValue(const Expr * E)7905   void VisitIgnoredValue(const Expr *E) {
7906     EvaluateIgnoredValue(Info, E);
7907   }
7908 
7909   /// Potentially visit a MemberExpr's base expression.
VisitIgnoredBaseExpression(const Expr * E)7910   void VisitIgnoredBaseExpression(const Expr *E) {
7911     // While MSVC doesn't evaluate the base expression, it does diagnose the
7912     // presence of side-effecting behavior.
7913     if (Info.getLangOpts().MSVCCompat && !E->HasSideEffects(Info.Ctx))
7914       return;
7915     VisitIgnoredValue(E);
7916   }
7917 };
7918 
7919 } // namespace
7920 
7921 //===----------------------------------------------------------------------===//
7922 // Common base class for lvalue and temporary evaluation.
7923 //===----------------------------------------------------------------------===//
7924 namespace {
7925 template<class Derived>
7926 class LValueExprEvaluatorBase
7927   : public ExprEvaluatorBase<Derived> {
7928 protected:
7929   LValue &Result;
7930   bool InvalidBaseOK;
7931   typedef LValueExprEvaluatorBase LValueExprEvaluatorBaseTy;
7932   typedef ExprEvaluatorBase<Derived> ExprEvaluatorBaseTy;
7933 
Success(APValue::LValueBase B)7934   bool Success(APValue::LValueBase B) {
7935     Result.set(B);
7936     return true;
7937   }
7938 
evaluatePointer(const Expr * E,LValue & Result)7939   bool evaluatePointer(const Expr *E, LValue &Result) {
7940     return EvaluatePointer(E, Result, this->Info, InvalidBaseOK);
7941   }
7942 
7943 public:
LValueExprEvaluatorBase(EvalInfo & Info,LValue & Result,bool InvalidBaseOK)7944   LValueExprEvaluatorBase(EvalInfo &Info, LValue &Result, bool InvalidBaseOK)
7945       : ExprEvaluatorBaseTy(Info), Result(Result),
7946         InvalidBaseOK(InvalidBaseOK) {}
7947 
Success(const APValue & V,const Expr * E)7948   bool Success(const APValue &V, const Expr *E) {
7949     Result.setFrom(this->Info.Ctx, V);
7950     return true;
7951   }
7952 
VisitMemberExpr(const MemberExpr * E)7953   bool VisitMemberExpr(const MemberExpr *E) {
7954     // Handle non-static data members.
7955     QualType BaseTy;
7956     bool EvalOK;
7957     if (E->isArrow()) {
7958       EvalOK = evaluatePointer(E->getBase(), Result);
7959       BaseTy = E->getBase()->getType()->castAs<PointerType>()->getPointeeType();
7960     } else if (E->getBase()->isPRValue()) {
7961       assert(E->getBase()->getType()->isRecordType());
7962       EvalOK = EvaluateTemporary(E->getBase(), Result, this->Info);
7963       BaseTy = E->getBase()->getType();
7964     } else {
7965       EvalOK = this->Visit(E->getBase());
7966       BaseTy = E->getBase()->getType();
7967     }
7968     if (!EvalOK) {
7969       if (!InvalidBaseOK)
7970         return false;
7971       Result.setInvalid(E);
7972       return true;
7973     }
7974 
7975     const ValueDecl *MD = E->getMemberDecl();
7976     if (const FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl())) {
7977       assert(BaseTy->castAs<RecordType>()->getDecl()->getCanonicalDecl() ==
7978              FD->getParent()->getCanonicalDecl() && "record / field mismatch");
7979       (void)BaseTy;
7980       if (!HandleLValueMember(this->Info, E, Result, FD))
7981         return false;
7982     } else if (const IndirectFieldDecl *IFD = dyn_cast<IndirectFieldDecl>(MD)) {
7983       if (!HandleLValueIndirectMember(this->Info, E, Result, IFD))
7984         return false;
7985     } else
7986       return this->Error(E);
7987 
7988     if (MD->getType()->isReferenceType()) {
7989       APValue RefValue;
7990       if (!handleLValueToRValueConversion(this->Info, E, MD->getType(), Result,
7991                                           RefValue))
7992         return false;
7993       return Success(RefValue, E);
7994     }
7995     return true;
7996   }
7997 
VisitBinaryOperator(const BinaryOperator * E)7998   bool VisitBinaryOperator(const BinaryOperator *E) {
7999     switch (E->getOpcode()) {
8000     default:
8001       return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
8002 
8003     case BO_PtrMemD:
8004     case BO_PtrMemI:
8005       return HandleMemberPointerAccess(this->Info, E, Result);
8006     }
8007   }
8008 
VisitCastExpr(const CastExpr * E)8009   bool VisitCastExpr(const CastExpr *E) {
8010     switch (E->getCastKind()) {
8011     default:
8012       return ExprEvaluatorBaseTy::VisitCastExpr(E);
8013 
8014     case CK_DerivedToBase:
8015     case CK_UncheckedDerivedToBase:
8016       if (!this->Visit(E->getSubExpr()))
8017         return false;
8018 
8019       // Now figure out the necessary offset to add to the base LV to get from
8020       // the derived class to the base class.
8021       return HandleLValueBasePath(this->Info, E, E->getSubExpr()->getType(),
8022                                   Result);
8023     }
8024   }
8025 };
8026 }
8027 
8028 //===----------------------------------------------------------------------===//
8029 // LValue Evaluation
8030 //
8031 // This is used for evaluating lvalues (in C and C++), xvalues (in C++11),
8032 // function designators (in C), decl references to void objects (in C), and
8033 // temporaries (if building with -Wno-address-of-temporary).
8034 //
8035 // LValue evaluation produces values comprising a base expression of one of the
8036 // following types:
8037 // - Declarations
8038 //  * VarDecl
8039 //  * FunctionDecl
8040 // - Literals
8041 //  * CompoundLiteralExpr in C (and in global scope in C++)
8042 //  * StringLiteral
8043 //  * PredefinedExpr
8044 //  * ObjCStringLiteralExpr
8045 //  * ObjCEncodeExpr
8046 //  * AddrLabelExpr
8047 //  * BlockExpr
8048 //  * CallExpr for a MakeStringConstant builtin
8049 // - typeid(T) expressions, as TypeInfoLValues
8050 // - Locals and temporaries
8051 //  * MaterializeTemporaryExpr
8052 //  * Any Expr, with a CallIndex indicating the function in which the temporary
8053 //    was evaluated, for cases where the MaterializeTemporaryExpr is missing
8054 //    from the AST (FIXME).
8055 //  * A MaterializeTemporaryExpr that has static storage duration, with no
8056 //    CallIndex, for a lifetime-extended temporary.
8057 //  * The ConstantExpr that is currently being evaluated during evaluation of an
8058 //    immediate invocation.
8059 // plus an offset in bytes.
8060 //===----------------------------------------------------------------------===//
8061 namespace {
8062 class LValueExprEvaluator
8063   : public LValueExprEvaluatorBase<LValueExprEvaluator> {
8064 public:
LValueExprEvaluator(EvalInfo & Info,LValue & Result,bool InvalidBaseOK)8065   LValueExprEvaluator(EvalInfo &Info, LValue &Result, bool InvalidBaseOK) :
8066     LValueExprEvaluatorBaseTy(Info, Result, InvalidBaseOK) {}
8067 
8068   bool VisitVarDecl(const Expr *E, const VarDecl *VD);
8069   bool VisitUnaryPreIncDec(const UnaryOperator *UO);
8070 
8071   bool VisitDeclRefExpr(const DeclRefExpr *E);
VisitPredefinedExpr(const PredefinedExpr * E)8072   bool VisitPredefinedExpr(const PredefinedExpr *E) { return Success(E); }
8073   bool VisitMaterializeTemporaryExpr(const MaterializeTemporaryExpr *E);
8074   bool VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
8075   bool VisitMemberExpr(const MemberExpr *E);
VisitStringLiteral(const StringLiteral * E)8076   bool VisitStringLiteral(const StringLiteral *E) { return Success(E); }
VisitObjCEncodeExpr(const ObjCEncodeExpr * E)8077   bool VisitObjCEncodeExpr(const ObjCEncodeExpr *E) { return Success(E); }
8078   bool VisitCXXTypeidExpr(const CXXTypeidExpr *E);
8079   bool VisitCXXUuidofExpr(const CXXUuidofExpr *E);
8080   bool VisitArraySubscriptExpr(const ArraySubscriptExpr *E);
8081   bool VisitUnaryDeref(const UnaryOperator *E);
8082   bool VisitUnaryReal(const UnaryOperator *E);
8083   bool VisitUnaryImag(const UnaryOperator *E);
VisitUnaryPreInc(const UnaryOperator * UO)8084   bool VisitUnaryPreInc(const UnaryOperator *UO) {
8085     return VisitUnaryPreIncDec(UO);
8086   }
VisitUnaryPreDec(const UnaryOperator * UO)8087   bool VisitUnaryPreDec(const UnaryOperator *UO) {
8088     return VisitUnaryPreIncDec(UO);
8089   }
8090   bool VisitBinAssign(const BinaryOperator *BO);
8091   bool VisitCompoundAssignOperator(const CompoundAssignOperator *CAO);
8092 
VisitCastExpr(const CastExpr * E)8093   bool VisitCastExpr(const CastExpr *E) {
8094     switch (E->getCastKind()) {
8095     default:
8096       return LValueExprEvaluatorBaseTy::VisitCastExpr(E);
8097 
8098     case CK_LValueBitCast:
8099       this->CCEDiag(E, diag::note_constexpr_invalid_cast) << 2;
8100       if (!Visit(E->getSubExpr()))
8101         return false;
8102       Result.Designator.setInvalid();
8103       return true;
8104 
8105     case CK_BaseToDerived:
8106       if (!Visit(E->getSubExpr()))
8107         return false;
8108       return HandleBaseToDerivedCast(Info, E, Result);
8109 
8110     case CK_Dynamic:
8111       if (!Visit(E->getSubExpr()))
8112         return false;
8113       return HandleDynamicCast(Info, cast<ExplicitCastExpr>(E), Result);
8114     }
8115   }
8116 };
8117 } // end anonymous namespace
8118 
8119 /// Evaluate an expression as an lvalue. This can be legitimately called on
8120 /// expressions which are not glvalues, in three cases:
8121 ///  * function designators in C, and
8122 ///  * "extern void" objects
8123 ///  * @selector() expressions in Objective-C
EvaluateLValue(const Expr * E,LValue & Result,EvalInfo & Info,bool InvalidBaseOK)8124 static bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info,
8125                            bool InvalidBaseOK) {
8126   assert(!E->isValueDependent());
8127   assert(E->isGLValue() || E->getType()->isFunctionType() ||
8128          E->getType()->isVoidType() || isa<ObjCSelectorExpr>(E));
8129   return LValueExprEvaluator(Info, Result, InvalidBaseOK).Visit(E);
8130 }
8131 
VisitDeclRefExpr(const DeclRefExpr * E)8132 bool LValueExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
8133   const NamedDecl *D = E->getDecl();
8134   if (isa<FunctionDecl, MSGuidDecl, TemplateParamObjectDecl>(D))
8135     return Success(cast<ValueDecl>(D));
8136   if (const VarDecl *VD = dyn_cast<VarDecl>(D))
8137     return VisitVarDecl(E, VD);
8138   if (const BindingDecl *BD = dyn_cast<BindingDecl>(D))
8139     return Visit(BD->getBinding());
8140   return Error(E);
8141 }
8142 
8143 
VisitVarDecl(const Expr * E,const VarDecl * VD)8144 bool LValueExprEvaluator::VisitVarDecl(const Expr *E, const VarDecl *VD) {
8145 
8146   // If we are within a lambda's call operator, check whether the 'VD' referred
8147   // to within 'E' actually represents a lambda-capture that maps to a
8148   // data-member/field within the closure object, and if so, evaluate to the
8149   // field or what the field refers to.
8150   if (Info.CurrentCall && isLambdaCallOperator(Info.CurrentCall->Callee) &&
8151       isa<DeclRefExpr>(E) &&
8152       cast<DeclRefExpr>(E)->refersToEnclosingVariableOrCapture()) {
8153     // We don't always have a complete capture-map when checking or inferring if
8154     // the function call operator meets the requirements of a constexpr function
8155     // - but we don't need to evaluate the captures to determine constexprness
8156     // (dcl.constexpr C++17).
8157     if (Info.checkingPotentialConstantExpression())
8158       return false;
8159 
8160     if (auto *FD = Info.CurrentCall->LambdaCaptureFields.lookup(VD)) {
8161       // Start with 'Result' referring to the complete closure object...
8162       Result = *Info.CurrentCall->This;
8163       // ... then update it to refer to the field of the closure object
8164       // that represents the capture.
8165       if (!HandleLValueMember(Info, E, Result, FD))
8166         return false;
8167       // And if the field is of reference type, update 'Result' to refer to what
8168       // the field refers to.
8169       if (FD->getType()->isReferenceType()) {
8170         APValue RVal;
8171         if (!handleLValueToRValueConversion(Info, E, FD->getType(), Result,
8172                                             RVal))
8173           return false;
8174         Result.setFrom(Info.Ctx, RVal);
8175       }
8176       return true;
8177     }
8178   }
8179 
8180   CallStackFrame *Frame = nullptr;
8181   unsigned Version = 0;
8182   if (VD->hasLocalStorage()) {
8183     // Only if a local variable was declared in the function currently being
8184     // evaluated, do we expect to be able to find its value in the current
8185     // frame. (Otherwise it was likely declared in an enclosing context and
8186     // could either have a valid evaluatable value (for e.g. a constexpr
8187     // variable) or be ill-formed (and trigger an appropriate evaluation
8188     // diagnostic)).
8189     CallStackFrame *CurrFrame = Info.CurrentCall;
8190     if (CurrFrame->Callee && CurrFrame->Callee->Equals(VD->getDeclContext())) {
8191       // Function parameters are stored in some caller's frame. (Usually the
8192       // immediate caller, but for an inherited constructor they may be more
8193       // distant.)
8194       if (auto *PVD = dyn_cast<ParmVarDecl>(VD)) {
8195         if (CurrFrame->Arguments) {
8196           VD = CurrFrame->Arguments.getOrigParam(PVD);
8197           Frame =
8198               Info.getCallFrameAndDepth(CurrFrame->Arguments.CallIndex).first;
8199           Version = CurrFrame->Arguments.Version;
8200         }
8201       } else {
8202         Frame = CurrFrame;
8203         Version = CurrFrame->getCurrentTemporaryVersion(VD);
8204       }
8205     }
8206   }
8207 
8208   if (!VD->getType()->isReferenceType()) {
8209     if (Frame) {
8210       Result.set({VD, Frame->Index, Version});
8211       return true;
8212     }
8213     return Success(VD);
8214   }
8215 
8216   if (!Info.getLangOpts().CPlusPlus11) {
8217     Info.CCEDiag(E, diag::note_constexpr_ltor_non_integral, 1)
8218         << VD << VD->getType();
8219     Info.Note(VD->getLocation(), diag::note_declared_at);
8220   }
8221 
8222   APValue *V;
8223   if (!evaluateVarDeclInit(Info, E, VD, Frame, Version, V))
8224     return false;
8225   if (!V->hasValue()) {
8226     // FIXME: Is it possible for V to be indeterminate here? If so, we should
8227     // adjust the diagnostic to say that.
8228     if (!Info.checkingPotentialConstantExpression())
8229       Info.FFDiag(E, diag::note_constexpr_use_uninit_reference);
8230     return false;
8231   }
8232   return Success(*V, E);
8233 }
8234 
VisitMaterializeTemporaryExpr(const MaterializeTemporaryExpr * E)8235 bool LValueExprEvaluator::VisitMaterializeTemporaryExpr(
8236     const MaterializeTemporaryExpr *E) {
8237   // Walk through the expression to find the materialized temporary itself.
8238   SmallVector<const Expr *, 2> CommaLHSs;
8239   SmallVector<SubobjectAdjustment, 2> Adjustments;
8240   const Expr *Inner =
8241       E->getSubExpr()->skipRValueSubobjectAdjustments(CommaLHSs, Adjustments);
8242 
8243   // If we passed any comma operators, evaluate their LHSs.
8244   for (unsigned I = 0, N = CommaLHSs.size(); I != N; ++I)
8245     if (!EvaluateIgnoredValue(Info, CommaLHSs[I]))
8246       return false;
8247 
8248   // A materialized temporary with static storage duration can appear within the
8249   // result of a constant expression evaluation, so we need to preserve its
8250   // value for use outside this evaluation.
8251   APValue *Value;
8252   if (E->getStorageDuration() == SD_Static) {
8253     // FIXME: What about SD_Thread?
8254     Value = E->getOrCreateValue(true);
8255     *Value = APValue();
8256     Result.set(E);
8257   } else {
8258     Value = &Info.CurrentCall->createTemporary(
8259         E, E->getType(),
8260         E->getStorageDuration() == SD_FullExpression ? ScopeKind::FullExpression
8261                                                      : ScopeKind::Block,
8262         Result);
8263   }
8264 
8265   QualType Type = Inner->getType();
8266 
8267   // Materialize the temporary itself.
8268   if (!EvaluateInPlace(*Value, Info, Result, Inner)) {
8269     *Value = APValue();
8270     return false;
8271   }
8272 
8273   // Adjust our lvalue to refer to the desired subobject.
8274   for (unsigned I = Adjustments.size(); I != 0; /**/) {
8275     --I;
8276     switch (Adjustments[I].Kind) {
8277     case SubobjectAdjustment::DerivedToBaseAdjustment:
8278       if (!HandleLValueBasePath(Info, Adjustments[I].DerivedToBase.BasePath,
8279                                 Type, Result))
8280         return false;
8281       Type = Adjustments[I].DerivedToBase.BasePath->getType();
8282       break;
8283 
8284     case SubobjectAdjustment::FieldAdjustment:
8285       if (!HandleLValueMember(Info, E, Result, Adjustments[I].Field))
8286         return false;
8287       Type = Adjustments[I].Field->getType();
8288       break;
8289 
8290     case SubobjectAdjustment::MemberPointerAdjustment:
8291       if (!HandleMemberPointerAccess(this->Info, Type, Result,
8292                                      Adjustments[I].Ptr.RHS))
8293         return false;
8294       Type = Adjustments[I].Ptr.MPT->getPointeeType();
8295       break;
8296     }
8297   }
8298 
8299   return true;
8300 }
8301 
8302 bool
VisitCompoundLiteralExpr(const CompoundLiteralExpr * E)8303 LValueExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
8304   assert((!Info.getLangOpts().CPlusPlus || E->isFileScope()) &&
8305          "lvalue compound literal in c++?");
8306   // Defer visiting the literal until the lvalue-to-rvalue conversion. We can
8307   // only see this when folding in C, so there's no standard to follow here.
8308   return Success(E);
8309 }
8310 
VisitCXXTypeidExpr(const CXXTypeidExpr * E)8311 bool LValueExprEvaluator::VisitCXXTypeidExpr(const CXXTypeidExpr *E) {
8312   TypeInfoLValue TypeInfo;
8313 
8314   if (!E->isPotentiallyEvaluated()) {
8315     if (E->isTypeOperand())
8316       TypeInfo = TypeInfoLValue(E->getTypeOperand(Info.Ctx).getTypePtr());
8317     else
8318       TypeInfo = TypeInfoLValue(E->getExprOperand()->getType().getTypePtr());
8319   } else {
8320     if (!Info.Ctx.getLangOpts().CPlusPlus20) {
8321       Info.CCEDiag(E, diag::note_constexpr_typeid_polymorphic)
8322         << E->getExprOperand()->getType()
8323         << E->getExprOperand()->getSourceRange();
8324     }
8325 
8326     if (!Visit(E->getExprOperand()))
8327       return false;
8328 
8329     Optional<DynamicType> DynType =
8330         ComputeDynamicType(Info, E, Result, AK_TypeId);
8331     if (!DynType)
8332       return false;
8333 
8334     TypeInfo =
8335         TypeInfoLValue(Info.Ctx.getRecordType(DynType->Type).getTypePtr());
8336   }
8337 
8338   return Success(APValue::LValueBase::getTypeInfo(TypeInfo, E->getType()));
8339 }
8340 
VisitCXXUuidofExpr(const CXXUuidofExpr * E)8341 bool LValueExprEvaluator::VisitCXXUuidofExpr(const CXXUuidofExpr *E) {
8342   return Success(E->getGuidDecl());
8343 }
8344 
VisitMemberExpr(const MemberExpr * E)8345 bool LValueExprEvaluator::VisitMemberExpr(const MemberExpr *E) {
8346   // Handle static data members.
8347   if (const VarDecl *VD = dyn_cast<VarDecl>(E->getMemberDecl())) {
8348     VisitIgnoredBaseExpression(E->getBase());
8349     return VisitVarDecl(E, VD);
8350   }
8351 
8352   // Handle static member functions.
8353   if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(E->getMemberDecl())) {
8354     if (MD->isStatic()) {
8355       VisitIgnoredBaseExpression(E->getBase());
8356       return Success(MD);
8357     }
8358   }
8359 
8360   // Handle non-static data members.
8361   return LValueExprEvaluatorBaseTy::VisitMemberExpr(E);
8362 }
8363 
VisitArraySubscriptExpr(const ArraySubscriptExpr * E)8364 bool LValueExprEvaluator::VisitArraySubscriptExpr(const ArraySubscriptExpr *E) {
8365   // FIXME: Deal with vectors as array subscript bases.
8366   if (E->getBase()->getType()->isVectorType())
8367     return Error(E);
8368 
8369   APSInt Index;
8370   bool Success = true;
8371 
8372   // C++17's rules require us to evaluate the LHS first, regardless of which
8373   // side is the base.
8374   for (const Expr *SubExpr : {E->getLHS(), E->getRHS()}) {
8375     if (SubExpr == E->getBase() ? !evaluatePointer(SubExpr, Result)
8376                                 : !EvaluateInteger(SubExpr, Index, Info)) {
8377       if (!Info.noteFailure())
8378         return false;
8379       Success = false;
8380     }
8381   }
8382 
8383   return Success &&
8384          HandleLValueArrayAdjustment(Info, E, Result, E->getType(), Index);
8385 }
8386 
VisitUnaryDeref(const UnaryOperator * E)8387 bool LValueExprEvaluator::VisitUnaryDeref(const UnaryOperator *E) {
8388   return evaluatePointer(E->getSubExpr(), Result);
8389 }
8390 
VisitUnaryReal(const UnaryOperator * E)8391 bool LValueExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
8392   if (!Visit(E->getSubExpr()))
8393     return false;
8394   // __real is a no-op on scalar lvalues.
8395   if (E->getSubExpr()->getType()->isAnyComplexType())
8396     HandleLValueComplexElement(Info, E, Result, E->getType(), false);
8397   return true;
8398 }
8399 
VisitUnaryImag(const UnaryOperator * E)8400 bool LValueExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
8401   assert(E->getSubExpr()->getType()->isAnyComplexType() &&
8402          "lvalue __imag__ on scalar?");
8403   if (!Visit(E->getSubExpr()))
8404     return false;
8405   HandleLValueComplexElement(Info, E, Result, E->getType(), true);
8406   return true;
8407 }
8408 
VisitUnaryPreIncDec(const UnaryOperator * UO)8409 bool LValueExprEvaluator::VisitUnaryPreIncDec(const UnaryOperator *UO) {
8410   if (!Info.getLangOpts().CPlusPlus14 && !Info.keepEvaluatingAfterFailure())
8411     return Error(UO);
8412 
8413   if (!this->Visit(UO->getSubExpr()))
8414     return false;
8415 
8416   return handleIncDec(
8417       this->Info, UO, Result, UO->getSubExpr()->getType(),
8418       UO->isIncrementOp(), nullptr);
8419 }
8420 
VisitCompoundAssignOperator(const CompoundAssignOperator * CAO)8421 bool LValueExprEvaluator::VisitCompoundAssignOperator(
8422     const CompoundAssignOperator *CAO) {
8423   if (!Info.getLangOpts().CPlusPlus14 && !Info.keepEvaluatingAfterFailure())
8424     return Error(CAO);
8425 
8426   bool Success = true;
8427 
8428   // C++17 onwards require that we evaluate the RHS first.
8429   APValue RHS;
8430   if (!Evaluate(RHS, this->Info, CAO->getRHS())) {
8431     if (!Info.noteFailure())
8432       return false;
8433     Success = false;
8434   }
8435 
8436   // The overall lvalue result is the result of evaluating the LHS.
8437   if (!this->Visit(CAO->getLHS()) || !Success)
8438     return false;
8439 
8440   return handleCompoundAssignment(
8441       this->Info, CAO,
8442       Result, CAO->getLHS()->getType(), CAO->getComputationLHSType(),
8443       CAO->getOpForCompoundAssignment(CAO->getOpcode()), RHS);
8444 }
8445 
VisitBinAssign(const BinaryOperator * E)8446 bool LValueExprEvaluator::VisitBinAssign(const BinaryOperator *E) {
8447   if (!Info.getLangOpts().CPlusPlus14 && !Info.keepEvaluatingAfterFailure())
8448     return Error(E);
8449 
8450   bool Success = true;
8451 
8452   // C++17 onwards require that we evaluate the RHS first.
8453   APValue NewVal;
8454   if (!Evaluate(NewVal, this->Info, E->getRHS())) {
8455     if (!Info.noteFailure())
8456       return false;
8457     Success = false;
8458   }
8459 
8460   if (!this->Visit(E->getLHS()) || !Success)
8461     return false;
8462 
8463   if (Info.getLangOpts().CPlusPlus20 &&
8464       !HandleUnionActiveMemberChange(Info, E->getLHS(), Result))
8465     return false;
8466 
8467   return handleAssignment(this->Info, E, Result, E->getLHS()->getType(),
8468                           NewVal);
8469 }
8470 
8471 //===----------------------------------------------------------------------===//
8472 // Pointer Evaluation
8473 //===----------------------------------------------------------------------===//
8474 
8475 /// Attempts to compute the number of bytes available at the pointer
8476 /// returned by a function with the alloc_size attribute. Returns true if we
8477 /// were successful. Places an unsigned number into `Result`.
8478 ///
8479 /// This expects the given CallExpr to be a call to a function with an
8480 /// alloc_size attribute.
getBytesReturnedByAllocSizeCall(const ASTContext & Ctx,const CallExpr * Call,llvm::APInt & Result)8481 static bool getBytesReturnedByAllocSizeCall(const ASTContext &Ctx,
8482                                             const CallExpr *Call,
8483                                             llvm::APInt &Result) {
8484   const AllocSizeAttr *AllocSize = getAllocSizeAttr(Call);
8485 
8486   assert(AllocSize && AllocSize->getElemSizeParam().isValid());
8487   unsigned SizeArgNo = AllocSize->getElemSizeParam().getASTIndex();
8488   unsigned BitsInSizeT = Ctx.getTypeSize(Ctx.getSizeType());
8489   if (Call->getNumArgs() <= SizeArgNo)
8490     return false;
8491 
8492   auto EvaluateAsSizeT = [&](const Expr *E, APSInt &Into) {
8493     Expr::EvalResult ExprResult;
8494     if (!E->EvaluateAsInt(ExprResult, Ctx, Expr::SE_AllowSideEffects))
8495       return false;
8496     Into = ExprResult.Val.getInt();
8497     if (Into.isNegative() || !Into.isIntN(BitsInSizeT))
8498       return false;
8499     Into = Into.zextOrSelf(BitsInSizeT);
8500     return true;
8501   };
8502 
8503   APSInt SizeOfElem;
8504   if (!EvaluateAsSizeT(Call->getArg(SizeArgNo), SizeOfElem))
8505     return false;
8506 
8507   if (!AllocSize->getNumElemsParam().isValid()) {
8508     Result = std::move(SizeOfElem);
8509     return true;
8510   }
8511 
8512   APSInt NumberOfElems;
8513   unsigned NumArgNo = AllocSize->getNumElemsParam().getASTIndex();
8514   if (!EvaluateAsSizeT(Call->getArg(NumArgNo), NumberOfElems))
8515     return false;
8516 
8517   bool Overflow;
8518   llvm::APInt BytesAvailable = SizeOfElem.umul_ov(NumberOfElems, Overflow);
8519   if (Overflow)
8520     return false;
8521 
8522   Result = std::move(BytesAvailable);
8523   return true;
8524 }
8525 
8526 /// Convenience function. LVal's base must be a call to an alloc_size
8527 /// function.
getBytesReturnedByAllocSizeCall(const ASTContext & Ctx,const LValue & LVal,llvm::APInt & Result)8528 static bool getBytesReturnedByAllocSizeCall(const ASTContext &Ctx,
8529                                             const LValue &LVal,
8530                                             llvm::APInt &Result) {
8531   assert(isBaseAnAllocSizeCall(LVal.getLValueBase()) &&
8532          "Can't get the size of a non alloc_size function");
8533   const auto *Base = LVal.getLValueBase().get<const Expr *>();
8534   const CallExpr *CE = tryUnwrapAllocSizeCall(Base);
8535   return getBytesReturnedByAllocSizeCall(Ctx, CE, Result);
8536 }
8537 
8538 /// Attempts to evaluate the given LValueBase as the result of a call to
8539 /// a function with the alloc_size attribute. If it was possible to do so, this
8540 /// function will return true, make Result's Base point to said function call,
8541 /// and mark Result's Base as invalid.
evaluateLValueAsAllocSize(EvalInfo & Info,APValue::LValueBase Base,LValue & Result)8542 static bool evaluateLValueAsAllocSize(EvalInfo &Info, APValue::LValueBase Base,
8543                                       LValue &Result) {
8544   if (Base.isNull())
8545     return false;
8546 
8547   // Because we do no form of static analysis, we only support const variables.
8548   //
8549   // Additionally, we can't support parameters, nor can we support static
8550   // variables (in the latter case, use-before-assign isn't UB; in the former,
8551   // we have no clue what they'll be assigned to).
8552   const auto *VD =
8553       dyn_cast_or_null<VarDecl>(Base.dyn_cast<const ValueDecl *>());
8554   if (!VD || !VD->isLocalVarDecl() || !VD->getType().isConstQualified())
8555     return false;
8556 
8557   const Expr *Init = VD->getAnyInitializer();
8558   if (!Init)
8559     return false;
8560 
8561   const Expr *E = Init->IgnoreParens();
8562   if (!tryUnwrapAllocSizeCall(E))
8563     return false;
8564 
8565   // Store E instead of E unwrapped so that the type of the LValue's base is
8566   // what the user wanted.
8567   Result.setInvalid(E);
8568 
8569   QualType Pointee = E->getType()->castAs<PointerType>()->getPointeeType();
8570   Result.addUnsizedArray(Info, E, Pointee);
8571   return true;
8572 }
8573 
8574 namespace {
8575 class PointerExprEvaluator
8576   : public ExprEvaluatorBase<PointerExprEvaluator> {
8577   LValue &Result;
8578   bool InvalidBaseOK;
8579 
Success(const Expr * E)8580   bool Success(const Expr *E) {
8581     Result.set(E);
8582     return true;
8583   }
8584 
evaluateLValue(const Expr * E,LValue & Result)8585   bool evaluateLValue(const Expr *E, LValue &Result) {
8586     return EvaluateLValue(E, Result, Info, InvalidBaseOK);
8587   }
8588 
evaluatePointer(const Expr * E,LValue & Result)8589   bool evaluatePointer(const Expr *E, LValue &Result) {
8590     return EvaluatePointer(E, Result, Info, InvalidBaseOK);
8591   }
8592 
8593   bool visitNonBuiltinCallExpr(const CallExpr *E);
8594 public:
8595 
PointerExprEvaluator(EvalInfo & info,LValue & Result,bool InvalidBaseOK)8596   PointerExprEvaluator(EvalInfo &info, LValue &Result, bool InvalidBaseOK)
8597       : ExprEvaluatorBaseTy(info), Result(Result),
8598         InvalidBaseOK(InvalidBaseOK) {}
8599 
Success(const APValue & V,const Expr * E)8600   bool Success(const APValue &V, const Expr *E) {
8601     Result.setFrom(Info.Ctx, V);
8602     return true;
8603   }
ZeroInitialization(const Expr * E)8604   bool ZeroInitialization(const Expr *E) {
8605     Result.setNull(Info.Ctx, E->getType());
8606     return true;
8607   }
8608 
8609   bool VisitBinaryOperator(const BinaryOperator *E);
8610   bool VisitCastExpr(const CastExpr* E);
8611   bool VisitUnaryAddrOf(const UnaryOperator *E);
VisitObjCStringLiteral(const ObjCStringLiteral * E)8612   bool VisitObjCStringLiteral(const ObjCStringLiteral *E)
8613       { return Success(E); }
VisitObjCBoxedExpr(const ObjCBoxedExpr * E)8614   bool VisitObjCBoxedExpr(const ObjCBoxedExpr *E) {
8615     if (E->isExpressibleAsConstantInitializer())
8616       return Success(E);
8617     if (Info.noteFailure())
8618       EvaluateIgnoredValue(Info, E->getSubExpr());
8619     return Error(E);
8620   }
VisitAddrLabelExpr(const AddrLabelExpr * E)8621   bool VisitAddrLabelExpr(const AddrLabelExpr *E)
8622       { return Success(E); }
8623   bool VisitCallExpr(const CallExpr *E);
8624   bool VisitBuiltinCallExpr(const CallExpr *E, unsigned BuiltinOp);
VisitBlockExpr(const BlockExpr * E)8625   bool VisitBlockExpr(const BlockExpr *E) {
8626     if (!E->getBlockDecl()->hasCaptures())
8627       return Success(E);
8628     return Error(E);
8629   }
VisitCXXThisExpr(const CXXThisExpr * E)8630   bool VisitCXXThisExpr(const CXXThisExpr *E) {
8631     // Can't look at 'this' when checking a potential constant expression.
8632     if (Info.checkingPotentialConstantExpression())
8633       return false;
8634     if (!Info.CurrentCall->This) {
8635       if (Info.getLangOpts().CPlusPlus11)
8636         Info.FFDiag(E, diag::note_constexpr_this) << E->isImplicit();
8637       else
8638         Info.FFDiag(E);
8639       return false;
8640     }
8641     Result = *Info.CurrentCall->This;
8642     // If we are inside a lambda's call operator, the 'this' expression refers
8643     // to the enclosing '*this' object (either by value or reference) which is
8644     // either copied into the closure object's field that represents the '*this'
8645     // or refers to '*this'.
8646     if (isLambdaCallOperator(Info.CurrentCall->Callee)) {
8647       // Ensure we actually have captured 'this'. (an error will have
8648       // been previously reported if not).
8649       if (!Info.CurrentCall->LambdaThisCaptureField)
8650         return false;
8651 
8652       // Update 'Result' to refer to the data member/field of the closure object
8653       // that represents the '*this' capture.
8654       if (!HandleLValueMember(Info, E, Result,
8655                              Info.CurrentCall->LambdaThisCaptureField))
8656         return false;
8657       // If we captured '*this' by reference, replace the field with its referent.
8658       if (Info.CurrentCall->LambdaThisCaptureField->getType()
8659               ->isPointerType()) {
8660         APValue RVal;
8661         if (!handleLValueToRValueConversion(Info, E, E->getType(), Result,
8662                                             RVal))
8663           return false;
8664 
8665         Result.setFrom(Info.Ctx, RVal);
8666       }
8667     }
8668     return true;
8669   }
8670 
8671   bool VisitCXXNewExpr(const CXXNewExpr *E);
8672 
VisitSourceLocExpr(const SourceLocExpr * E)8673   bool VisitSourceLocExpr(const SourceLocExpr *E) {
8674     assert(E->isStringType() && "SourceLocExpr isn't a pointer type?");
8675     APValue LValResult = E->EvaluateInContext(
8676         Info.Ctx, Info.CurrentCall->CurSourceLocExprScope.getDefaultExpr());
8677     Result.setFrom(Info.Ctx, LValResult);
8678     return true;
8679   }
8680 
VisitSYCLUniqueStableNameExpr(const SYCLUniqueStableNameExpr * E)8681   bool VisitSYCLUniqueStableNameExpr(const SYCLUniqueStableNameExpr *E) {
8682     std::string ResultStr = E->ComputeName(Info.Ctx);
8683 
8684     Info.Ctx.SYCLUniqueStableNameEvaluatedValues[E] = ResultStr;
8685 
8686     QualType CharTy = Info.Ctx.CharTy.withConst();
8687     APInt Size(Info.Ctx.getTypeSize(Info.Ctx.getSizeType()),
8688                ResultStr.size() + 1);
8689     QualType ArrayTy = Info.Ctx.getConstantArrayType(CharTy, Size, nullptr,
8690                                                      ArrayType::Normal, 0);
8691 
8692     StringLiteral *SL =
8693         StringLiteral::Create(Info.Ctx, ResultStr, StringLiteral::Ascii,
8694                               /*Pascal*/ false, ArrayTy, E->getLocation());
8695 
8696     evaluateLValue(SL, Result);
8697     Result.addArray(Info, E, cast<ConstantArrayType>(ArrayTy));
8698     return true;
8699   }
8700 
8701   // FIXME: Missing: @protocol, @selector
8702 };
8703 } // end anonymous namespace
8704 
EvaluatePointer(const Expr * E,LValue & Result,EvalInfo & Info,bool InvalidBaseOK)8705 static bool EvaluatePointer(const Expr* E, LValue& Result, EvalInfo &Info,
8706                             bool InvalidBaseOK) {
8707   assert(!E->isValueDependent());
8708   assert(E->isPRValue() && E->getType()->hasPointerRepresentation());
8709   return PointerExprEvaluator(Info, Result, InvalidBaseOK).Visit(E);
8710 }
8711 
VisitBinaryOperator(const BinaryOperator * E)8712 bool PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
8713   if (E->getOpcode() != BO_Add &&
8714       E->getOpcode() != BO_Sub)
8715     return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
8716 
8717   const Expr *PExp = E->getLHS();
8718   const Expr *IExp = E->getRHS();
8719   if (IExp->getType()->isPointerType())
8720     std::swap(PExp, IExp);
8721 
8722   bool EvalPtrOK = evaluatePointer(PExp, Result);
8723   if (!EvalPtrOK && !Info.noteFailure())
8724     return false;
8725 
8726   llvm::APSInt Offset;
8727   if (!EvaluateInteger(IExp, Offset, Info) || !EvalPtrOK)
8728     return false;
8729 
8730   if (E->getOpcode() == BO_Sub)
8731     negateAsSigned(Offset);
8732 
8733   QualType Pointee = PExp->getType()->castAs<PointerType>()->getPointeeType();
8734   return HandleLValueArrayAdjustment(Info, E, Result, Pointee, Offset);
8735 }
8736 
VisitUnaryAddrOf(const UnaryOperator * E)8737 bool PointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
8738   return evaluateLValue(E->getSubExpr(), Result);
8739 }
8740 
VisitCastExpr(const CastExpr * E)8741 bool PointerExprEvaluator::VisitCastExpr(const CastExpr *E) {
8742   const Expr *SubExpr = E->getSubExpr();
8743 
8744   switch (E->getCastKind()) {
8745   default:
8746     break;
8747   case CK_BitCast:
8748   case CK_CPointerToObjCPointerCast:
8749   case CK_BlockPointerToObjCPointerCast:
8750   case CK_AnyPointerToBlockPointerCast:
8751   case CK_AddressSpaceConversion:
8752     if (!Visit(SubExpr))
8753       return false;
8754     // Bitcasts to cv void* are static_casts, not reinterpret_casts, so are
8755     // permitted in constant expressions in C++11. Bitcasts from cv void* are
8756     // also static_casts, but we disallow them as a resolution to DR1312.
8757     if (!E->getType()->isVoidPointerType()) {
8758       if (!Result.InvalidBase && !Result.Designator.Invalid &&
8759           !Result.IsNullPtr &&
8760           Info.Ctx.hasSameUnqualifiedType(Result.Designator.getType(Info.Ctx),
8761                                           E->getType()->getPointeeType()) &&
8762           Info.getStdAllocatorCaller("allocate")) {
8763         // Inside a call to std::allocator::allocate and friends, we permit
8764         // casting from void* back to cv1 T* for a pointer that points to a
8765         // cv2 T.
8766       } else {
8767         Result.Designator.setInvalid();
8768         if (SubExpr->getType()->isVoidPointerType())
8769           CCEDiag(E, diag::note_constexpr_invalid_cast)
8770             << 3 << SubExpr->getType();
8771         else
8772           CCEDiag(E, diag::note_constexpr_invalid_cast) << 2;
8773       }
8774     }
8775     if (E->getCastKind() == CK_AddressSpaceConversion && Result.IsNullPtr)
8776       ZeroInitialization(E);
8777     return true;
8778 
8779   case CK_DerivedToBase:
8780   case CK_UncheckedDerivedToBase:
8781     if (!evaluatePointer(E->getSubExpr(), Result))
8782       return false;
8783     if (!Result.Base && Result.Offset.isZero())
8784       return true;
8785 
8786     // Now figure out the necessary offset to add to the base LV to get from
8787     // the derived class to the base class.
8788     return HandleLValueBasePath(Info, E, E->getSubExpr()->getType()->
8789                                   castAs<PointerType>()->getPointeeType(),
8790                                 Result);
8791 
8792   case CK_BaseToDerived:
8793     if (!Visit(E->getSubExpr()))
8794       return false;
8795     if (!Result.Base && Result.Offset.isZero())
8796       return true;
8797     return HandleBaseToDerivedCast(Info, E, Result);
8798 
8799   case CK_Dynamic:
8800     if (!Visit(E->getSubExpr()))
8801       return false;
8802     return HandleDynamicCast(Info, cast<ExplicitCastExpr>(E), Result);
8803 
8804   case CK_NullToPointer:
8805     VisitIgnoredValue(E->getSubExpr());
8806     return ZeroInitialization(E);
8807 
8808   case CK_IntegralToPointer: {
8809     CCEDiag(E, diag::note_constexpr_invalid_cast) << 2;
8810 
8811     APValue Value;
8812     if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
8813       break;
8814 
8815     if (Value.isInt()) {
8816       unsigned Size = Info.Ctx.getTypeSize(E->getType());
8817       uint64_t N = Value.getInt().extOrTrunc(Size).getZExtValue();
8818       Result.Base = (Expr*)nullptr;
8819       Result.InvalidBase = false;
8820       Result.Offset = CharUnits::fromQuantity(N);
8821       Result.Designator.setInvalid();
8822       Result.IsNullPtr = false;
8823       return true;
8824     } else {
8825       // Cast is of an lvalue, no need to change value.
8826       Result.setFrom(Info.Ctx, Value);
8827       return true;
8828     }
8829   }
8830 
8831   case CK_ArrayToPointerDecay: {
8832     if (SubExpr->isGLValue()) {
8833       if (!evaluateLValue(SubExpr, Result))
8834         return false;
8835     } else {
8836       APValue &Value = Info.CurrentCall->createTemporary(
8837           SubExpr, SubExpr->getType(), ScopeKind::FullExpression, Result);
8838       if (!EvaluateInPlace(Value, Info, Result, SubExpr))
8839         return false;
8840     }
8841     // The result is a pointer to the first element of the array.
8842     auto *AT = Info.Ctx.getAsArrayType(SubExpr->getType());
8843     if (auto *CAT = dyn_cast<ConstantArrayType>(AT))
8844       Result.addArray(Info, E, CAT);
8845     else
8846       Result.addUnsizedArray(Info, E, AT->getElementType());
8847     return true;
8848   }
8849 
8850   case CK_FunctionToPointerDecay:
8851     return evaluateLValue(SubExpr, Result);
8852 
8853   case CK_LValueToRValue: {
8854     LValue LVal;
8855     if (!evaluateLValue(E->getSubExpr(), LVal))
8856       return false;
8857 
8858     APValue RVal;
8859     // Note, we use the subexpression's type in order to retain cv-qualifiers.
8860     if (!handleLValueToRValueConversion(Info, E, E->getSubExpr()->getType(),
8861                                         LVal, RVal))
8862       return InvalidBaseOK &&
8863              evaluateLValueAsAllocSize(Info, LVal.Base, Result);
8864     return Success(RVal, E);
8865   }
8866   }
8867 
8868   return ExprEvaluatorBaseTy::VisitCastExpr(E);
8869 }
8870 
GetAlignOfType(EvalInfo & Info,QualType T,UnaryExprOrTypeTrait ExprKind)8871 static CharUnits GetAlignOfType(EvalInfo &Info, QualType T,
8872                                 UnaryExprOrTypeTrait ExprKind) {
8873   // C++ [expr.alignof]p3:
8874   //     When alignof is applied to a reference type, the result is the
8875   //     alignment of the referenced type.
8876   if (const ReferenceType *Ref = T->getAs<ReferenceType>())
8877     T = Ref->getPointeeType();
8878 
8879   if (T.getQualifiers().hasUnaligned())
8880     return CharUnits::One();
8881 
8882   const bool AlignOfReturnsPreferred =
8883       Info.Ctx.getLangOpts().getClangABICompat() <= LangOptions::ClangABI::Ver7;
8884 
8885   // __alignof is defined to return the preferred alignment.
8886   // Before 8, clang returned the preferred alignment for alignof and _Alignof
8887   // as well.
8888   if (ExprKind == UETT_PreferredAlignOf || AlignOfReturnsPreferred)
8889     return Info.Ctx.toCharUnitsFromBits(
8890       Info.Ctx.getPreferredTypeAlign(T.getTypePtr()));
8891   // alignof and _Alignof are defined to return the ABI alignment.
8892   else if (ExprKind == UETT_AlignOf)
8893     return Info.Ctx.getTypeAlignInChars(T.getTypePtr());
8894   else
8895     llvm_unreachable("GetAlignOfType on a non-alignment ExprKind");
8896 }
8897 
GetAlignOfExpr(EvalInfo & Info,const Expr * E,UnaryExprOrTypeTrait ExprKind)8898 static CharUnits GetAlignOfExpr(EvalInfo &Info, const Expr *E,
8899                                 UnaryExprOrTypeTrait ExprKind) {
8900   E = E->IgnoreParens();
8901 
8902   // The kinds of expressions that we have special-case logic here for
8903   // should be kept up to date with the special checks for those
8904   // expressions in Sema.
8905 
8906   // alignof decl is always accepted, even if it doesn't make sense: we default
8907   // to 1 in those cases.
8908   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
8909     return Info.Ctx.getDeclAlign(DRE->getDecl(),
8910                                  /*RefAsPointee*/true);
8911 
8912   if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
8913     return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
8914                                  /*RefAsPointee*/true);
8915 
8916   return GetAlignOfType(Info, E->getType(), ExprKind);
8917 }
8918 
getBaseAlignment(EvalInfo & Info,const LValue & Value)8919 static CharUnits getBaseAlignment(EvalInfo &Info, const LValue &Value) {
8920   if (const auto *VD = Value.Base.dyn_cast<const ValueDecl *>())
8921     return Info.Ctx.getDeclAlign(VD);
8922   if (const auto *E = Value.Base.dyn_cast<const Expr *>())
8923     return GetAlignOfExpr(Info, E, UETT_AlignOf);
8924   return GetAlignOfType(Info, Value.Base.getTypeInfoType(), UETT_AlignOf);
8925 }
8926 
8927 /// Evaluate the value of the alignment argument to __builtin_align_{up,down},
8928 /// __builtin_is_aligned and __builtin_assume_aligned.
getAlignmentArgument(const Expr * E,QualType ForType,EvalInfo & Info,APSInt & Alignment)8929 static bool getAlignmentArgument(const Expr *E, QualType ForType,
8930                                  EvalInfo &Info, APSInt &Alignment) {
8931   if (!EvaluateInteger(E, Alignment, Info))
8932     return false;
8933   if (Alignment < 0 || !Alignment.isPowerOf2()) {
8934     Info.FFDiag(E, diag::note_constexpr_invalid_alignment) << Alignment;
8935     return false;
8936   }
8937   unsigned SrcWidth = Info.Ctx.getIntWidth(ForType);
8938   APSInt MaxValue(APInt::getOneBitSet(SrcWidth, SrcWidth - 1));
8939   if (APSInt::compareValues(Alignment, MaxValue) > 0) {
8940     Info.FFDiag(E, diag::note_constexpr_alignment_too_big)
8941         << MaxValue << ForType << Alignment;
8942     return false;
8943   }
8944   // Ensure both alignment and source value have the same bit width so that we
8945   // don't assert when computing the resulting value.
8946   APSInt ExtAlignment =
8947       APSInt(Alignment.zextOrTrunc(SrcWidth), /*isUnsigned=*/true);
8948   assert(APSInt::compareValues(Alignment, ExtAlignment) == 0 &&
8949          "Alignment should not be changed by ext/trunc");
8950   Alignment = ExtAlignment;
8951   assert(Alignment.getBitWidth() == SrcWidth);
8952   return true;
8953 }
8954 
8955 // To be clear: this happily visits unsupported builtins. Better name welcomed.
visitNonBuiltinCallExpr(const CallExpr * E)8956 bool PointerExprEvaluator::visitNonBuiltinCallExpr(const CallExpr *E) {
8957   if (ExprEvaluatorBaseTy::VisitCallExpr(E))
8958     return true;
8959 
8960   if (!(InvalidBaseOK && getAllocSizeAttr(E)))
8961     return false;
8962 
8963   Result.setInvalid(E);
8964   QualType PointeeTy = E->getType()->castAs<PointerType>()->getPointeeType();
8965   Result.addUnsizedArray(Info, E, PointeeTy);
8966   return true;
8967 }
8968 
VisitCallExpr(const CallExpr * E)8969 bool PointerExprEvaluator::VisitCallExpr(const CallExpr *E) {
8970   if (IsStringLiteralCall(E))
8971     return Success(E);
8972 
8973   if (unsigned BuiltinOp = E->getBuiltinCallee())
8974     return VisitBuiltinCallExpr(E, BuiltinOp);
8975 
8976   return visitNonBuiltinCallExpr(E);
8977 }
8978 
8979 // Determine if T is a character type for which we guarantee that
8980 // sizeof(T) == 1.
isOneByteCharacterType(QualType T)8981 static bool isOneByteCharacterType(QualType T) {
8982   return T->isCharType() || T->isChar8Type();
8983 }
8984 
VisitBuiltinCallExpr(const CallExpr * E,unsigned BuiltinOp)8985 bool PointerExprEvaluator::VisitBuiltinCallExpr(const CallExpr *E,
8986                                                 unsigned BuiltinOp) {
8987   switch (BuiltinOp) {
8988   case Builtin::BI__builtin_addressof:
8989     return evaluateLValue(E->getArg(0), Result);
8990   case Builtin::BI__builtin_assume_aligned: {
8991     // We need to be very careful here because: if the pointer does not have the
8992     // asserted alignment, then the behavior is undefined, and undefined
8993     // behavior is non-constant.
8994     if (!evaluatePointer(E->getArg(0), Result))
8995       return false;
8996 
8997     LValue OffsetResult(Result);
8998     APSInt Alignment;
8999     if (!getAlignmentArgument(E->getArg(1), E->getArg(0)->getType(), Info,
9000                               Alignment))
9001       return false;
9002     CharUnits Align = CharUnits::fromQuantity(Alignment.getZExtValue());
9003 
9004     if (E->getNumArgs() > 2) {
9005       APSInt Offset;
9006       if (!EvaluateInteger(E->getArg(2), Offset, Info))
9007         return false;
9008 
9009       int64_t AdditionalOffset = -Offset.getZExtValue();
9010       OffsetResult.Offset += CharUnits::fromQuantity(AdditionalOffset);
9011     }
9012 
9013     // If there is a base object, then it must have the correct alignment.
9014     if (OffsetResult.Base) {
9015       CharUnits BaseAlignment = getBaseAlignment(Info, OffsetResult);
9016 
9017       if (BaseAlignment < Align) {
9018         Result.Designator.setInvalid();
9019         // FIXME: Add support to Diagnostic for long / long long.
9020         CCEDiag(E->getArg(0),
9021                 diag::note_constexpr_baa_insufficient_alignment) << 0
9022           << (unsigned)BaseAlignment.getQuantity()
9023           << (unsigned)Align.getQuantity();
9024         return false;
9025       }
9026     }
9027 
9028     // The offset must also have the correct alignment.
9029     if (OffsetResult.Offset.alignTo(Align) != OffsetResult.Offset) {
9030       Result.Designator.setInvalid();
9031 
9032       (OffsetResult.Base
9033            ? CCEDiag(E->getArg(0),
9034                      diag::note_constexpr_baa_insufficient_alignment) << 1
9035            : CCEDiag(E->getArg(0),
9036                      diag::note_constexpr_baa_value_insufficient_alignment))
9037         << (int)OffsetResult.Offset.getQuantity()
9038         << (unsigned)Align.getQuantity();
9039       return false;
9040     }
9041 
9042     return true;
9043   }
9044   case Builtin::BI__builtin_align_up:
9045   case Builtin::BI__builtin_align_down: {
9046     if (!evaluatePointer(E->getArg(0), Result))
9047       return false;
9048     APSInt Alignment;
9049     if (!getAlignmentArgument(E->getArg(1), E->getArg(0)->getType(), Info,
9050                               Alignment))
9051       return false;
9052     CharUnits BaseAlignment = getBaseAlignment(Info, Result);
9053     CharUnits PtrAlign = BaseAlignment.alignmentAtOffset(Result.Offset);
9054     // For align_up/align_down, we can return the same value if the alignment
9055     // is known to be greater or equal to the requested value.
9056     if (PtrAlign.getQuantity() >= Alignment)
9057       return true;
9058 
9059     // The alignment could be greater than the minimum at run-time, so we cannot
9060     // infer much about the resulting pointer value. One case is possible:
9061     // For `_Alignas(32) char buf[N]; __builtin_align_down(&buf[idx], 32)` we
9062     // can infer the correct index if the requested alignment is smaller than
9063     // the base alignment so we can perform the computation on the offset.
9064     if (BaseAlignment.getQuantity() >= Alignment) {
9065       assert(Alignment.getBitWidth() <= 64 &&
9066              "Cannot handle > 64-bit address-space");
9067       uint64_t Alignment64 = Alignment.getZExtValue();
9068       CharUnits NewOffset = CharUnits::fromQuantity(
9069           BuiltinOp == Builtin::BI__builtin_align_down
9070               ? llvm::alignDown(Result.Offset.getQuantity(), Alignment64)
9071               : llvm::alignTo(Result.Offset.getQuantity(), Alignment64));
9072       Result.adjustOffset(NewOffset - Result.Offset);
9073       // TODO: diagnose out-of-bounds values/only allow for arrays?
9074       return true;
9075     }
9076     // Otherwise, we cannot constant-evaluate the result.
9077     Info.FFDiag(E->getArg(0), diag::note_constexpr_alignment_adjust)
9078         << Alignment;
9079     return false;
9080   }
9081   case Builtin::BI__builtin_operator_new:
9082     return HandleOperatorNewCall(Info, E, Result);
9083   case Builtin::BI__builtin_launder:
9084     return evaluatePointer(E->getArg(0), Result);
9085   case Builtin::BIstrchr:
9086   case Builtin::BIwcschr:
9087   case Builtin::BImemchr:
9088   case Builtin::BIwmemchr:
9089     if (Info.getLangOpts().CPlusPlus11)
9090       Info.CCEDiag(E, diag::note_constexpr_invalid_function)
9091         << /*isConstexpr*/0 << /*isConstructor*/0
9092         << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'");
9093     else
9094       Info.CCEDiag(E, diag::note_invalid_subexpr_in_const_expr);
9095     LLVM_FALLTHROUGH;
9096   case Builtin::BI__builtin_strchr:
9097   case Builtin::BI__builtin_wcschr:
9098   case Builtin::BI__builtin_memchr:
9099   case Builtin::BI__builtin_char_memchr:
9100   case Builtin::BI__builtin_wmemchr: {
9101     if (!Visit(E->getArg(0)))
9102       return false;
9103     APSInt Desired;
9104     if (!EvaluateInteger(E->getArg(1), Desired, Info))
9105       return false;
9106     uint64_t MaxLength = uint64_t(-1);
9107     if (BuiltinOp != Builtin::BIstrchr &&
9108         BuiltinOp != Builtin::BIwcschr &&
9109         BuiltinOp != Builtin::BI__builtin_strchr &&
9110         BuiltinOp != Builtin::BI__builtin_wcschr) {
9111       APSInt N;
9112       if (!EvaluateInteger(E->getArg(2), N, Info))
9113         return false;
9114       MaxLength = N.getExtValue();
9115     }
9116     // We cannot find the value if there are no candidates to match against.
9117     if (MaxLength == 0u)
9118       return ZeroInitialization(E);
9119     if (!Result.checkNullPointerForFoldAccess(Info, E, AK_Read) ||
9120         Result.Designator.Invalid)
9121       return false;
9122     QualType CharTy = Result.Designator.getType(Info.Ctx);
9123     bool IsRawByte = BuiltinOp == Builtin::BImemchr ||
9124                      BuiltinOp == Builtin::BI__builtin_memchr;
9125     assert(IsRawByte ||
9126            Info.Ctx.hasSameUnqualifiedType(
9127                CharTy, E->getArg(0)->getType()->getPointeeType()));
9128     // Pointers to const void may point to objects of incomplete type.
9129     if (IsRawByte && CharTy->isIncompleteType()) {
9130       Info.FFDiag(E, diag::note_constexpr_ltor_incomplete_type) << CharTy;
9131       return false;
9132     }
9133     // Give up on byte-oriented matching against multibyte elements.
9134     // FIXME: We can compare the bytes in the correct order.
9135     if (IsRawByte && !isOneByteCharacterType(CharTy)) {
9136       Info.FFDiag(E, diag::note_constexpr_memchr_unsupported)
9137           << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'")
9138           << CharTy;
9139       return false;
9140     }
9141     // Figure out what value we're actually looking for (after converting to
9142     // the corresponding unsigned type if necessary).
9143     uint64_t DesiredVal;
9144     bool StopAtNull = false;
9145     switch (BuiltinOp) {
9146     case Builtin::BIstrchr:
9147     case Builtin::BI__builtin_strchr:
9148       // strchr compares directly to the passed integer, and therefore
9149       // always fails if given an int that is not a char.
9150       if (!APSInt::isSameValue(HandleIntToIntCast(Info, E, CharTy,
9151                                                   E->getArg(1)->getType(),
9152                                                   Desired),
9153                                Desired))
9154         return ZeroInitialization(E);
9155       StopAtNull = true;
9156       LLVM_FALLTHROUGH;
9157     case Builtin::BImemchr:
9158     case Builtin::BI__builtin_memchr:
9159     case Builtin::BI__builtin_char_memchr:
9160       // memchr compares by converting both sides to unsigned char. That's also
9161       // correct for strchr if we get this far (to cope with plain char being
9162       // unsigned in the strchr case).
9163       DesiredVal = Desired.trunc(Info.Ctx.getCharWidth()).getZExtValue();
9164       break;
9165 
9166     case Builtin::BIwcschr:
9167     case Builtin::BI__builtin_wcschr:
9168       StopAtNull = true;
9169       LLVM_FALLTHROUGH;
9170     case Builtin::BIwmemchr:
9171     case Builtin::BI__builtin_wmemchr:
9172       // wcschr and wmemchr are given a wchar_t to look for. Just use it.
9173       DesiredVal = Desired.getZExtValue();
9174       break;
9175     }
9176 
9177     for (; MaxLength; --MaxLength) {
9178       APValue Char;
9179       if (!handleLValueToRValueConversion(Info, E, CharTy, Result, Char) ||
9180           !Char.isInt())
9181         return false;
9182       if (Char.getInt().getZExtValue() == DesiredVal)
9183         return true;
9184       if (StopAtNull && !Char.getInt())
9185         break;
9186       if (!HandleLValueArrayAdjustment(Info, E, Result, CharTy, 1))
9187         return false;
9188     }
9189     // Not found: return nullptr.
9190     return ZeroInitialization(E);
9191   }
9192 
9193   case Builtin::BImemcpy:
9194   case Builtin::BImemmove:
9195   case Builtin::BIwmemcpy:
9196   case Builtin::BIwmemmove:
9197     if (Info.getLangOpts().CPlusPlus11)
9198       Info.CCEDiag(E, diag::note_constexpr_invalid_function)
9199         << /*isConstexpr*/0 << /*isConstructor*/0
9200         << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'");
9201     else
9202       Info.CCEDiag(E, diag::note_invalid_subexpr_in_const_expr);
9203     LLVM_FALLTHROUGH;
9204   case Builtin::BI__builtin_memcpy:
9205   case Builtin::BI__builtin_memmove:
9206   case Builtin::BI__builtin_wmemcpy:
9207   case Builtin::BI__builtin_wmemmove: {
9208     bool WChar = BuiltinOp == Builtin::BIwmemcpy ||
9209                  BuiltinOp == Builtin::BIwmemmove ||
9210                  BuiltinOp == Builtin::BI__builtin_wmemcpy ||
9211                  BuiltinOp == Builtin::BI__builtin_wmemmove;
9212     bool Move = BuiltinOp == Builtin::BImemmove ||
9213                 BuiltinOp == Builtin::BIwmemmove ||
9214                 BuiltinOp == Builtin::BI__builtin_memmove ||
9215                 BuiltinOp == Builtin::BI__builtin_wmemmove;
9216 
9217     // The result of mem* is the first argument.
9218     if (!Visit(E->getArg(0)))
9219       return false;
9220     LValue Dest = Result;
9221 
9222     LValue Src;
9223     if (!EvaluatePointer(E->getArg(1), Src, Info))
9224       return false;
9225 
9226     APSInt N;
9227     if (!EvaluateInteger(E->getArg(2), N, Info))
9228       return false;
9229     assert(!N.isSigned() && "memcpy and friends take an unsigned size");
9230 
9231     // If the size is zero, we treat this as always being a valid no-op.
9232     // (Even if one of the src and dest pointers is null.)
9233     if (!N)
9234       return true;
9235 
9236     // Otherwise, if either of the operands is null, we can't proceed. Don't
9237     // try to determine the type of the copied objects, because there aren't
9238     // any.
9239     if (!Src.Base || !Dest.Base) {
9240       APValue Val;
9241       (!Src.Base ? Src : Dest).moveInto(Val);
9242       Info.FFDiag(E, diag::note_constexpr_memcpy_null)
9243           << Move << WChar << !!Src.Base
9244           << Val.getAsString(Info.Ctx, E->getArg(0)->getType());
9245       return false;
9246     }
9247     if (Src.Designator.Invalid || Dest.Designator.Invalid)
9248       return false;
9249 
9250     // We require that Src and Dest are both pointers to arrays of
9251     // trivially-copyable type. (For the wide version, the designator will be
9252     // invalid if the designated object is not a wchar_t.)
9253     QualType T = Dest.Designator.getType(Info.Ctx);
9254     QualType SrcT = Src.Designator.getType(Info.Ctx);
9255     if (!Info.Ctx.hasSameUnqualifiedType(T, SrcT)) {
9256       // FIXME: Consider using our bit_cast implementation to support this.
9257       Info.FFDiag(E, diag::note_constexpr_memcpy_type_pun) << Move << SrcT << T;
9258       return false;
9259     }
9260     if (T->isIncompleteType()) {
9261       Info.FFDiag(E, diag::note_constexpr_memcpy_incomplete_type) << Move << T;
9262       return false;
9263     }
9264     if (!T.isTriviallyCopyableType(Info.Ctx)) {
9265       Info.FFDiag(E, diag::note_constexpr_memcpy_nontrivial) << Move << T;
9266       return false;
9267     }
9268 
9269     // Figure out how many T's we're copying.
9270     uint64_t TSize = Info.Ctx.getTypeSizeInChars(T).getQuantity();
9271     if (!WChar) {
9272       uint64_t Remainder;
9273       llvm::APInt OrigN = N;
9274       llvm::APInt::udivrem(OrigN, TSize, N, Remainder);
9275       if (Remainder) {
9276         Info.FFDiag(E, diag::note_constexpr_memcpy_unsupported)
9277             << Move << WChar << 0 << T << toString(OrigN, 10, /*Signed*/false)
9278             << (unsigned)TSize;
9279         return false;
9280       }
9281     }
9282 
9283     // Check that the copying will remain within the arrays, just so that we
9284     // can give a more meaningful diagnostic. This implicitly also checks that
9285     // N fits into 64 bits.
9286     uint64_t RemainingSrcSize = Src.Designator.validIndexAdjustments().second;
9287     uint64_t RemainingDestSize = Dest.Designator.validIndexAdjustments().second;
9288     if (N.ugt(RemainingSrcSize) || N.ugt(RemainingDestSize)) {
9289       Info.FFDiag(E, diag::note_constexpr_memcpy_unsupported)
9290           << Move << WChar << (N.ugt(RemainingSrcSize) ? 1 : 2) << T
9291           << toString(N, 10, /*Signed*/false);
9292       return false;
9293     }
9294     uint64_t NElems = N.getZExtValue();
9295     uint64_t NBytes = NElems * TSize;
9296 
9297     // Check for overlap.
9298     int Direction = 1;
9299     if (HasSameBase(Src, Dest)) {
9300       uint64_t SrcOffset = Src.getLValueOffset().getQuantity();
9301       uint64_t DestOffset = Dest.getLValueOffset().getQuantity();
9302       if (DestOffset >= SrcOffset && DestOffset - SrcOffset < NBytes) {
9303         // Dest is inside the source region.
9304         if (!Move) {
9305           Info.FFDiag(E, diag::note_constexpr_memcpy_overlap) << WChar;
9306           return false;
9307         }
9308         // For memmove and friends, copy backwards.
9309         if (!HandleLValueArrayAdjustment(Info, E, Src, T, NElems - 1) ||
9310             !HandleLValueArrayAdjustment(Info, E, Dest, T, NElems - 1))
9311           return false;
9312         Direction = -1;
9313       } else if (!Move && SrcOffset >= DestOffset &&
9314                  SrcOffset - DestOffset < NBytes) {
9315         // Src is inside the destination region for memcpy: invalid.
9316         Info.FFDiag(E, diag::note_constexpr_memcpy_overlap) << WChar;
9317         return false;
9318       }
9319     }
9320 
9321     while (true) {
9322       APValue Val;
9323       // FIXME: Set WantObjectRepresentation to true if we're copying a
9324       // char-like type?
9325       if (!handleLValueToRValueConversion(Info, E, T, Src, Val) ||
9326           !handleAssignment(Info, E, Dest, T, Val))
9327         return false;
9328       // Do not iterate past the last element; if we're copying backwards, that
9329       // might take us off the start of the array.
9330       if (--NElems == 0)
9331         return true;
9332       if (!HandleLValueArrayAdjustment(Info, E, Src, T, Direction) ||
9333           !HandleLValueArrayAdjustment(Info, E, Dest, T, Direction))
9334         return false;
9335     }
9336   }
9337 
9338   default:
9339     break;
9340   }
9341 
9342   return visitNonBuiltinCallExpr(E);
9343 }
9344 
9345 static bool EvaluateArrayNewInitList(EvalInfo &Info, LValue &This,
9346                                      APValue &Result, const InitListExpr *ILE,
9347                                      QualType AllocType);
9348 static bool EvaluateArrayNewConstructExpr(EvalInfo &Info, LValue &This,
9349                                           APValue &Result,
9350                                           const CXXConstructExpr *CCE,
9351                                           QualType AllocType);
9352 
VisitCXXNewExpr(const CXXNewExpr * E)9353 bool PointerExprEvaluator::VisitCXXNewExpr(const CXXNewExpr *E) {
9354   if (!Info.getLangOpts().CPlusPlus20)
9355     Info.CCEDiag(E, diag::note_constexpr_new);
9356 
9357   // We cannot speculatively evaluate a delete expression.
9358   if (Info.SpeculativeEvaluationDepth)
9359     return false;
9360 
9361   FunctionDecl *OperatorNew = E->getOperatorNew();
9362 
9363   bool IsNothrow = false;
9364   bool IsPlacement = false;
9365   if (OperatorNew->isReservedGlobalPlacementOperator() &&
9366       Info.CurrentCall->isStdFunction() && !E->isArray()) {
9367     // FIXME Support array placement new.
9368     assert(E->getNumPlacementArgs() == 1);
9369     if (!EvaluatePointer(E->getPlacementArg(0), Result, Info))
9370       return false;
9371     if (Result.Designator.Invalid)
9372       return false;
9373     IsPlacement = true;
9374   } else if (!OperatorNew->isReplaceableGlobalAllocationFunction()) {
9375     Info.FFDiag(E, diag::note_constexpr_new_non_replaceable)
9376         << isa<CXXMethodDecl>(OperatorNew) << OperatorNew;
9377     return false;
9378   } else if (E->getNumPlacementArgs()) {
9379     // The only new-placement list we support is of the form (std::nothrow).
9380     //
9381     // FIXME: There is no restriction on this, but it's not clear that any
9382     // other form makes any sense. We get here for cases such as:
9383     //
9384     //   new (std::align_val_t{N}) X(int)
9385     //
9386     // (which should presumably be valid only if N is a multiple of
9387     // alignof(int), and in any case can't be deallocated unless N is
9388     // alignof(X) and X has new-extended alignment).
9389     if (E->getNumPlacementArgs() != 1 ||
9390         !E->getPlacementArg(0)->getType()->isNothrowT())
9391       return Error(E, diag::note_constexpr_new_placement);
9392 
9393     LValue Nothrow;
9394     if (!EvaluateLValue(E->getPlacementArg(0), Nothrow, Info))
9395       return false;
9396     IsNothrow = true;
9397   }
9398 
9399   const Expr *Init = E->getInitializer();
9400   const InitListExpr *ResizedArrayILE = nullptr;
9401   const CXXConstructExpr *ResizedArrayCCE = nullptr;
9402   bool ValueInit = false;
9403 
9404   QualType AllocType = E->getAllocatedType();
9405   if (Optional<const Expr*> ArraySize = E->getArraySize()) {
9406     const Expr *Stripped = *ArraySize;
9407     for (; auto *ICE = dyn_cast<ImplicitCastExpr>(Stripped);
9408          Stripped = ICE->getSubExpr())
9409       if (ICE->getCastKind() != CK_NoOp &&
9410           ICE->getCastKind() != CK_IntegralCast)
9411         break;
9412 
9413     llvm::APSInt ArrayBound;
9414     if (!EvaluateInteger(Stripped, ArrayBound, Info))
9415       return false;
9416 
9417     // C++ [expr.new]p9:
9418     //   The expression is erroneous if:
9419     //   -- [...] its value before converting to size_t [or] applying the
9420     //      second standard conversion sequence is less than zero
9421     if (ArrayBound.isSigned() && ArrayBound.isNegative()) {
9422       if (IsNothrow)
9423         return ZeroInitialization(E);
9424 
9425       Info.FFDiag(*ArraySize, diag::note_constexpr_new_negative)
9426           << ArrayBound << (*ArraySize)->getSourceRange();
9427       return false;
9428     }
9429 
9430     //   -- its value is such that the size of the allocated object would
9431     //      exceed the implementation-defined limit
9432     if (ConstantArrayType::getNumAddressingBits(Info.Ctx, AllocType,
9433                                                 ArrayBound) >
9434         ConstantArrayType::getMaxSizeBits(Info.Ctx)) {
9435       if (IsNothrow)
9436         return ZeroInitialization(E);
9437 
9438       Info.FFDiag(*ArraySize, diag::note_constexpr_new_too_large)
9439         << ArrayBound << (*ArraySize)->getSourceRange();
9440       return false;
9441     }
9442 
9443     //   -- the new-initializer is a braced-init-list and the number of
9444     //      array elements for which initializers are provided [...]
9445     //      exceeds the number of elements to initialize
9446     if (!Init) {
9447       // No initialization is performed.
9448     } else if (isa<CXXScalarValueInitExpr>(Init) ||
9449                isa<ImplicitValueInitExpr>(Init)) {
9450       ValueInit = true;
9451     } else if (auto *CCE = dyn_cast<CXXConstructExpr>(Init)) {
9452       ResizedArrayCCE = CCE;
9453     } else {
9454       auto *CAT = Info.Ctx.getAsConstantArrayType(Init->getType());
9455       assert(CAT && "unexpected type for array initializer");
9456 
9457       unsigned Bits =
9458           std::max(CAT->getSize().getBitWidth(), ArrayBound.getBitWidth());
9459       llvm::APInt InitBound = CAT->getSize().zextOrSelf(Bits);
9460       llvm::APInt AllocBound = ArrayBound.zextOrSelf(Bits);
9461       if (InitBound.ugt(AllocBound)) {
9462         if (IsNothrow)
9463           return ZeroInitialization(E);
9464 
9465         Info.FFDiag(*ArraySize, diag::note_constexpr_new_too_small)
9466             << toString(AllocBound, 10, /*Signed=*/false)
9467             << toString(InitBound, 10, /*Signed=*/false)
9468             << (*ArraySize)->getSourceRange();
9469         return false;
9470       }
9471 
9472       // If the sizes differ, we must have an initializer list, and we need
9473       // special handling for this case when we initialize.
9474       if (InitBound != AllocBound)
9475         ResizedArrayILE = cast<InitListExpr>(Init);
9476     }
9477 
9478     AllocType = Info.Ctx.getConstantArrayType(AllocType, ArrayBound, nullptr,
9479                                               ArrayType::Normal, 0);
9480   } else {
9481     assert(!AllocType->isArrayType() &&
9482            "array allocation with non-array new");
9483   }
9484 
9485   APValue *Val;
9486   if (IsPlacement) {
9487     AccessKinds AK = AK_Construct;
9488     struct FindObjectHandler {
9489       EvalInfo &Info;
9490       const Expr *E;
9491       QualType AllocType;
9492       const AccessKinds AccessKind;
9493       APValue *Value;
9494 
9495       typedef bool result_type;
9496       bool failed() { return false; }
9497       bool found(APValue &Subobj, QualType SubobjType) {
9498         // FIXME: Reject the cases where [basic.life]p8 would not permit the
9499         // old name of the object to be used to name the new object.
9500         if (!Info.Ctx.hasSameUnqualifiedType(SubobjType, AllocType)) {
9501           Info.FFDiag(E, diag::note_constexpr_placement_new_wrong_type) <<
9502             SubobjType << AllocType;
9503           return false;
9504         }
9505         Value = &Subobj;
9506         return true;
9507       }
9508       bool found(APSInt &Value, QualType SubobjType) {
9509         Info.FFDiag(E, diag::note_constexpr_construct_complex_elem);
9510         return false;
9511       }
9512       bool found(APFloat &Value, QualType SubobjType) {
9513         Info.FFDiag(E, diag::note_constexpr_construct_complex_elem);
9514         return false;
9515       }
9516     } Handler = {Info, E, AllocType, AK, nullptr};
9517 
9518     CompleteObject Obj = findCompleteObject(Info, E, AK, Result, AllocType);
9519     if (!Obj || !findSubobject(Info, E, Obj, Result.Designator, Handler))
9520       return false;
9521 
9522     Val = Handler.Value;
9523 
9524     // [basic.life]p1:
9525     //   The lifetime of an object o of type T ends when [...] the storage
9526     //   which the object occupies is [...] reused by an object that is not
9527     //   nested within o (6.6.2).
9528     *Val = APValue();
9529   } else {
9530     // Perform the allocation and obtain a pointer to the resulting object.
9531     Val = Info.createHeapAlloc(E, AllocType, Result);
9532     if (!Val)
9533       return false;
9534   }
9535 
9536   if (ValueInit) {
9537     ImplicitValueInitExpr VIE(AllocType);
9538     if (!EvaluateInPlace(*Val, Info, Result, &VIE))
9539       return false;
9540   } else if (ResizedArrayILE) {
9541     if (!EvaluateArrayNewInitList(Info, Result, *Val, ResizedArrayILE,
9542                                   AllocType))
9543       return false;
9544   } else if (ResizedArrayCCE) {
9545     if (!EvaluateArrayNewConstructExpr(Info, Result, *Val, ResizedArrayCCE,
9546                                        AllocType))
9547       return false;
9548   } else if (Init) {
9549     if (!EvaluateInPlace(*Val, Info, Result, Init))
9550       return false;
9551   } else if (!getDefaultInitValue(AllocType, *Val)) {
9552     return false;
9553   }
9554 
9555   // Array new returns a pointer to the first element, not a pointer to the
9556   // array.
9557   if (auto *AT = AllocType->getAsArrayTypeUnsafe())
9558     Result.addArray(Info, E, cast<ConstantArrayType>(AT));
9559 
9560   return true;
9561 }
9562 //===----------------------------------------------------------------------===//
9563 // Member Pointer Evaluation
9564 //===----------------------------------------------------------------------===//
9565 
9566 namespace {
9567 class MemberPointerExprEvaluator
9568   : public ExprEvaluatorBase<MemberPointerExprEvaluator> {
9569   MemberPtr &Result;
9570 
Success(const ValueDecl * D)9571   bool Success(const ValueDecl *D) {
9572     Result = MemberPtr(D);
9573     return true;
9574   }
9575 public:
9576 
MemberPointerExprEvaluator(EvalInfo & Info,MemberPtr & Result)9577   MemberPointerExprEvaluator(EvalInfo &Info, MemberPtr &Result)
9578     : ExprEvaluatorBaseTy(Info), Result(Result) {}
9579 
Success(const APValue & V,const Expr * E)9580   bool Success(const APValue &V, const Expr *E) {
9581     Result.setFrom(V);
9582     return true;
9583   }
ZeroInitialization(const Expr * E)9584   bool ZeroInitialization(const Expr *E) {
9585     return Success((const ValueDecl*)nullptr);
9586   }
9587 
9588   bool VisitCastExpr(const CastExpr *E);
9589   bool VisitUnaryAddrOf(const UnaryOperator *E);
9590 };
9591 } // end anonymous namespace
9592 
EvaluateMemberPointer(const Expr * E,MemberPtr & Result,EvalInfo & Info)9593 static bool EvaluateMemberPointer(const Expr *E, MemberPtr &Result,
9594                                   EvalInfo &Info) {
9595   assert(!E->isValueDependent());
9596   assert(E->isPRValue() && E->getType()->isMemberPointerType());
9597   return MemberPointerExprEvaluator(Info, Result).Visit(E);
9598 }
9599 
VisitCastExpr(const CastExpr * E)9600 bool MemberPointerExprEvaluator::VisitCastExpr(const CastExpr *E) {
9601   switch (E->getCastKind()) {
9602   default:
9603     return ExprEvaluatorBaseTy::VisitCastExpr(E);
9604 
9605   case CK_NullToMemberPointer:
9606     VisitIgnoredValue(E->getSubExpr());
9607     return ZeroInitialization(E);
9608 
9609   case CK_BaseToDerivedMemberPointer: {
9610     if (!Visit(E->getSubExpr()))
9611       return false;
9612     if (E->path_empty())
9613       return true;
9614     // Base-to-derived member pointer casts store the path in derived-to-base
9615     // order, so iterate backwards. The CXXBaseSpecifier also provides us with
9616     // the wrong end of the derived->base arc, so stagger the path by one class.
9617     typedef std::reverse_iterator<CastExpr::path_const_iterator> ReverseIter;
9618     for (ReverseIter PathI(E->path_end() - 1), PathE(E->path_begin());
9619          PathI != PathE; ++PathI) {
9620       assert(!(*PathI)->isVirtual() && "memptr cast through vbase");
9621       const CXXRecordDecl *Derived = (*PathI)->getType()->getAsCXXRecordDecl();
9622       if (!Result.castToDerived(Derived))
9623         return Error(E);
9624     }
9625     const Type *FinalTy = E->getType()->castAs<MemberPointerType>()->getClass();
9626     if (!Result.castToDerived(FinalTy->getAsCXXRecordDecl()))
9627       return Error(E);
9628     return true;
9629   }
9630 
9631   case CK_DerivedToBaseMemberPointer:
9632     if (!Visit(E->getSubExpr()))
9633       return false;
9634     for (CastExpr::path_const_iterator PathI = E->path_begin(),
9635          PathE = E->path_end(); PathI != PathE; ++PathI) {
9636       assert(!(*PathI)->isVirtual() && "memptr cast through vbase");
9637       const CXXRecordDecl *Base = (*PathI)->getType()->getAsCXXRecordDecl();
9638       if (!Result.castToBase(Base))
9639         return Error(E);
9640     }
9641     return true;
9642   }
9643 }
9644 
VisitUnaryAddrOf(const UnaryOperator * E)9645 bool MemberPointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
9646   // C++11 [expr.unary.op]p3 has very strict rules on how the address of a
9647   // member can be formed.
9648   return Success(cast<DeclRefExpr>(E->getSubExpr())->getDecl());
9649 }
9650 
9651 //===----------------------------------------------------------------------===//
9652 // Record Evaluation
9653 //===----------------------------------------------------------------------===//
9654 
9655 namespace {
9656   class RecordExprEvaluator
9657   : public ExprEvaluatorBase<RecordExprEvaluator> {
9658     const LValue &This;
9659     APValue &Result;
9660   public:
9661 
RecordExprEvaluator(EvalInfo & info,const LValue & This,APValue & Result)9662     RecordExprEvaluator(EvalInfo &info, const LValue &This, APValue &Result)
9663       : ExprEvaluatorBaseTy(info), This(This), Result(Result) {}
9664 
Success(const APValue & V,const Expr * E)9665     bool Success(const APValue &V, const Expr *E) {
9666       Result = V;
9667       return true;
9668     }
ZeroInitialization(const Expr * E)9669     bool ZeroInitialization(const Expr *E) {
9670       return ZeroInitialization(E, E->getType());
9671     }
9672     bool ZeroInitialization(const Expr *E, QualType T);
9673 
VisitCallExpr(const CallExpr * E)9674     bool VisitCallExpr(const CallExpr *E) {
9675       return handleCallExpr(E, Result, &This);
9676     }
9677     bool VisitCastExpr(const CastExpr *E);
9678     bool VisitInitListExpr(const InitListExpr *E);
VisitCXXConstructExpr(const CXXConstructExpr * E)9679     bool VisitCXXConstructExpr(const CXXConstructExpr *E) {
9680       return VisitCXXConstructExpr(E, E->getType());
9681     }
9682     bool VisitLambdaExpr(const LambdaExpr *E);
9683     bool VisitCXXInheritedCtorInitExpr(const CXXInheritedCtorInitExpr *E);
9684     bool VisitCXXConstructExpr(const CXXConstructExpr *E, QualType T);
9685     bool VisitCXXStdInitializerListExpr(const CXXStdInitializerListExpr *E);
9686     bool VisitBinCmp(const BinaryOperator *E);
9687   };
9688 }
9689 
9690 /// Perform zero-initialization on an object of non-union class type.
9691 /// C++11 [dcl.init]p5:
9692 ///  To zero-initialize an object or reference of type T means:
9693 ///    [...]
9694 ///    -- if T is a (possibly cv-qualified) non-union class type,
9695 ///       each non-static data member and each base-class subobject is
9696 ///       zero-initialized
HandleClassZeroInitialization(EvalInfo & Info,const Expr * E,const RecordDecl * RD,const LValue & This,APValue & Result)9697 static bool HandleClassZeroInitialization(EvalInfo &Info, const Expr *E,
9698                                           const RecordDecl *RD,
9699                                           const LValue &This, APValue &Result) {
9700   assert(!RD->isUnion() && "Expected non-union class type");
9701   const CXXRecordDecl *CD = dyn_cast<CXXRecordDecl>(RD);
9702   Result = APValue(APValue::UninitStruct(), CD ? CD->getNumBases() : 0,
9703                    std::distance(RD->field_begin(), RD->field_end()));
9704 
9705   if (RD->isInvalidDecl()) return false;
9706   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
9707 
9708   if (CD) {
9709     unsigned Index = 0;
9710     for (CXXRecordDecl::base_class_const_iterator I = CD->bases_begin(),
9711            End = CD->bases_end(); I != End; ++I, ++Index) {
9712       const CXXRecordDecl *Base = I->getType()->getAsCXXRecordDecl();
9713       LValue Subobject = This;
9714       if (!HandleLValueDirectBase(Info, E, Subobject, CD, Base, &Layout))
9715         return false;
9716       if (!HandleClassZeroInitialization(Info, E, Base, Subobject,
9717                                          Result.getStructBase(Index)))
9718         return false;
9719     }
9720   }
9721 
9722   for (const auto *I : RD->fields()) {
9723     // -- if T is a reference type, no initialization is performed.
9724     if (I->isUnnamedBitfield() || I->getType()->isReferenceType())
9725       continue;
9726 
9727     LValue Subobject = This;
9728     if (!HandleLValueMember(Info, E, Subobject, I, &Layout))
9729       return false;
9730 
9731     ImplicitValueInitExpr VIE(I->getType());
9732     if (!EvaluateInPlace(
9733           Result.getStructField(I->getFieldIndex()), Info, Subobject, &VIE))
9734       return false;
9735   }
9736 
9737   return true;
9738 }
9739 
ZeroInitialization(const Expr * E,QualType T)9740 bool RecordExprEvaluator::ZeroInitialization(const Expr *E, QualType T) {
9741   const RecordDecl *RD = T->castAs<RecordType>()->getDecl();
9742   if (RD->isInvalidDecl()) return false;
9743   if (RD->isUnion()) {
9744     // C++11 [dcl.init]p5: If T is a (possibly cv-qualified) union type, the
9745     // object's first non-static named data member is zero-initialized
9746     RecordDecl::field_iterator I = RD->field_begin();
9747     while (I != RD->field_end() && (*I)->isUnnamedBitfield())
9748       ++I;
9749     if (I == RD->field_end()) {
9750       Result = APValue((const FieldDecl*)nullptr);
9751       return true;
9752     }
9753 
9754     LValue Subobject = This;
9755     if (!HandleLValueMember(Info, E, Subobject, *I))
9756       return false;
9757     Result = APValue(*I);
9758     ImplicitValueInitExpr VIE(I->getType());
9759     return EvaluateInPlace(Result.getUnionValue(), Info, Subobject, &VIE);
9760   }
9761 
9762   if (isa<CXXRecordDecl>(RD) && cast<CXXRecordDecl>(RD)->getNumVBases()) {
9763     Info.FFDiag(E, diag::note_constexpr_virtual_base) << RD;
9764     return false;
9765   }
9766 
9767   return HandleClassZeroInitialization(Info, E, RD, This, Result);
9768 }
9769 
VisitCastExpr(const CastExpr * E)9770 bool RecordExprEvaluator::VisitCastExpr(const CastExpr *E) {
9771   switch (E->getCastKind()) {
9772   default:
9773     return ExprEvaluatorBaseTy::VisitCastExpr(E);
9774 
9775   case CK_ConstructorConversion:
9776     return Visit(E->getSubExpr());
9777 
9778   case CK_DerivedToBase:
9779   case CK_UncheckedDerivedToBase: {
9780     APValue DerivedObject;
9781     if (!Evaluate(DerivedObject, Info, E->getSubExpr()))
9782       return false;
9783     if (!DerivedObject.isStruct())
9784       return Error(E->getSubExpr());
9785 
9786     // Derived-to-base rvalue conversion: just slice off the derived part.
9787     APValue *Value = &DerivedObject;
9788     const CXXRecordDecl *RD = E->getSubExpr()->getType()->getAsCXXRecordDecl();
9789     for (CastExpr::path_const_iterator PathI = E->path_begin(),
9790          PathE = E->path_end(); PathI != PathE; ++PathI) {
9791       assert(!(*PathI)->isVirtual() && "record rvalue with virtual base");
9792       const CXXRecordDecl *Base = (*PathI)->getType()->getAsCXXRecordDecl();
9793       Value = &Value->getStructBase(getBaseIndex(RD, Base));
9794       RD = Base;
9795     }
9796     Result = *Value;
9797     return true;
9798   }
9799   }
9800 }
9801 
VisitInitListExpr(const InitListExpr * E)9802 bool RecordExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
9803   if (E->isTransparent())
9804     return Visit(E->getInit(0));
9805 
9806   const RecordDecl *RD = E->getType()->castAs<RecordType>()->getDecl();
9807   if (RD->isInvalidDecl()) return false;
9808   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(RD);
9809   auto *CXXRD = dyn_cast<CXXRecordDecl>(RD);
9810 
9811   EvalInfo::EvaluatingConstructorRAII EvalObj(
9812       Info,
9813       ObjectUnderConstruction{This.getLValueBase(), This.Designator.Entries},
9814       CXXRD && CXXRD->getNumBases());
9815 
9816   if (RD->isUnion()) {
9817     const FieldDecl *Field = E->getInitializedFieldInUnion();
9818     Result = APValue(Field);
9819     if (!Field)
9820       return true;
9821 
9822     // If the initializer list for a union does not contain any elements, the
9823     // first element of the union is value-initialized.
9824     // FIXME: The element should be initialized from an initializer list.
9825     //        Is this difference ever observable for initializer lists which
9826     //        we don't build?
9827     ImplicitValueInitExpr VIE(Field->getType());
9828     const Expr *InitExpr = E->getNumInits() ? E->getInit(0) : &VIE;
9829 
9830     LValue Subobject = This;
9831     if (!HandleLValueMember(Info, InitExpr, Subobject, Field, &Layout))
9832       return false;
9833 
9834     // Temporarily override This, in case there's a CXXDefaultInitExpr in here.
9835     ThisOverrideRAII ThisOverride(*Info.CurrentCall, &This,
9836                                   isa<CXXDefaultInitExpr>(InitExpr));
9837 
9838     if (EvaluateInPlace(Result.getUnionValue(), Info, Subobject, InitExpr)) {
9839       if (Field->isBitField())
9840         return truncateBitfieldValue(Info, InitExpr, Result.getUnionValue(),
9841                                      Field);
9842       return true;
9843     }
9844 
9845     return false;
9846   }
9847 
9848   if (!Result.hasValue())
9849     Result = APValue(APValue::UninitStruct(), CXXRD ? CXXRD->getNumBases() : 0,
9850                      std::distance(RD->field_begin(), RD->field_end()));
9851   unsigned ElementNo = 0;
9852   bool Success = true;
9853 
9854   // Initialize base classes.
9855   if (CXXRD && CXXRD->getNumBases()) {
9856     for (const auto &Base : CXXRD->bases()) {
9857       assert(ElementNo < E->getNumInits() && "missing init for base class");
9858       const Expr *Init = E->getInit(ElementNo);
9859 
9860       LValue Subobject = This;
9861       if (!HandleLValueBase(Info, Init, Subobject, CXXRD, &Base))
9862         return false;
9863 
9864       APValue &FieldVal = Result.getStructBase(ElementNo);
9865       if (!EvaluateInPlace(FieldVal, Info, Subobject, Init)) {
9866         if (!Info.noteFailure())
9867           return false;
9868         Success = false;
9869       }
9870       ++ElementNo;
9871     }
9872 
9873     EvalObj.finishedConstructingBases();
9874   }
9875 
9876   // Initialize members.
9877   for (const auto *Field : RD->fields()) {
9878     // Anonymous bit-fields are not considered members of the class for
9879     // purposes of aggregate initialization.
9880     if (Field->isUnnamedBitfield())
9881       continue;
9882 
9883     LValue Subobject = This;
9884 
9885     bool HaveInit = ElementNo < E->getNumInits();
9886 
9887     // FIXME: Diagnostics here should point to the end of the initializer
9888     // list, not the start.
9889     if (!HandleLValueMember(Info, HaveInit ? E->getInit(ElementNo) : E,
9890                             Subobject, Field, &Layout))
9891       return false;
9892 
9893     // Perform an implicit value-initialization for members beyond the end of
9894     // the initializer list.
9895     ImplicitValueInitExpr VIE(HaveInit ? Info.Ctx.IntTy : Field->getType());
9896     const Expr *Init = HaveInit ? E->getInit(ElementNo++) : &VIE;
9897 
9898     // Temporarily override This, in case there's a CXXDefaultInitExpr in here.
9899     ThisOverrideRAII ThisOverride(*Info.CurrentCall, &This,
9900                                   isa<CXXDefaultInitExpr>(Init));
9901 
9902     APValue &FieldVal = Result.getStructField(Field->getFieldIndex());
9903     if (!EvaluateInPlace(FieldVal, Info, Subobject, Init) ||
9904         (Field->isBitField() && !truncateBitfieldValue(Info, Init,
9905                                                        FieldVal, Field))) {
9906       if (!Info.noteFailure())
9907         return false;
9908       Success = false;
9909     }
9910   }
9911 
9912   EvalObj.finishedConstructingFields();
9913 
9914   return Success;
9915 }
9916 
VisitCXXConstructExpr(const CXXConstructExpr * E,QualType T)9917 bool RecordExprEvaluator::VisitCXXConstructExpr(const CXXConstructExpr *E,
9918                                                 QualType T) {
9919   // Note that E's type is not necessarily the type of our class here; we might
9920   // be initializing an array element instead.
9921   const CXXConstructorDecl *FD = E->getConstructor();
9922   if (FD->isInvalidDecl() || FD->getParent()->isInvalidDecl()) return false;
9923 
9924   bool ZeroInit = E->requiresZeroInitialization();
9925   if (CheckTrivialDefaultConstructor(Info, E->getExprLoc(), FD, ZeroInit)) {
9926     // If we've already performed zero-initialization, we're already done.
9927     if (Result.hasValue())
9928       return true;
9929 
9930     if (ZeroInit)
9931       return ZeroInitialization(E, T);
9932 
9933     return getDefaultInitValue(T, Result);
9934   }
9935 
9936   const FunctionDecl *Definition = nullptr;
9937   auto Body = FD->getBody(Definition);
9938 
9939   if (!CheckConstexprFunction(Info, E->getExprLoc(), FD, Definition, Body))
9940     return false;
9941 
9942   // Avoid materializing a temporary for an elidable copy/move constructor.
9943   if (E->isElidable() && !ZeroInit) {
9944     // FIXME: This only handles the simplest case, where the source object
9945     //        is passed directly as the first argument to the constructor.
9946     //        This should also handle stepping though implicit casts and
9947     //        and conversion sequences which involve two steps, with a
9948     //        conversion operator followed by a converting constructor.
9949     const Expr *SrcObj = E->getArg(0);
9950     assert(SrcObj->isTemporaryObject(Info.Ctx, FD->getParent()));
9951     assert(Info.Ctx.hasSameUnqualifiedType(E->getType(), SrcObj->getType()));
9952     if (const MaterializeTemporaryExpr *ME =
9953             dyn_cast<MaterializeTemporaryExpr>(SrcObj))
9954       return Visit(ME->getSubExpr());
9955   }
9956 
9957   if (ZeroInit && !ZeroInitialization(E, T))
9958     return false;
9959 
9960   auto Args = llvm::makeArrayRef(E->getArgs(), E->getNumArgs());
9961   return HandleConstructorCall(E, This, Args,
9962                                cast<CXXConstructorDecl>(Definition), Info,
9963                                Result);
9964 }
9965 
VisitCXXInheritedCtorInitExpr(const CXXInheritedCtorInitExpr * E)9966 bool RecordExprEvaluator::VisitCXXInheritedCtorInitExpr(
9967     const CXXInheritedCtorInitExpr *E) {
9968   if (!Info.CurrentCall) {
9969     assert(Info.checkingPotentialConstantExpression());
9970     return false;
9971   }
9972 
9973   const CXXConstructorDecl *FD = E->getConstructor();
9974   if (FD->isInvalidDecl() || FD->getParent()->isInvalidDecl())
9975     return false;
9976 
9977   const FunctionDecl *Definition = nullptr;
9978   auto Body = FD->getBody(Definition);
9979 
9980   if (!CheckConstexprFunction(Info, E->getExprLoc(), FD, Definition, Body))
9981     return false;
9982 
9983   return HandleConstructorCall(E, This, Info.CurrentCall->Arguments,
9984                                cast<CXXConstructorDecl>(Definition), Info,
9985                                Result);
9986 }
9987 
VisitCXXStdInitializerListExpr(const CXXStdInitializerListExpr * E)9988 bool RecordExprEvaluator::VisitCXXStdInitializerListExpr(
9989     const CXXStdInitializerListExpr *E) {
9990   const ConstantArrayType *ArrayType =
9991       Info.Ctx.getAsConstantArrayType(E->getSubExpr()->getType());
9992 
9993   LValue Array;
9994   if (!EvaluateLValue(E->getSubExpr(), Array, Info))
9995     return false;
9996 
9997   // Get a pointer to the first element of the array.
9998   Array.addArray(Info, E, ArrayType);
9999 
10000   auto InvalidType = [&] {
10001     Info.FFDiag(E, diag::note_constexpr_unsupported_layout)
10002       << E->getType();
10003     return false;
10004   };
10005 
10006   // FIXME: Perform the checks on the field types in SemaInit.
10007   RecordDecl *Record = E->getType()->castAs<RecordType>()->getDecl();
10008   RecordDecl::field_iterator Field = Record->field_begin();
10009   if (Field == Record->field_end())
10010     return InvalidType();
10011 
10012   // Start pointer.
10013   if (!Field->getType()->isPointerType() ||
10014       !Info.Ctx.hasSameType(Field->getType()->getPointeeType(),
10015                             ArrayType->getElementType()))
10016     return InvalidType();
10017 
10018   // FIXME: What if the initializer_list type has base classes, etc?
10019   Result = APValue(APValue::UninitStruct(), 0, 2);
10020   Array.moveInto(Result.getStructField(0));
10021 
10022   if (++Field == Record->field_end())
10023     return InvalidType();
10024 
10025   if (Field->getType()->isPointerType() &&
10026       Info.Ctx.hasSameType(Field->getType()->getPointeeType(),
10027                            ArrayType->getElementType())) {
10028     // End pointer.
10029     if (!HandleLValueArrayAdjustment(Info, E, Array,
10030                                      ArrayType->getElementType(),
10031                                      ArrayType->getSize().getZExtValue()))
10032       return false;
10033     Array.moveInto(Result.getStructField(1));
10034   } else if (Info.Ctx.hasSameType(Field->getType(), Info.Ctx.getSizeType()))
10035     // Length.
10036     Result.getStructField(1) = APValue(APSInt(ArrayType->getSize()));
10037   else
10038     return InvalidType();
10039 
10040   if (++Field != Record->field_end())
10041     return InvalidType();
10042 
10043   return true;
10044 }
10045 
VisitLambdaExpr(const LambdaExpr * E)10046 bool RecordExprEvaluator::VisitLambdaExpr(const LambdaExpr *E) {
10047   const CXXRecordDecl *ClosureClass = E->getLambdaClass();
10048   if (ClosureClass->isInvalidDecl())
10049     return false;
10050 
10051   const size_t NumFields =
10052       std::distance(ClosureClass->field_begin(), ClosureClass->field_end());
10053 
10054   assert(NumFields == (size_t)std::distance(E->capture_init_begin(),
10055                                             E->capture_init_end()) &&
10056          "The number of lambda capture initializers should equal the number of "
10057          "fields within the closure type");
10058 
10059   Result = APValue(APValue::UninitStruct(), /*NumBases*/0, NumFields);
10060   // Iterate through all the lambda's closure object's fields and initialize
10061   // them.
10062   auto *CaptureInitIt = E->capture_init_begin();
10063   const LambdaCapture *CaptureIt = ClosureClass->captures_begin();
10064   bool Success = true;
10065   const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(ClosureClass);
10066   for (const auto *Field : ClosureClass->fields()) {
10067     assert(CaptureInitIt != E->capture_init_end());
10068     // Get the initializer for this field
10069     Expr *const CurFieldInit = *CaptureInitIt++;
10070 
10071     // If there is no initializer, either this is a VLA or an error has
10072     // occurred.
10073     if (!CurFieldInit)
10074       return Error(E);
10075 
10076     LValue Subobject = This;
10077 
10078     if (!HandleLValueMember(Info, E, Subobject, Field, &Layout))
10079       return false;
10080 
10081     APValue &FieldVal = Result.getStructField(Field->getFieldIndex());
10082     if (!EvaluateInPlace(FieldVal, Info, Subobject, CurFieldInit)) {
10083       if (!Info.keepEvaluatingAfterFailure())
10084         return false;
10085       Success = false;
10086     }
10087     ++CaptureIt;
10088   }
10089   return Success;
10090 }
10091 
EvaluateRecord(const Expr * E,const LValue & This,APValue & Result,EvalInfo & Info)10092 static bool EvaluateRecord(const Expr *E, const LValue &This,
10093                            APValue &Result, EvalInfo &Info) {
10094   assert(!E->isValueDependent());
10095   assert(E->isPRValue() && E->getType()->isRecordType() &&
10096          "can't evaluate expression as a record rvalue");
10097   return RecordExprEvaluator(Info, This, Result).Visit(E);
10098 }
10099 
10100 //===----------------------------------------------------------------------===//
10101 // Temporary Evaluation
10102 //
10103 // Temporaries are represented in the AST as rvalues, but generally behave like
10104 // lvalues. The full-object of which the temporary is a subobject is implicitly
10105 // materialized so that a reference can bind to it.
10106 //===----------------------------------------------------------------------===//
10107 namespace {
10108 class TemporaryExprEvaluator
10109   : public LValueExprEvaluatorBase<TemporaryExprEvaluator> {
10110 public:
TemporaryExprEvaluator(EvalInfo & Info,LValue & Result)10111   TemporaryExprEvaluator(EvalInfo &Info, LValue &Result) :
10112     LValueExprEvaluatorBaseTy(Info, Result, false) {}
10113 
10114   /// Visit an expression which constructs the value of this temporary.
VisitConstructExpr(const Expr * E)10115   bool VisitConstructExpr(const Expr *E) {
10116     APValue &Value = Info.CurrentCall->createTemporary(
10117         E, E->getType(), ScopeKind::FullExpression, Result);
10118     return EvaluateInPlace(Value, Info, Result, E);
10119   }
10120 
VisitCastExpr(const CastExpr * E)10121   bool VisitCastExpr(const CastExpr *E) {
10122     switch (E->getCastKind()) {
10123     default:
10124       return LValueExprEvaluatorBaseTy::VisitCastExpr(E);
10125 
10126     case CK_ConstructorConversion:
10127       return VisitConstructExpr(E->getSubExpr());
10128     }
10129   }
VisitInitListExpr(const InitListExpr * E)10130   bool VisitInitListExpr(const InitListExpr *E) {
10131     return VisitConstructExpr(E);
10132   }
VisitCXXConstructExpr(const CXXConstructExpr * E)10133   bool VisitCXXConstructExpr(const CXXConstructExpr *E) {
10134     return VisitConstructExpr(E);
10135   }
VisitCallExpr(const CallExpr * E)10136   bool VisitCallExpr(const CallExpr *E) {
10137     return VisitConstructExpr(E);
10138   }
VisitCXXStdInitializerListExpr(const CXXStdInitializerListExpr * E)10139   bool VisitCXXStdInitializerListExpr(const CXXStdInitializerListExpr *E) {
10140     return VisitConstructExpr(E);
10141   }
VisitLambdaExpr(const LambdaExpr * E)10142   bool VisitLambdaExpr(const LambdaExpr *E) {
10143     return VisitConstructExpr(E);
10144   }
10145 };
10146 } // end anonymous namespace
10147 
10148 /// Evaluate an expression of record type as a temporary.
EvaluateTemporary(const Expr * E,LValue & Result,EvalInfo & Info)10149 static bool EvaluateTemporary(const Expr *E, LValue &Result, EvalInfo &Info) {
10150   assert(!E->isValueDependent());
10151   assert(E->isPRValue() && E->getType()->isRecordType());
10152   return TemporaryExprEvaluator(Info, Result).Visit(E);
10153 }
10154 
10155 //===----------------------------------------------------------------------===//
10156 // Vector Evaluation
10157 //===----------------------------------------------------------------------===//
10158 
10159 namespace {
10160   class VectorExprEvaluator
10161   : public ExprEvaluatorBase<VectorExprEvaluator> {
10162     APValue &Result;
10163   public:
10164 
VectorExprEvaluator(EvalInfo & info,APValue & Result)10165     VectorExprEvaluator(EvalInfo &info, APValue &Result)
10166       : ExprEvaluatorBaseTy(info), Result(Result) {}
10167 
Success(ArrayRef<APValue> V,const Expr * E)10168     bool Success(ArrayRef<APValue> V, const Expr *E) {
10169       assert(V.size() == E->getType()->castAs<VectorType>()->getNumElements());
10170       // FIXME: remove this APValue copy.
10171       Result = APValue(V.data(), V.size());
10172       return true;
10173     }
Success(const APValue & V,const Expr * E)10174     bool Success(const APValue &V, const Expr *E) {
10175       assert(V.isVector());
10176       Result = V;
10177       return true;
10178     }
10179     bool ZeroInitialization(const Expr *E);
10180 
VisitUnaryReal(const UnaryOperator * E)10181     bool VisitUnaryReal(const UnaryOperator *E)
10182       { return Visit(E->getSubExpr()); }
10183     bool VisitCastExpr(const CastExpr* E);
10184     bool VisitInitListExpr(const InitListExpr *E);
10185     bool VisitUnaryImag(const UnaryOperator *E);
10186     bool VisitBinaryOperator(const BinaryOperator *E);
10187     // FIXME: Missing: unary -, unary ~, conditional operator (for GNU
10188     //                 conditional select), shufflevector, ExtVectorElementExpr
10189   };
10190 } // end anonymous namespace
10191 
EvaluateVector(const Expr * E,APValue & Result,EvalInfo & Info)10192 static bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
10193   assert(E->isPRValue() && E->getType()->isVectorType() &&
10194          "not a vector prvalue");
10195   return VectorExprEvaluator(Info, Result).Visit(E);
10196 }
10197 
VisitCastExpr(const CastExpr * E)10198 bool VectorExprEvaluator::VisitCastExpr(const CastExpr *E) {
10199   const VectorType *VTy = E->getType()->castAs<VectorType>();
10200   unsigned NElts = VTy->getNumElements();
10201 
10202   const Expr *SE = E->getSubExpr();
10203   QualType SETy = SE->getType();
10204 
10205   switch (E->getCastKind()) {
10206   case CK_VectorSplat: {
10207     APValue Val = APValue();
10208     if (SETy->isIntegerType()) {
10209       APSInt IntResult;
10210       if (!EvaluateInteger(SE, IntResult, Info))
10211         return false;
10212       Val = APValue(std::move(IntResult));
10213     } else if (SETy->isRealFloatingType()) {
10214       APFloat FloatResult(0.0);
10215       if (!EvaluateFloat(SE, FloatResult, Info))
10216         return false;
10217       Val = APValue(std::move(FloatResult));
10218     } else {
10219       return Error(E);
10220     }
10221 
10222     // Splat and create vector APValue.
10223     SmallVector<APValue, 4> Elts(NElts, Val);
10224     return Success(Elts, E);
10225   }
10226   case CK_BitCast: {
10227     // Evaluate the operand into an APInt we can extract from.
10228     llvm::APInt SValInt;
10229     if (!EvalAndBitcastToAPInt(Info, SE, SValInt))
10230       return false;
10231     // Extract the elements
10232     QualType EltTy = VTy->getElementType();
10233     unsigned EltSize = Info.Ctx.getTypeSize(EltTy);
10234     bool BigEndian = Info.Ctx.getTargetInfo().isBigEndian();
10235     SmallVector<APValue, 4> Elts;
10236     if (EltTy->isRealFloatingType()) {
10237       const llvm::fltSemantics &Sem = Info.Ctx.getFloatTypeSemantics(EltTy);
10238       unsigned FloatEltSize = EltSize;
10239       if (&Sem == &APFloat::x87DoubleExtended())
10240         FloatEltSize = 80;
10241       for (unsigned i = 0; i < NElts; i++) {
10242         llvm::APInt Elt;
10243         if (BigEndian)
10244           Elt = SValInt.rotl(i*EltSize+FloatEltSize).trunc(FloatEltSize);
10245         else
10246           Elt = SValInt.rotr(i*EltSize).trunc(FloatEltSize);
10247         Elts.push_back(APValue(APFloat(Sem, Elt)));
10248       }
10249     } else if (EltTy->isIntegerType()) {
10250       for (unsigned i = 0; i < NElts; i++) {
10251         llvm::APInt Elt;
10252         if (BigEndian)
10253           Elt = SValInt.rotl(i*EltSize+EltSize).zextOrTrunc(EltSize);
10254         else
10255           Elt = SValInt.rotr(i*EltSize).zextOrTrunc(EltSize);
10256         Elts.push_back(APValue(APSInt(Elt, !EltTy->isSignedIntegerType())));
10257       }
10258     } else {
10259       return Error(E);
10260     }
10261     return Success(Elts, E);
10262   }
10263   default:
10264     return ExprEvaluatorBaseTy::VisitCastExpr(E);
10265   }
10266 }
10267 
10268 bool
VisitInitListExpr(const InitListExpr * E)10269 VectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
10270   const VectorType *VT = E->getType()->castAs<VectorType>();
10271   unsigned NumInits = E->getNumInits();
10272   unsigned NumElements = VT->getNumElements();
10273 
10274   QualType EltTy = VT->getElementType();
10275   SmallVector<APValue, 4> Elements;
10276 
10277   // The number of initializers can be less than the number of
10278   // vector elements. For OpenCL, this can be due to nested vector
10279   // initialization. For GCC compatibility, missing trailing elements
10280   // should be initialized with zeroes.
10281   unsigned CountInits = 0, CountElts = 0;
10282   while (CountElts < NumElements) {
10283     // Handle nested vector initialization.
10284     if (CountInits < NumInits
10285         && E->getInit(CountInits)->getType()->isVectorType()) {
10286       APValue v;
10287       if (!EvaluateVector(E->getInit(CountInits), v, Info))
10288         return Error(E);
10289       unsigned vlen = v.getVectorLength();
10290       for (unsigned j = 0; j < vlen; j++)
10291         Elements.push_back(v.getVectorElt(j));
10292       CountElts += vlen;
10293     } else if (EltTy->isIntegerType()) {
10294       llvm::APSInt sInt(32);
10295       if (CountInits < NumInits) {
10296         if (!EvaluateInteger(E->getInit(CountInits), sInt, Info))
10297           return false;
10298       } else // trailing integer zero.
10299         sInt = Info.Ctx.MakeIntValue(0, EltTy);
10300       Elements.push_back(APValue(sInt));
10301       CountElts++;
10302     } else {
10303       llvm::APFloat f(0.0);
10304       if (CountInits < NumInits) {
10305         if (!EvaluateFloat(E->getInit(CountInits), f, Info))
10306           return false;
10307       } else // trailing float zero.
10308         f = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy));
10309       Elements.push_back(APValue(f));
10310       CountElts++;
10311     }
10312     CountInits++;
10313   }
10314   return Success(Elements, E);
10315 }
10316 
10317 bool
ZeroInitialization(const Expr * E)10318 VectorExprEvaluator::ZeroInitialization(const Expr *E) {
10319   const auto *VT = E->getType()->castAs<VectorType>();
10320   QualType EltTy = VT->getElementType();
10321   APValue ZeroElement;
10322   if (EltTy->isIntegerType())
10323     ZeroElement = APValue(Info.Ctx.MakeIntValue(0, EltTy));
10324   else
10325     ZeroElement =
10326         APValue(APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy)));
10327 
10328   SmallVector<APValue, 4> Elements(VT->getNumElements(), ZeroElement);
10329   return Success(Elements, E);
10330 }
10331 
VisitUnaryImag(const UnaryOperator * E)10332 bool VectorExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
10333   VisitIgnoredValue(E->getSubExpr());
10334   return ZeroInitialization(E);
10335 }
10336 
VisitBinaryOperator(const BinaryOperator * E)10337 bool VectorExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
10338   BinaryOperatorKind Op = E->getOpcode();
10339   assert(Op != BO_PtrMemD && Op != BO_PtrMemI && Op != BO_Cmp &&
10340          "Operation not supported on vector types");
10341 
10342   if (Op == BO_Comma)
10343     return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
10344 
10345   Expr *LHS = E->getLHS();
10346   Expr *RHS = E->getRHS();
10347 
10348   assert(LHS->getType()->isVectorType() && RHS->getType()->isVectorType() &&
10349          "Must both be vector types");
10350   // Checking JUST the types are the same would be fine, except shifts don't
10351   // need to have their types be the same (since you always shift by an int).
10352   assert(LHS->getType()->castAs<VectorType>()->getNumElements() ==
10353              E->getType()->castAs<VectorType>()->getNumElements() &&
10354          RHS->getType()->castAs<VectorType>()->getNumElements() ==
10355              E->getType()->castAs<VectorType>()->getNumElements() &&
10356          "All operands must be the same size.");
10357 
10358   APValue LHSValue;
10359   APValue RHSValue;
10360   bool LHSOK = Evaluate(LHSValue, Info, LHS);
10361   if (!LHSOK && !Info.noteFailure())
10362     return false;
10363   if (!Evaluate(RHSValue, Info, RHS) || !LHSOK)
10364     return false;
10365 
10366   if (!handleVectorVectorBinOp(Info, E, Op, LHSValue, RHSValue))
10367     return false;
10368 
10369   return Success(LHSValue, E);
10370 }
10371 
10372 //===----------------------------------------------------------------------===//
10373 // Array Evaluation
10374 //===----------------------------------------------------------------------===//
10375 
10376 namespace {
10377   class ArrayExprEvaluator
10378   : public ExprEvaluatorBase<ArrayExprEvaluator> {
10379     const LValue &This;
10380     APValue &Result;
10381   public:
10382 
ArrayExprEvaluator(EvalInfo & Info,const LValue & This,APValue & Result)10383     ArrayExprEvaluator(EvalInfo &Info, const LValue &This, APValue &Result)
10384       : ExprEvaluatorBaseTy(Info), This(This), Result(Result) {}
10385 
Success(const APValue & V,const Expr * E)10386     bool Success(const APValue &V, const Expr *E) {
10387       assert(V.isArray() && "expected array");
10388       Result = V;
10389       return true;
10390     }
10391 
ZeroInitialization(const Expr * E)10392     bool ZeroInitialization(const Expr *E) {
10393       const ConstantArrayType *CAT =
10394           Info.Ctx.getAsConstantArrayType(E->getType());
10395       if (!CAT) {
10396         if (E->getType()->isIncompleteArrayType()) {
10397           // We can be asked to zero-initialize a flexible array member; this
10398           // is represented as an ImplicitValueInitExpr of incomplete array
10399           // type. In this case, the array has zero elements.
10400           Result = APValue(APValue::UninitArray(), 0, 0);
10401           return true;
10402         }
10403         // FIXME: We could handle VLAs here.
10404         return Error(E);
10405       }
10406 
10407       Result = APValue(APValue::UninitArray(), 0,
10408                        CAT->getSize().getZExtValue());
10409       if (!Result.hasArrayFiller())
10410         return true;
10411 
10412       // Zero-initialize all elements.
10413       LValue Subobject = This;
10414       Subobject.addArray(Info, E, CAT);
10415       ImplicitValueInitExpr VIE(CAT->getElementType());
10416       return EvaluateInPlace(Result.getArrayFiller(), Info, Subobject, &VIE);
10417     }
10418 
VisitCallExpr(const CallExpr * E)10419     bool VisitCallExpr(const CallExpr *E) {
10420       return handleCallExpr(E, Result, &This);
10421     }
10422     bool VisitInitListExpr(const InitListExpr *E,
10423                            QualType AllocType = QualType());
10424     bool VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E);
10425     bool VisitCXXConstructExpr(const CXXConstructExpr *E);
10426     bool VisitCXXConstructExpr(const CXXConstructExpr *E,
10427                                const LValue &Subobject,
10428                                APValue *Value, QualType Type);
VisitStringLiteral(const StringLiteral * E,QualType AllocType=QualType ())10429     bool VisitStringLiteral(const StringLiteral *E,
10430                             QualType AllocType = QualType()) {
10431       expandStringLiteral(Info, E, Result, AllocType);
10432       return true;
10433     }
10434   };
10435 } // end anonymous namespace
10436 
EvaluateArray(const Expr * E,const LValue & This,APValue & Result,EvalInfo & Info)10437 static bool EvaluateArray(const Expr *E, const LValue &This,
10438                           APValue &Result, EvalInfo &Info) {
10439   assert(!E->isValueDependent());
10440   assert(E->isPRValue() && E->getType()->isArrayType() &&
10441          "not an array prvalue");
10442   return ArrayExprEvaluator(Info, This, Result).Visit(E);
10443 }
10444 
EvaluateArrayNewInitList(EvalInfo & Info,LValue & This,APValue & Result,const InitListExpr * ILE,QualType AllocType)10445 static bool EvaluateArrayNewInitList(EvalInfo &Info, LValue &This,
10446                                      APValue &Result, const InitListExpr *ILE,
10447                                      QualType AllocType) {
10448   assert(!ILE->isValueDependent());
10449   assert(ILE->isPRValue() && ILE->getType()->isArrayType() &&
10450          "not an array prvalue");
10451   return ArrayExprEvaluator(Info, This, Result)
10452       .VisitInitListExpr(ILE, AllocType);
10453 }
10454 
EvaluateArrayNewConstructExpr(EvalInfo & Info,LValue & This,APValue & Result,const CXXConstructExpr * CCE,QualType AllocType)10455 static bool EvaluateArrayNewConstructExpr(EvalInfo &Info, LValue &This,
10456                                           APValue &Result,
10457                                           const CXXConstructExpr *CCE,
10458                                           QualType AllocType) {
10459   assert(!CCE->isValueDependent());
10460   assert(CCE->isPRValue() && CCE->getType()->isArrayType() &&
10461          "not an array prvalue");
10462   return ArrayExprEvaluator(Info, This, Result)
10463       .VisitCXXConstructExpr(CCE, This, &Result, AllocType);
10464 }
10465 
10466 // Return true iff the given array filler may depend on the element index.
MaybeElementDependentArrayFiller(const Expr * FillerExpr)10467 static bool MaybeElementDependentArrayFiller(const Expr *FillerExpr) {
10468   // For now, just allow non-class value-initialization and initialization
10469   // lists comprised of them.
10470   if (isa<ImplicitValueInitExpr>(FillerExpr))
10471     return false;
10472   if (const InitListExpr *ILE = dyn_cast<InitListExpr>(FillerExpr)) {
10473     for (unsigned I = 0, E = ILE->getNumInits(); I != E; ++I) {
10474       if (MaybeElementDependentArrayFiller(ILE->getInit(I)))
10475         return true;
10476     }
10477     return false;
10478   }
10479   return true;
10480 }
10481 
VisitInitListExpr(const InitListExpr * E,QualType AllocType)10482 bool ArrayExprEvaluator::VisitInitListExpr(const InitListExpr *E,
10483                                            QualType AllocType) {
10484   const ConstantArrayType *CAT = Info.Ctx.getAsConstantArrayType(
10485       AllocType.isNull() ? E->getType() : AllocType);
10486   if (!CAT)
10487     return Error(E);
10488 
10489   // C++11 [dcl.init.string]p1: A char array [...] can be initialized by [...]
10490   // an appropriately-typed string literal enclosed in braces.
10491   if (E->isStringLiteralInit()) {
10492     auto *SL = dyn_cast<StringLiteral>(E->getInit(0)->IgnoreParens());
10493     // FIXME: Support ObjCEncodeExpr here once we support it in
10494     // ArrayExprEvaluator generally.
10495     if (!SL)
10496       return Error(E);
10497     return VisitStringLiteral(SL, AllocType);
10498   }
10499 
10500   bool Success = true;
10501 
10502   assert((!Result.isArray() || Result.getArrayInitializedElts() == 0) &&
10503          "zero-initialized array shouldn't have any initialized elts");
10504   APValue Filler;
10505   if (Result.isArray() && Result.hasArrayFiller())
10506     Filler = Result.getArrayFiller();
10507 
10508   unsigned NumEltsToInit = E->getNumInits();
10509   unsigned NumElts = CAT->getSize().getZExtValue();
10510   const Expr *FillerExpr = E->hasArrayFiller() ? E->getArrayFiller() : nullptr;
10511 
10512   // If the initializer might depend on the array index, run it for each
10513   // array element.
10514   if (NumEltsToInit != NumElts && MaybeElementDependentArrayFiller(FillerExpr))
10515     NumEltsToInit = NumElts;
10516 
10517   LLVM_DEBUG(llvm::dbgs() << "The number of elements to initialize: "
10518                           << NumEltsToInit << ".\n");
10519 
10520   Result = APValue(APValue::UninitArray(), NumEltsToInit, NumElts);
10521 
10522   // If the array was previously zero-initialized, preserve the
10523   // zero-initialized values.
10524   if (Filler.hasValue()) {
10525     for (unsigned I = 0, E = Result.getArrayInitializedElts(); I != E; ++I)
10526       Result.getArrayInitializedElt(I) = Filler;
10527     if (Result.hasArrayFiller())
10528       Result.getArrayFiller() = Filler;
10529   }
10530 
10531   LValue Subobject = This;
10532   Subobject.addArray(Info, E, CAT);
10533   for (unsigned Index = 0; Index != NumEltsToInit; ++Index) {
10534     const Expr *Init =
10535         Index < E->getNumInits() ? E->getInit(Index) : FillerExpr;
10536     if (!EvaluateInPlace(Result.getArrayInitializedElt(Index),
10537                          Info, Subobject, Init) ||
10538         !HandleLValueArrayAdjustment(Info, Init, Subobject,
10539                                      CAT->getElementType(), 1)) {
10540       if (!Info.noteFailure())
10541         return false;
10542       Success = false;
10543     }
10544   }
10545 
10546   if (!Result.hasArrayFiller())
10547     return Success;
10548 
10549   // If we get here, we have a trivial filler, which we can just evaluate
10550   // once and splat over the rest of the array elements.
10551   assert(FillerExpr && "no array filler for incomplete init list");
10552   return EvaluateInPlace(Result.getArrayFiller(), Info, Subobject,
10553                          FillerExpr) && Success;
10554 }
10555 
VisitArrayInitLoopExpr(const ArrayInitLoopExpr * E)10556 bool ArrayExprEvaluator::VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E) {
10557   LValue CommonLV;
10558   if (E->getCommonExpr() &&
10559       !Evaluate(Info.CurrentCall->createTemporary(
10560                     E->getCommonExpr(),
10561                     getStorageType(Info.Ctx, E->getCommonExpr()),
10562                     ScopeKind::FullExpression, CommonLV),
10563                 Info, E->getCommonExpr()->getSourceExpr()))
10564     return false;
10565 
10566   auto *CAT = cast<ConstantArrayType>(E->getType()->castAsArrayTypeUnsafe());
10567 
10568   uint64_t Elements = CAT->getSize().getZExtValue();
10569   Result = APValue(APValue::UninitArray(), Elements, Elements);
10570 
10571   LValue Subobject = This;
10572   Subobject.addArray(Info, E, CAT);
10573 
10574   bool Success = true;
10575   for (EvalInfo::ArrayInitLoopIndex Index(Info); Index != Elements; ++Index) {
10576     if (!EvaluateInPlace(Result.getArrayInitializedElt(Index),
10577                          Info, Subobject, E->getSubExpr()) ||
10578         !HandleLValueArrayAdjustment(Info, E, Subobject,
10579                                      CAT->getElementType(), 1)) {
10580       if (!Info.noteFailure())
10581         return false;
10582       Success = false;
10583     }
10584   }
10585 
10586   return Success;
10587 }
10588 
VisitCXXConstructExpr(const CXXConstructExpr * E)10589 bool ArrayExprEvaluator::VisitCXXConstructExpr(const CXXConstructExpr *E) {
10590   return VisitCXXConstructExpr(E, This, &Result, E->getType());
10591 }
10592 
VisitCXXConstructExpr(const CXXConstructExpr * E,const LValue & Subobject,APValue * Value,QualType Type)10593 bool ArrayExprEvaluator::VisitCXXConstructExpr(const CXXConstructExpr *E,
10594                                                const LValue &Subobject,
10595                                                APValue *Value,
10596                                                QualType Type) {
10597   bool HadZeroInit = Value->hasValue();
10598 
10599   if (const ConstantArrayType *CAT = Info.Ctx.getAsConstantArrayType(Type)) {
10600     unsigned N = CAT->getSize().getZExtValue();
10601 
10602     // Preserve the array filler if we had prior zero-initialization.
10603     APValue Filler =
10604       HadZeroInit && Value->hasArrayFiller() ? Value->getArrayFiller()
10605                                              : APValue();
10606 
10607     *Value = APValue(APValue::UninitArray(), N, N);
10608 
10609     if (HadZeroInit)
10610       for (unsigned I = 0; I != N; ++I)
10611         Value->getArrayInitializedElt(I) = Filler;
10612 
10613     // Initialize the elements.
10614     LValue ArrayElt = Subobject;
10615     ArrayElt.addArray(Info, E, CAT);
10616     for (unsigned I = 0; I != N; ++I)
10617       if (!VisitCXXConstructExpr(E, ArrayElt, &Value->getArrayInitializedElt(I),
10618                                  CAT->getElementType()) ||
10619           !HandleLValueArrayAdjustment(Info, E, ArrayElt,
10620                                        CAT->getElementType(), 1))
10621         return false;
10622 
10623     return true;
10624   }
10625 
10626   if (!Type->isRecordType())
10627     return Error(E);
10628 
10629   return RecordExprEvaluator(Info, Subobject, *Value)
10630              .VisitCXXConstructExpr(E, Type);
10631 }
10632 
10633 //===----------------------------------------------------------------------===//
10634 // Integer Evaluation
10635 //
10636 // As a GNU extension, we support casting pointers to sufficiently-wide integer
10637 // types and back in constant folding. Integer values are thus represented
10638 // either as an integer-valued APValue, or as an lvalue-valued APValue.
10639 //===----------------------------------------------------------------------===//
10640 
10641 namespace {
10642 class IntExprEvaluator
10643         : public ExprEvaluatorBase<IntExprEvaluator> {
10644   APValue &Result;
10645 public:
IntExprEvaluator(EvalInfo & info,APValue & result)10646   IntExprEvaluator(EvalInfo &info, APValue &result)
10647       : ExprEvaluatorBaseTy(info), Result(result) {}
10648 
Success(const llvm::APSInt & SI,const Expr * E,APValue & Result)10649   bool Success(const llvm::APSInt &SI, const Expr *E, APValue &Result) {
10650     assert(E->getType()->isIntegralOrEnumerationType() &&
10651            "Invalid evaluation result.");
10652     assert(SI.isSigned() == E->getType()->isSignedIntegerOrEnumerationType() &&
10653            "Invalid evaluation result.");
10654     assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
10655            "Invalid evaluation result.");
10656     Result = APValue(SI);
10657     return true;
10658   }
Success(const llvm::APSInt & SI,const Expr * E)10659   bool Success(const llvm::APSInt &SI, const Expr *E) {
10660     return Success(SI, E, Result);
10661   }
10662 
Success(const llvm::APInt & I,const Expr * E,APValue & Result)10663   bool Success(const llvm::APInt &I, const Expr *E, APValue &Result) {
10664     assert(E->getType()->isIntegralOrEnumerationType() &&
10665            "Invalid evaluation result.");
10666     assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
10667            "Invalid evaluation result.");
10668     Result = APValue(APSInt(I));
10669     Result.getInt().setIsUnsigned(
10670                             E->getType()->isUnsignedIntegerOrEnumerationType());
10671     return true;
10672   }
Success(const llvm::APInt & I,const Expr * E)10673   bool Success(const llvm::APInt &I, const Expr *E) {
10674     return Success(I, E, Result);
10675   }
10676 
Success(uint64_t Value,const Expr * E,APValue & Result)10677   bool Success(uint64_t Value, const Expr *E, APValue &Result) {
10678     assert(E->getType()->isIntegralOrEnumerationType() &&
10679            "Invalid evaluation result.");
10680     Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
10681     return true;
10682   }
Success(uint64_t Value,const Expr * E)10683   bool Success(uint64_t Value, const Expr *E) {
10684     return Success(Value, E, Result);
10685   }
10686 
Success(CharUnits Size,const Expr * E)10687   bool Success(CharUnits Size, const Expr *E) {
10688     return Success(Size.getQuantity(), E);
10689   }
10690 
Success(const APValue & V,const Expr * E)10691   bool Success(const APValue &V, const Expr *E) {
10692     if (V.isLValue() || V.isAddrLabelDiff() || V.isIndeterminate()) {
10693       Result = V;
10694       return true;
10695     }
10696     return Success(V.getInt(), E);
10697   }
10698 
ZeroInitialization(const Expr * E)10699   bool ZeroInitialization(const Expr *E) { return Success(0, E); }
10700 
10701   //===--------------------------------------------------------------------===//
10702   //                            Visitor Methods
10703   //===--------------------------------------------------------------------===//
10704 
VisitIntegerLiteral(const IntegerLiteral * E)10705   bool VisitIntegerLiteral(const IntegerLiteral *E) {
10706     return Success(E->getValue(), E);
10707   }
VisitCharacterLiteral(const CharacterLiteral * E)10708   bool VisitCharacterLiteral(const CharacterLiteral *E) {
10709     return Success(E->getValue(), E);
10710   }
10711 
10712   bool CheckReferencedDecl(const Expr *E, const Decl *D);
VisitDeclRefExpr(const DeclRefExpr * E)10713   bool VisitDeclRefExpr(const DeclRefExpr *E) {
10714     if (CheckReferencedDecl(E, E->getDecl()))
10715       return true;
10716 
10717     return ExprEvaluatorBaseTy::VisitDeclRefExpr(E);
10718   }
VisitMemberExpr(const MemberExpr * E)10719   bool VisitMemberExpr(const MemberExpr *E) {
10720     if (CheckReferencedDecl(E, E->getMemberDecl())) {
10721       VisitIgnoredBaseExpression(E->getBase());
10722       return true;
10723     }
10724 
10725     return ExprEvaluatorBaseTy::VisitMemberExpr(E);
10726   }
10727 
10728   bool VisitCallExpr(const CallExpr *E);
10729   bool VisitBuiltinCallExpr(const CallExpr *E, unsigned BuiltinOp);
10730   bool VisitBinaryOperator(const BinaryOperator *E);
10731   bool VisitOffsetOfExpr(const OffsetOfExpr *E);
10732   bool VisitUnaryOperator(const UnaryOperator *E);
10733 
10734   bool VisitCastExpr(const CastExpr* E);
10735   bool VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
10736 
VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr * E)10737   bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
10738     return Success(E->getValue(), E);
10739   }
10740 
VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr * E)10741   bool VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr *E) {
10742     return Success(E->getValue(), E);
10743   }
10744 
VisitArrayInitIndexExpr(const ArrayInitIndexExpr * E)10745   bool VisitArrayInitIndexExpr(const ArrayInitIndexExpr *E) {
10746     if (Info.ArrayInitIndex == uint64_t(-1)) {
10747       // We were asked to evaluate this subexpression independent of the
10748       // enclosing ArrayInitLoopExpr. We can't do that.
10749       Info.FFDiag(E);
10750       return false;
10751     }
10752     return Success(Info.ArrayInitIndex, E);
10753   }
10754 
10755   // Note, GNU defines __null as an integer, not a pointer.
VisitGNUNullExpr(const GNUNullExpr * E)10756   bool VisitGNUNullExpr(const GNUNullExpr *E) {
10757     return ZeroInitialization(E);
10758   }
10759 
VisitTypeTraitExpr(const TypeTraitExpr * E)10760   bool VisitTypeTraitExpr(const TypeTraitExpr *E) {
10761     return Success(E->getValue(), E);
10762   }
10763 
VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr * E)10764   bool VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
10765     return Success(E->getValue(), E);
10766   }
10767 
VisitExpressionTraitExpr(const ExpressionTraitExpr * E)10768   bool VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
10769     return Success(E->getValue(), E);
10770   }
10771 
10772   bool VisitUnaryReal(const UnaryOperator *E);
10773   bool VisitUnaryImag(const UnaryOperator *E);
10774 
10775   bool VisitCXXNoexceptExpr(const CXXNoexceptExpr *E);
10776   bool VisitSizeOfPackExpr(const SizeOfPackExpr *E);
10777   bool VisitSourceLocExpr(const SourceLocExpr *E);
10778   bool VisitConceptSpecializationExpr(const ConceptSpecializationExpr *E);
10779   bool VisitRequiresExpr(const RequiresExpr *E);
10780   // FIXME: Missing: array subscript of vector, member of vector
10781 };
10782 
10783 class FixedPointExprEvaluator
10784     : public ExprEvaluatorBase<FixedPointExprEvaluator> {
10785   APValue &Result;
10786 
10787  public:
FixedPointExprEvaluator(EvalInfo & info,APValue & result)10788   FixedPointExprEvaluator(EvalInfo &info, APValue &result)
10789       : ExprEvaluatorBaseTy(info), Result(result) {}
10790 
Success(const llvm::APInt & I,const Expr * E)10791   bool Success(const llvm::APInt &I, const Expr *E) {
10792     return Success(
10793         APFixedPoint(I, Info.Ctx.getFixedPointSemantics(E->getType())), E);
10794   }
10795 
Success(uint64_t Value,const Expr * E)10796   bool Success(uint64_t Value, const Expr *E) {
10797     return Success(
10798         APFixedPoint(Value, Info.Ctx.getFixedPointSemantics(E->getType())), E);
10799   }
10800 
Success(const APValue & V,const Expr * E)10801   bool Success(const APValue &V, const Expr *E) {
10802     return Success(V.getFixedPoint(), E);
10803   }
10804 
Success(const APFixedPoint & V,const Expr * E)10805   bool Success(const APFixedPoint &V, const Expr *E) {
10806     assert(E->getType()->isFixedPointType() && "Invalid evaluation result.");
10807     assert(V.getWidth() == Info.Ctx.getIntWidth(E->getType()) &&
10808            "Invalid evaluation result.");
10809     Result = APValue(V);
10810     return true;
10811   }
10812 
10813   //===--------------------------------------------------------------------===//
10814   //                            Visitor Methods
10815   //===--------------------------------------------------------------------===//
10816 
VisitFixedPointLiteral(const FixedPointLiteral * E)10817   bool VisitFixedPointLiteral(const FixedPointLiteral *E) {
10818     return Success(E->getValue(), E);
10819   }
10820 
10821   bool VisitCastExpr(const CastExpr *E);
10822   bool VisitUnaryOperator(const UnaryOperator *E);
10823   bool VisitBinaryOperator(const BinaryOperator *E);
10824 };
10825 } // end anonymous namespace
10826 
10827 /// EvaluateIntegerOrLValue - Evaluate an rvalue integral-typed expression, and
10828 /// produce either the integer value or a pointer.
10829 ///
10830 /// GCC has a heinous extension which folds casts between pointer types and
10831 /// pointer-sized integral types. We support this by allowing the evaluation of
10832 /// an integer rvalue to produce a pointer (represented as an lvalue) instead.
10833 /// Some simple arithmetic on such values is supported (they are treated much
10834 /// like char*).
EvaluateIntegerOrLValue(const Expr * E,APValue & Result,EvalInfo & Info)10835 static bool EvaluateIntegerOrLValue(const Expr *E, APValue &Result,
10836                                     EvalInfo &Info) {
10837   assert(!E->isValueDependent());
10838   assert(E->isPRValue() && E->getType()->isIntegralOrEnumerationType());
10839   return IntExprEvaluator(Info, Result).Visit(E);
10840 }
10841 
EvaluateInteger(const Expr * E,APSInt & Result,EvalInfo & Info)10842 static bool EvaluateInteger(const Expr *E, APSInt &Result, EvalInfo &Info) {
10843   assert(!E->isValueDependent());
10844   APValue Val;
10845   if (!EvaluateIntegerOrLValue(E, Val, Info))
10846     return false;
10847   if (!Val.isInt()) {
10848     // FIXME: It would be better to produce the diagnostic for casting
10849     //        a pointer to an integer.
10850     Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
10851     return false;
10852   }
10853   Result = Val.getInt();
10854   return true;
10855 }
10856 
VisitSourceLocExpr(const SourceLocExpr * E)10857 bool IntExprEvaluator::VisitSourceLocExpr(const SourceLocExpr *E) {
10858   APValue Evaluated = E->EvaluateInContext(
10859       Info.Ctx, Info.CurrentCall->CurSourceLocExprScope.getDefaultExpr());
10860   return Success(Evaluated, E);
10861 }
10862 
EvaluateFixedPoint(const Expr * E,APFixedPoint & Result,EvalInfo & Info)10863 static bool EvaluateFixedPoint(const Expr *E, APFixedPoint &Result,
10864                                EvalInfo &Info) {
10865   assert(!E->isValueDependent());
10866   if (E->getType()->isFixedPointType()) {
10867     APValue Val;
10868     if (!FixedPointExprEvaluator(Info, Val).Visit(E))
10869       return false;
10870     if (!Val.isFixedPoint())
10871       return false;
10872 
10873     Result = Val.getFixedPoint();
10874     return true;
10875   }
10876   return false;
10877 }
10878 
EvaluateFixedPointOrInteger(const Expr * E,APFixedPoint & Result,EvalInfo & Info)10879 static bool EvaluateFixedPointOrInteger(const Expr *E, APFixedPoint &Result,
10880                                         EvalInfo &Info) {
10881   assert(!E->isValueDependent());
10882   if (E->getType()->isIntegerType()) {
10883     auto FXSema = Info.Ctx.getFixedPointSemantics(E->getType());
10884     APSInt Val;
10885     if (!EvaluateInteger(E, Val, Info))
10886       return false;
10887     Result = APFixedPoint(Val, FXSema);
10888     return true;
10889   } else if (E->getType()->isFixedPointType()) {
10890     return EvaluateFixedPoint(E, Result, Info);
10891   }
10892   return false;
10893 }
10894 
10895 /// Check whether the given declaration can be directly converted to an integral
10896 /// rvalue. If not, no diagnostic is produced; there are other things we can
10897 /// try.
CheckReferencedDecl(const Expr * E,const Decl * D)10898 bool IntExprEvaluator::CheckReferencedDecl(const Expr* E, const Decl* D) {
10899   // Enums are integer constant exprs.
10900   if (const EnumConstantDecl *ECD = dyn_cast<EnumConstantDecl>(D)) {
10901     // Check for signedness/width mismatches between E type and ECD value.
10902     bool SameSign = (ECD->getInitVal().isSigned()
10903                      == E->getType()->isSignedIntegerOrEnumerationType());
10904     bool SameWidth = (ECD->getInitVal().getBitWidth()
10905                       == Info.Ctx.getIntWidth(E->getType()));
10906     if (SameSign && SameWidth)
10907       return Success(ECD->getInitVal(), E);
10908     else {
10909       // Get rid of mismatch (otherwise Success assertions will fail)
10910       // by computing a new value matching the type of E.
10911       llvm::APSInt Val = ECD->getInitVal();
10912       if (!SameSign)
10913         Val.setIsSigned(!ECD->getInitVal().isSigned());
10914       if (!SameWidth)
10915         Val = Val.extOrTrunc(Info.Ctx.getIntWidth(E->getType()));
10916       return Success(Val, E);
10917     }
10918   }
10919   return false;
10920 }
10921 
10922 /// Values returned by __builtin_classify_type, chosen to match the values
10923 /// produced by GCC's builtin.
10924 enum class GCCTypeClass {
10925   None = -1,
10926   Void = 0,
10927   Integer = 1,
10928   // GCC reserves 2 for character types, but instead classifies them as
10929   // integers.
10930   Enum = 3,
10931   Bool = 4,
10932   Pointer = 5,
10933   // GCC reserves 6 for references, but appears to never use it (because
10934   // expressions never have reference type, presumably).
10935   PointerToDataMember = 7,
10936   RealFloat = 8,
10937   Complex = 9,
10938   // GCC reserves 10 for functions, but does not use it since GCC version 6 due
10939   // to decay to pointer. (Prior to version 6 it was only used in C++ mode).
10940   // GCC claims to reserve 11 for pointers to member functions, but *actually*
10941   // uses 12 for that purpose, same as for a class or struct. Maybe it
10942   // internally implements a pointer to member as a struct?  Who knows.
10943   PointerToMemberFunction = 12, // Not a bug, see above.
10944   ClassOrStruct = 12,
10945   Union = 13,
10946   // GCC reserves 14 for arrays, but does not use it since GCC version 6 due to
10947   // decay to pointer. (Prior to version 6 it was only used in C++ mode).
10948   // GCC reserves 15 for strings, but actually uses 5 (pointer) for string
10949   // literals.
10950 };
10951 
10952 /// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
10953 /// as GCC.
10954 static GCCTypeClass
EvaluateBuiltinClassifyType(QualType T,const LangOptions & LangOpts)10955 EvaluateBuiltinClassifyType(QualType T, const LangOptions &LangOpts) {
10956   assert(!T->isDependentType() && "unexpected dependent type");
10957 
10958   QualType CanTy = T.getCanonicalType();
10959   const BuiltinType *BT = dyn_cast<BuiltinType>(CanTy);
10960 
10961   switch (CanTy->getTypeClass()) {
10962 #define TYPE(ID, BASE)
10963 #define DEPENDENT_TYPE(ID, BASE) case Type::ID:
10964 #define NON_CANONICAL_TYPE(ID, BASE) case Type::ID:
10965 #define NON_CANONICAL_UNLESS_DEPENDENT_TYPE(ID, BASE) case Type::ID:
10966 #include "clang/AST/TypeNodes.inc"
10967   case Type::Auto:
10968   case Type::DeducedTemplateSpecialization:
10969       llvm_unreachable("unexpected non-canonical or dependent type");
10970 
10971   case Type::Builtin:
10972     switch (BT->getKind()) {
10973 #define BUILTIN_TYPE(ID, SINGLETON_ID)
10974 #define SIGNED_TYPE(ID, SINGLETON_ID) \
10975     case BuiltinType::ID: return GCCTypeClass::Integer;
10976 #define FLOATING_TYPE(ID, SINGLETON_ID) \
10977     case BuiltinType::ID: return GCCTypeClass::RealFloat;
10978 #define PLACEHOLDER_TYPE(ID, SINGLETON_ID) \
10979     case BuiltinType::ID: break;
10980 #include "clang/AST/BuiltinTypes.def"
10981     case BuiltinType::Void:
10982       return GCCTypeClass::Void;
10983 
10984     case BuiltinType::Bool:
10985       return GCCTypeClass::Bool;
10986 
10987     case BuiltinType::Char_U:
10988     case BuiltinType::UChar:
10989     case BuiltinType::WChar_U:
10990     case BuiltinType::Char8:
10991     case BuiltinType::Char16:
10992     case BuiltinType::Char32:
10993     case BuiltinType::UShort:
10994     case BuiltinType::UInt:
10995     case BuiltinType::ULong:
10996     case BuiltinType::ULongLong:
10997     case BuiltinType::UInt128:
10998       return GCCTypeClass::Integer;
10999 
11000     case BuiltinType::UShortAccum:
11001     case BuiltinType::UAccum:
11002     case BuiltinType::ULongAccum:
11003     case BuiltinType::UShortFract:
11004     case BuiltinType::UFract:
11005     case BuiltinType::ULongFract:
11006     case BuiltinType::SatUShortAccum:
11007     case BuiltinType::SatUAccum:
11008     case BuiltinType::SatULongAccum:
11009     case BuiltinType::SatUShortFract:
11010     case BuiltinType::SatUFract:
11011     case BuiltinType::SatULongFract:
11012       return GCCTypeClass::None;
11013 
11014     case BuiltinType::NullPtr:
11015 
11016     case BuiltinType::ObjCId:
11017     case BuiltinType::ObjCClass:
11018     case BuiltinType::ObjCSel:
11019 #define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
11020     case BuiltinType::Id:
11021 #include "clang/Basic/OpenCLImageTypes.def"
11022 #define EXT_OPAQUE_TYPE(ExtType, Id, Ext) \
11023     case BuiltinType::Id:
11024 #include "clang/Basic/OpenCLExtensionTypes.def"
11025     case BuiltinType::OCLSampler:
11026     case BuiltinType::OCLEvent:
11027     case BuiltinType::OCLClkEvent:
11028     case BuiltinType::OCLQueue:
11029     case BuiltinType::OCLReserveID:
11030 #define SVE_TYPE(Name, Id, SingletonId) \
11031     case BuiltinType::Id:
11032 #include "clang/Basic/AArch64SVEACLETypes.def"
11033 #define PPC_VECTOR_TYPE(Name, Id, Size) \
11034     case BuiltinType::Id:
11035 #include "clang/Basic/PPCTypes.def"
11036 #define RVV_TYPE(Name, Id, SingletonId) case BuiltinType::Id:
11037 #include "clang/Basic/RISCVVTypes.def"
11038       return GCCTypeClass::None;
11039 
11040     case BuiltinType::Dependent:
11041       llvm_unreachable("unexpected dependent type");
11042     };
11043     llvm_unreachable("unexpected placeholder type");
11044 
11045   case Type::Enum:
11046     return LangOpts.CPlusPlus ? GCCTypeClass::Enum : GCCTypeClass::Integer;
11047 
11048   case Type::Pointer:
11049   case Type::ConstantArray:
11050   case Type::VariableArray:
11051   case Type::IncompleteArray:
11052   case Type::FunctionNoProto:
11053   case Type::FunctionProto:
11054     return GCCTypeClass::Pointer;
11055 
11056   case Type::MemberPointer:
11057     return CanTy->isMemberDataPointerType()
11058                ? GCCTypeClass::PointerToDataMember
11059                : GCCTypeClass::PointerToMemberFunction;
11060 
11061   case Type::Complex:
11062     return GCCTypeClass::Complex;
11063 
11064   case Type::Record:
11065     return CanTy->isUnionType() ? GCCTypeClass::Union
11066                                 : GCCTypeClass::ClassOrStruct;
11067 
11068   case Type::Atomic:
11069     // GCC classifies _Atomic T the same as T.
11070     return EvaluateBuiltinClassifyType(
11071         CanTy->castAs<AtomicType>()->getValueType(), LangOpts);
11072 
11073   case Type::BlockPointer:
11074   case Type::Vector:
11075   case Type::ExtVector:
11076   case Type::ConstantMatrix:
11077   case Type::ObjCObject:
11078   case Type::ObjCInterface:
11079   case Type::ObjCObjectPointer:
11080   case Type::Pipe:
11081   case Type::ExtInt:
11082     // GCC classifies vectors as None. We follow its lead and classify all
11083     // other types that don't fit into the regular classification the same way.
11084     return GCCTypeClass::None;
11085 
11086   case Type::LValueReference:
11087   case Type::RValueReference:
11088     llvm_unreachable("invalid type for expression");
11089   }
11090 
11091   llvm_unreachable("unexpected type class");
11092 }
11093 
11094 /// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
11095 /// as GCC.
11096 static GCCTypeClass
EvaluateBuiltinClassifyType(const CallExpr * E,const LangOptions & LangOpts)11097 EvaluateBuiltinClassifyType(const CallExpr *E, const LangOptions &LangOpts) {
11098   // If no argument was supplied, default to None. This isn't
11099   // ideal, however it is what gcc does.
11100   if (E->getNumArgs() == 0)
11101     return GCCTypeClass::None;
11102 
11103   // FIXME: Bizarrely, GCC treats a call with more than one argument as not
11104   // being an ICE, but still folds it to a constant using the type of the first
11105   // argument.
11106   return EvaluateBuiltinClassifyType(E->getArg(0)->getType(), LangOpts);
11107 }
11108 
11109 /// EvaluateBuiltinConstantPForLValue - Determine the result of
11110 /// __builtin_constant_p when applied to the given pointer.
11111 ///
11112 /// A pointer is only "constant" if it is null (or a pointer cast to integer)
11113 /// or it points to the first character of a string literal.
EvaluateBuiltinConstantPForLValue(const APValue & LV)11114 static bool EvaluateBuiltinConstantPForLValue(const APValue &LV) {
11115   APValue::LValueBase Base = LV.getLValueBase();
11116   if (Base.isNull()) {
11117     // A null base is acceptable.
11118     return true;
11119   } else if (const Expr *E = Base.dyn_cast<const Expr *>()) {
11120     if (!isa<StringLiteral>(E))
11121       return false;
11122     return LV.getLValueOffset().isZero();
11123   } else if (Base.is<TypeInfoLValue>()) {
11124     // Surprisingly, GCC considers __builtin_constant_p(&typeid(int)) to
11125     // evaluate to true.
11126     return true;
11127   } else {
11128     // Any other base is not constant enough for GCC.
11129     return false;
11130   }
11131 }
11132 
11133 /// EvaluateBuiltinConstantP - Evaluate __builtin_constant_p as similarly to
11134 /// GCC as we can manage.
EvaluateBuiltinConstantP(EvalInfo & Info,const Expr * Arg)11135 static bool EvaluateBuiltinConstantP(EvalInfo &Info, const Expr *Arg) {
11136   // This evaluation is not permitted to have side-effects, so evaluate it in
11137   // a speculative evaluation context.
11138   SpeculativeEvaluationRAII SpeculativeEval(Info);
11139 
11140   // Constant-folding is always enabled for the operand of __builtin_constant_p
11141   // (even when the enclosing evaluation context otherwise requires a strict
11142   // language-specific constant expression).
11143   FoldConstant Fold(Info, true);
11144 
11145   QualType ArgType = Arg->getType();
11146 
11147   // __builtin_constant_p always has one operand. The rules which gcc follows
11148   // are not precisely documented, but are as follows:
11149   //
11150   //  - If the operand is of integral, floating, complex or enumeration type,
11151   //    and can be folded to a known value of that type, it returns 1.
11152   //  - If the operand can be folded to a pointer to the first character
11153   //    of a string literal (or such a pointer cast to an integral type)
11154   //    or to a null pointer or an integer cast to a pointer, it returns 1.
11155   //
11156   // Otherwise, it returns 0.
11157   //
11158   // FIXME: GCC also intends to return 1 for literals of aggregate types, but
11159   // its support for this did not work prior to GCC 9 and is not yet well
11160   // understood.
11161   if (ArgType->isIntegralOrEnumerationType() || ArgType->isFloatingType() ||
11162       ArgType->isAnyComplexType() || ArgType->isPointerType() ||
11163       ArgType->isNullPtrType()) {
11164     APValue V;
11165     if (!::EvaluateAsRValue(Info, Arg, V) || Info.EvalStatus.HasSideEffects) {
11166       Fold.keepDiagnostics();
11167       return false;
11168     }
11169 
11170     // For a pointer (possibly cast to integer), there are special rules.
11171     if (V.getKind() == APValue::LValue)
11172       return EvaluateBuiltinConstantPForLValue(V);
11173 
11174     // Otherwise, any constant value is good enough.
11175     return V.hasValue();
11176   }
11177 
11178   // Anything else isn't considered to be sufficiently constant.
11179   return false;
11180 }
11181 
11182 /// Retrieves the "underlying object type" of the given expression,
11183 /// as used by __builtin_object_size.
getObjectType(APValue::LValueBase B)11184 static QualType getObjectType(APValue::LValueBase B) {
11185   if (const ValueDecl *D = B.dyn_cast<const ValueDecl*>()) {
11186     if (const VarDecl *VD = dyn_cast<VarDecl>(D))
11187       return VD->getType();
11188   } else if (const Expr *E = B.dyn_cast<const Expr*>()) {
11189     if (isa<CompoundLiteralExpr>(E))
11190       return E->getType();
11191   } else if (B.is<TypeInfoLValue>()) {
11192     return B.getTypeInfoType();
11193   } else if (B.is<DynamicAllocLValue>()) {
11194     return B.getDynamicAllocType();
11195   }
11196 
11197   return QualType();
11198 }
11199 
11200 /// A more selective version of E->IgnoreParenCasts for
11201 /// tryEvaluateBuiltinObjectSize. This ignores some casts/parens that serve only
11202 /// to change the type of E.
11203 /// Ex. For E = `(short*)((char*)(&foo))`, returns `&foo`
11204 ///
11205 /// Always returns an RValue with a pointer representation.
ignorePointerCastsAndParens(const Expr * E)11206 static const Expr *ignorePointerCastsAndParens(const Expr *E) {
11207   assert(E->isPRValue() && E->getType()->hasPointerRepresentation());
11208 
11209   auto *NoParens = E->IgnoreParens();
11210   auto *Cast = dyn_cast<CastExpr>(NoParens);
11211   if (Cast == nullptr)
11212     return NoParens;
11213 
11214   // We only conservatively allow a few kinds of casts, because this code is
11215   // inherently a simple solution that seeks to support the common case.
11216   auto CastKind = Cast->getCastKind();
11217   if (CastKind != CK_NoOp && CastKind != CK_BitCast &&
11218       CastKind != CK_AddressSpaceConversion)
11219     return NoParens;
11220 
11221   auto *SubExpr = Cast->getSubExpr();
11222   if (!SubExpr->getType()->hasPointerRepresentation() || !SubExpr->isPRValue())
11223     return NoParens;
11224   return ignorePointerCastsAndParens(SubExpr);
11225 }
11226 
11227 /// Checks to see if the given LValue's Designator is at the end of the LValue's
11228 /// record layout. e.g.
11229 ///   struct { struct { int a, b; } fst, snd; } obj;
11230 ///   obj.fst   // no
11231 ///   obj.snd   // yes
11232 ///   obj.fst.a // no
11233 ///   obj.fst.b // no
11234 ///   obj.snd.a // no
11235 ///   obj.snd.b // yes
11236 ///
11237 /// Please note: this function is specialized for how __builtin_object_size
11238 /// views "objects".
11239 ///
11240 /// If this encounters an invalid RecordDecl or otherwise cannot determine the
11241 /// correct result, it will always return true.
isDesignatorAtObjectEnd(const ASTContext & Ctx,const LValue & LVal)11242 static bool isDesignatorAtObjectEnd(const ASTContext &Ctx, const LValue &LVal) {
11243   assert(!LVal.Designator.Invalid);
11244 
11245   auto IsLastOrInvalidFieldDecl = [&Ctx](const FieldDecl *FD, bool &Invalid) {
11246     const RecordDecl *Parent = FD->getParent();
11247     Invalid = Parent->isInvalidDecl();
11248     if (Invalid || Parent->isUnion())
11249       return true;
11250     const ASTRecordLayout &Layout = Ctx.getASTRecordLayout(Parent);
11251     return FD->getFieldIndex() + 1 == Layout.getFieldCount();
11252   };
11253 
11254   auto &Base = LVal.getLValueBase();
11255   if (auto *ME = dyn_cast_or_null<MemberExpr>(Base.dyn_cast<const Expr *>())) {
11256     if (auto *FD = dyn_cast<FieldDecl>(ME->getMemberDecl())) {
11257       bool Invalid;
11258       if (!IsLastOrInvalidFieldDecl(FD, Invalid))
11259         return Invalid;
11260     } else if (auto *IFD = dyn_cast<IndirectFieldDecl>(ME->getMemberDecl())) {
11261       for (auto *FD : IFD->chain()) {
11262         bool Invalid;
11263         if (!IsLastOrInvalidFieldDecl(cast<FieldDecl>(FD), Invalid))
11264           return Invalid;
11265       }
11266     }
11267   }
11268 
11269   unsigned I = 0;
11270   QualType BaseType = getType(Base);
11271   if (LVal.Designator.FirstEntryIsAnUnsizedArray) {
11272     // If we don't know the array bound, conservatively assume we're looking at
11273     // the final array element.
11274     ++I;
11275     if (BaseType->isIncompleteArrayType())
11276       BaseType = Ctx.getAsArrayType(BaseType)->getElementType();
11277     else
11278       BaseType = BaseType->castAs<PointerType>()->getPointeeType();
11279   }
11280 
11281   for (unsigned E = LVal.Designator.Entries.size(); I != E; ++I) {
11282     const auto &Entry = LVal.Designator.Entries[I];
11283     if (BaseType->isArrayType()) {
11284       // Because __builtin_object_size treats arrays as objects, we can ignore
11285       // the index iff this is the last array in the Designator.
11286       if (I + 1 == E)
11287         return true;
11288       const auto *CAT = cast<ConstantArrayType>(Ctx.getAsArrayType(BaseType));
11289       uint64_t Index = Entry.getAsArrayIndex();
11290       if (Index + 1 != CAT->getSize())
11291         return false;
11292       BaseType = CAT->getElementType();
11293     } else if (BaseType->isAnyComplexType()) {
11294       const auto *CT = BaseType->castAs<ComplexType>();
11295       uint64_t Index = Entry.getAsArrayIndex();
11296       if (Index != 1)
11297         return false;
11298       BaseType = CT->getElementType();
11299     } else if (auto *FD = getAsField(Entry)) {
11300       bool Invalid;
11301       if (!IsLastOrInvalidFieldDecl(FD, Invalid))
11302         return Invalid;
11303       BaseType = FD->getType();
11304     } else {
11305       assert(getAsBaseClass(Entry) && "Expecting cast to a base class");
11306       return false;
11307     }
11308   }
11309   return true;
11310 }
11311 
11312 /// Tests to see if the LValue has a user-specified designator (that isn't
11313 /// necessarily valid). Note that this always returns 'true' if the LValue has
11314 /// an unsized array as its first designator entry, because there's currently no
11315 /// way to tell if the user typed *foo or foo[0].
refersToCompleteObject(const LValue & LVal)11316 static bool refersToCompleteObject(const LValue &LVal) {
11317   if (LVal.Designator.Invalid)
11318     return false;
11319 
11320   if (!LVal.Designator.Entries.empty())
11321     return LVal.Designator.isMostDerivedAnUnsizedArray();
11322 
11323   if (!LVal.InvalidBase)
11324     return true;
11325 
11326   // If `E` is a MemberExpr, then the first part of the designator is hiding in
11327   // the LValueBase.
11328   const auto *E = LVal.Base.dyn_cast<const Expr *>();
11329   return !E || !isa<MemberExpr>(E);
11330 }
11331 
11332 /// Attempts to detect a user writing into a piece of memory that's impossible
11333 /// to figure out the size of by just using types.
isUserWritingOffTheEnd(const ASTContext & Ctx,const LValue & LVal)11334 static bool isUserWritingOffTheEnd(const ASTContext &Ctx, const LValue &LVal) {
11335   const SubobjectDesignator &Designator = LVal.Designator;
11336   // Notes:
11337   // - Users can only write off of the end when we have an invalid base. Invalid
11338   //   bases imply we don't know where the memory came from.
11339   // - We used to be a bit more aggressive here; we'd only be conservative if
11340   //   the array at the end was flexible, or if it had 0 or 1 elements. This
11341   //   broke some common standard library extensions (PR30346), but was
11342   //   otherwise seemingly fine. It may be useful to reintroduce this behavior
11343   //   with some sort of list. OTOH, it seems that GCC is always
11344   //   conservative with the last element in structs (if it's an array), so our
11345   //   current behavior is more compatible than an explicit list approach would
11346   //   be.
11347   return LVal.InvalidBase &&
11348          Designator.Entries.size() == Designator.MostDerivedPathLength &&
11349          Designator.MostDerivedIsArrayElement &&
11350          isDesignatorAtObjectEnd(Ctx, LVal);
11351 }
11352 
11353 /// Converts the given APInt to CharUnits, assuming the APInt is unsigned.
11354 /// Fails if the conversion would cause loss of precision.
convertUnsignedAPIntToCharUnits(const llvm::APInt & Int,CharUnits & Result)11355 static bool convertUnsignedAPIntToCharUnits(const llvm::APInt &Int,
11356                                             CharUnits &Result) {
11357   auto CharUnitsMax = std::numeric_limits<CharUnits::QuantityType>::max();
11358   if (Int.ugt(CharUnitsMax))
11359     return false;
11360   Result = CharUnits::fromQuantity(Int.getZExtValue());
11361   return true;
11362 }
11363 
11364 /// Helper for tryEvaluateBuiltinObjectSize -- Given an LValue, this will
11365 /// determine how many bytes exist from the beginning of the object to either
11366 /// the end of the current subobject, or the end of the object itself, depending
11367 /// on what the LValue looks like + the value of Type.
11368 ///
11369 /// If this returns false, the value of Result is undefined.
determineEndOffset(EvalInfo & Info,SourceLocation ExprLoc,unsigned Type,const LValue & LVal,CharUnits & EndOffset)11370 static bool determineEndOffset(EvalInfo &Info, SourceLocation ExprLoc,
11371                                unsigned Type, const LValue &LVal,
11372                                CharUnits &EndOffset) {
11373   bool DetermineForCompleteObject = refersToCompleteObject(LVal);
11374 
11375   auto CheckedHandleSizeof = [&](QualType Ty, CharUnits &Result) {
11376     if (Ty.isNull() || Ty->isIncompleteType() || Ty->isFunctionType())
11377       return false;
11378     return HandleSizeof(Info, ExprLoc, Ty, Result);
11379   };
11380 
11381   // We want to evaluate the size of the entire object. This is a valid fallback
11382   // for when Type=1 and the designator is invalid, because we're asked for an
11383   // upper-bound.
11384   if (!(Type & 1) || LVal.Designator.Invalid || DetermineForCompleteObject) {
11385     // Type=3 wants a lower bound, so we can't fall back to this.
11386     if (Type == 3 && !DetermineForCompleteObject)
11387       return false;
11388 
11389     llvm::APInt APEndOffset;
11390     if (isBaseAnAllocSizeCall(LVal.getLValueBase()) &&
11391         getBytesReturnedByAllocSizeCall(Info.Ctx, LVal, APEndOffset))
11392       return convertUnsignedAPIntToCharUnits(APEndOffset, EndOffset);
11393 
11394     if (LVal.InvalidBase)
11395       return false;
11396 
11397     QualType BaseTy = getObjectType(LVal.getLValueBase());
11398     return CheckedHandleSizeof(BaseTy, EndOffset);
11399   }
11400 
11401   // We want to evaluate the size of a subobject.
11402   const SubobjectDesignator &Designator = LVal.Designator;
11403 
11404   // The following is a moderately common idiom in C:
11405   //
11406   // struct Foo { int a; char c[1]; };
11407   // struct Foo *F = (struct Foo *)malloc(sizeof(struct Foo) + strlen(Bar));
11408   // strcpy(&F->c[0], Bar);
11409   //
11410   // In order to not break too much legacy code, we need to support it.
11411   if (isUserWritingOffTheEnd(Info.Ctx, LVal)) {
11412     // If we can resolve this to an alloc_size call, we can hand that back,
11413     // because we know for certain how many bytes there are to write to.
11414     llvm::APInt APEndOffset;
11415     if (isBaseAnAllocSizeCall(LVal.getLValueBase()) &&
11416         getBytesReturnedByAllocSizeCall(Info.Ctx, LVal, APEndOffset))
11417       return convertUnsignedAPIntToCharUnits(APEndOffset, EndOffset);
11418 
11419     // If we cannot determine the size of the initial allocation, then we can't
11420     // given an accurate upper-bound. However, we are still able to give
11421     // conservative lower-bounds for Type=3.
11422     if (Type == 1)
11423       return false;
11424   }
11425 
11426   CharUnits BytesPerElem;
11427   if (!CheckedHandleSizeof(Designator.MostDerivedType, BytesPerElem))
11428     return false;
11429 
11430   // According to the GCC documentation, we want the size of the subobject
11431   // denoted by the pointer. But that's not quite right -- what we actually
11432   // want is the size of the immediately-enclosing array, if there is one.
11433   int64_t ElemsRemaining;
11434   if (Designator.MostDerivedIsArrayElement &&
11435       Designator.Entries.size() == Designator.MostDerivedPathLength) {
11436     uint64_t ArraySize = Designator.getMostDerivedArraySize();
11437     uint64_t ArrayIndex = Designator.Entries.back().getAsArrayIndex();
11438     ElemsRemaining = ArraySize <= ArrayIndex ? 0 : ArraySize - ArrayIndex;
11439   } else {
11440     ElemsRemaining = Designator.isOnePastTheEnd() ? 0 : 1;
11441   }
11442 
11443   EndOffset = LVal.getLValueOffset() + BytesPerElem * ElemsRemaining;
11444   return true;
11445 }
11446 
11447 /// Tries to evaluate the __builtin_object_size for @p E. If successful,
11448 /// returns true and stores the result in @p Size.
11449 ///
11450 /// If @p WasError is non-null, this will report whether the failure to evaluate
11451 /// is to be treated as an Error in IntExprEvaluator.
tryEvaluateBuiltinObjectSize(const Expr * E,unsigned Type,EvalInfo & Info,uint64_t & Size)11452 static bool tryEvaluateBuiltinObjectSize(const Expr *E, unsigned Type,
11453                                          EvalInfo &Info, uint64_t &Size) {
11454   // Determine the denoted object.
11455   LValue LVal;
11456   {
11457     // The operand of __builtin_object_size is never evaluated for side-effects.
11458     // If there are any, but we can determine the pointed-to object anyway, then
11459     // ignore the side-effects.
11460     SpeculativeEvaluationRAII SpeculativeEval(Info);
11461     IgnoreSideEffectsRAII Fold(Info);
11462 
11463     if (E->isGLValue()) {
11464       // It's possible for us to be given GLValues if we're called via
11465       // Expr::tryEvaluateObjectSize.
11466       APValue RVal;
11467       if (!EvaluateAsRValue(Info, E, RVal))
11468         return false;
11469       LVal.setFrom(Info.Ctx, RVal);
11470     } else if (!EvaluatePointer(ignorePointerCastsAndParens(E), LVal, Info,
11471                                 /*InvalidBaseOK=*/true))
11472       return false;
11473   }
11474 
11475   // If we point to before the start of the object, there are no accessible
11476   // bytes.
11477   if (LVal.getLValueOffset().isNegative()) {
11478     Size = 0;
11479     return true;
11480   }
11481 
11482   CharUnits EndOffset;
11483   if (!determineEndOffset(Info, E->getExprLoc(), Type, LVal, EndOffset))
11484     return false;
11485 
11486   // If we've fallen outside of the end offset, just pretend there's nothing to
11487   // write to/read from.
11488   if (EndOffset <= LVal.getLValueOffset())
11489     Size = 0;
11490   else
11491     Size = (EndOffset - LVal.getLValueOffset()).getQuantity();
11492   return true;
11493 }
11494 
VisitCallExpr(const CallExpr * E)11495 bool IntExprEvaluator::VisitCallExpr(const CallExpr *E) {
11496   if (unsigned BuiltinOp = E->getBuiltinCallee())
11497     return VisitBuiltinCallExpr(E, BuiltinOp);
11498 
11499   return ExprEvaluatorBaseTy::VisitCallExpr(E);
11500 }
11501 
getBuiltinAlignArguments(const CallExpr * E,EvalInfo & Info,APValue & Val,APSInt & Alignment)11502 static bool getBuiltinAlignArguments(const CallExpr *E, EvalInfo &Info,
11503                                      APValue &Val, APSInt &Alignment) {
11504   QualType SrcTy = E->getArg(0)->getType();
11505   if (!getAlignmentArgument(E->getArg(1), SrcTy, Info, Alignment))
11506     return false;
11507   // Even though we are evaluating integer expressions we could get a pointer
11508   // argument for the __builtin_is_aligned() case.
11509   if (SrcTy->isPointerType()) {
11510     LValue Ptr;
11511     if (!EvaluatePointer(E->getArg(0), Ptr, Info))
11512       return false;
11513     Ptr.moveInto(Val);
11514   } else if (!SrcTy->isIntegralOrEnumerationType()) {
11515     Info.FFDiag(E->getArg(0));
11516     return false;
11517   } else {
11518     APSInt SrcInt;
11519     if (!EvaluateInteger(E->getArg(0), SrcInt, Info))
11520       return false;
11521     assert(SrcInt.getBitWidth() >= Alignment.getBitWidth() &&
11522            "Bit widths must be the same");
11523     Val = APValue(SrcInt);
11524   }
11525   assert(Val.hasValue());
11526   return true;
11527 }
11528 
VisitBuiltinCallExpr(const CallExpr * E,unsigned BuiltinOp)11529 bool IntExprEvaluator::VisitBuiltinCallExpr(const CallExpr *E,
11530                                             unsigned BuiltinOp) {
11531   switch (BuiltinOp) {
11532   default:
11533     return ExprEvaluatorBaseTy::VisitCallExpr(E);
11534 
11535   case Builtin::BI__builtin_dynamic_object_size:
11536   case Builtin::BI__builtin_object_size: {
11537     // The type was checked when we built the expression.
11538     unsigned Type =
11539         E->getArg(1)->EvaluateKnownConstInt(Info.Ctx).getZExtValue();
11540     assert(Type <= 3 && "unexpected type");
11541 
11542     uint64_t Size;
11543     if (tryEvaluateBuiltinObjectSize(E->getArg(0), Type, Info, Size))
11544       return Success(Size, E);
11545 
11546     if (E->getArg(0)->HasSideEffects(Info.Ctx))
11547       return Success((Type & 2) ? 0 : -1, E);
11548 
11549     // Expression had no side effects, but we couldn't statically determine the
11550     // size of the referenced object.
11551     switch (Info.EvalMode) {
11552     case EvalInfo::EM_ConstantExpression:
11553     case EvalInfo::EM_ConstantFold:
11554     case EvalInfo::EM_IgnoreSideEffects:
11555       // Leave it to IR generation.
11556       return Error(E);
11557     case EvalInfo::EM_ConstantExpressionUnevaluated:
11558       // Reduce it to a constant now.
11559       return Success((Type & 2) ? 0 : -1, E);
11560     }
11561 
11562     llvm_unreachable("unexpected EvalMode");
11563   }
11564 
11565   case Builtin::BI__builtin_os_log_format_buffer_size: {
11566     analyze_os_log::OSLogBufferLayout Layout;
11567     analyze_os_log::computeOSLogBufferLayout(Info.Ctx, E, Layout);
11568     return Success(Layout.size().getQuantity(), E);
11569   }
11570 
11571   case Builtin::BI__builtin_is_aligned: {
11572     APValue Src;
11573     APSInt Alignment;
11574     if (!getBuiltinAlignArguments(E, Info, Src, Alignment))
11575       return false;
11576     if (Src.isLValue()) {
11577       // If we evaluated a pointer, check the minimum known alignment.
11578       LValue Ptr;
11579       Ptr.setFrom(Info.Ctx, Src);
11580       CharUnits BaseAlignment = getBaseAlignment(Info, Ptr);
11581       CharUnits PtrAlign = BaseAlignment.alignmentAtOffset(Ptr.Offset);
11582       // We can return true if the known alignment at the computed offset is
11583       // greater than the requested alignment.
11584       assert(PtrAlign.isPowerOfTwo());
11585       assert(Alignment.isPowerOf2());
11586       if (PtrAlign.getQuantity() >= Alignment)
11587         return Success(1, E);
11588       // If the alignment is not known to be sufficient, some cases could still
11589       // be aligned at run time. However, if the requested alignment is less or
11590       // equal to the base alignment and the offset is not aligned, we know that
11591       // the run-time value can never be aligned.
11592       if (BaseAlignment.getQuantity() >= Alignment &&
11593           PtrAlign.getQuantity() < Alignment)
11594         return Success(0, E);
11595       // Otherwise we can't infer whether the value is sufficiently aligned.
11596       // TODO: __builtin_is_aligned(__builtin_align_{down,up{(expr, N), N)
11597       //  in cases where we can't fully evaluate the pointer.
11598       Info.FFDiag(E->getArg(0), diag::note_constexpr_alignment_compute)
11599           << Alignment;
11600       return false;
11601     }
11602     assert(Src.isInt());
11603     return Success((Src.getInt() & (Alignment - 1)) == 0 ? 1 : 0, E);
11604   }
11605   case Builtin::BI__builtin_align_up: {
11606     APValue Src;
11607     APSInt Alignment;
11608     if (!getBuiltinAlignArguments(E, Info, Src, Alignment))
11609       return false;
11610     if (!Src.isInt())
11611       return Error(E);
11612     APSInt AlignedVal =
11613         APSInt((Src.getInt() + (Alignment - 1)) & ~(Alignment - 1),
11614                Src.getInt().isUnsigned());
11615     assert(AlignedVal.getBitWidth() == Src.getInt().getBitWidth());
11616     return Success(AlignedVal, E);
11617   }
11618   case Builtin::BI__builtin_align_down: {
11619     APValue Src;
11620     APSInt Alignment;
11621     if (!getBuiltinAlignArguments(E, Info, Src, Alignment))
11622       return false;
11623     if (!Src.isInt())
11624       return Error(E);
11625     APSInt AlignedVal =
11626         APSInt(Src.getInt() & ~(Alignment - 1), Src.getInt().isUnsigned());
11627     assert(AlignedVal.getBitWidth() == Src.getInt().getBitWidth());
11628     return Success(AlignedVal, E);
11629   }
11630 
11631   case Builtin::BI__builtin_bitreverse8:
11632   case Builtin::BI__builtin_bitreverse16:
11633   case Builtin::BI__builtin_bitreverse32:
11634   case Builtin::BI__builtin_bitreverse64: {
11635     APSInt Val;
11636     if (!EvaluateInteger(E->getArg(0), Val, Info))
11637       return false;
11638 
11639     return Success(Val.reverseBits(), E);
11640   }
11641 
11642   case Builtin::BI__builtin_bswap16:
11643   case Builtin::BI__builtin_bswap32:
11644   case Builtin::BI__builtin_bswap64: {
11645     APSInt Val;
11646     if (!EvaluateInteger(E->getArg(0), Val, Info))
11647       return false;
11648 
11649     return Success(Val.byteSwap(), E);
11650   }
11651 
11652   case Builtin::BI__builtin_classify_type:
11653     return Success((int)EvaluateBuiltinClassifyType(E, Info.getLangOpts()), E);
11654 
11655   case Builtin::BI__builtin_clrsb:
11656   case Builtin::BI__builtin_clrsbl:
11657   case Builtin::BI__builtin_clrsbll: {
11658     APSInt Val;
11659     if (!EvaluateInteger(E->getArg(0), Val, Info))
11660       return false;
11661 
11662     return Success(Val.getBitWidth() - Val.getMinSignedBits(), E);
11663   }
11664 
11665   case Builtin::BI__builtin_clz:
11666   case Builtin::BI__builtin_clzl:
11667   case Builtin::BI__builtin_clzll:
11668   case Builtin::BI__builtin_clzs: {
11669     APSInt Val;
11670     if (!EvaluateInteger(E->getArg(0), Val, Info))
11671       return false;
11672     if (!Val)
11673       return Error(E);
11674 
11675     return Success(Val.countLeadingZeros(), E);
11676   }
11677 
11678   case Builtin::BI__builtin_constant_p: {
11679     const Expr *Arg = E->getArg(0);
11680     if (EvaluateBuiltinConstantP(Info, Arg))
11681       return Success(true, E);
11682     if (Info.InConstantContext || Arg->HasSideEffects(Info.Ctx)) {
11683       // Outside a constant context, eagerly evaluate to false in the presence
11684       // of side-effects in order to avoid -Wunsequenced false-positives in
11685       // a branch on __builtin_constant_p(expr).
11686       return Success(false, E);
11687     }
11688     Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
11689     return false;
11690   }
11691 
11692   case Builtin::BI__builtin_is_constant_evaluated: {
11693     const auto *Callee = Info.CurrentCall->getCallee();
11694     if (Info.InConstantContext && !Info.CheckingPotentialConstantExpression &&
11695         (Info.CallStackDepth == 1 ||
11696          (Info.CallStackDepth == 2 && Callee->isInStdNamespace() &&
11697           Callee->getIdentifier() &&
11698           Callee->getIdentifier()->isStr("is_constant_evaluated")))) {
11699       // FIXME: Find a better way to avoid duplicated diagnostics.
11700       if (Info.EvalStatus.Diag)
11701         Info.report((Info.CallStackDepth == 1) ? E->getExprLoc()
11702                                                : Info.CurrentCall->CallLoc,
11703                     diag::warn_is_constant_evaluated_always_true_constexpr)
11704             << (Info.CallStackDepth == 1 ? "__builtin_is_constant_evaluated"
11705                                          : "std::is_constant_evaluated");
11706     }
11707 
11708     return Success(Info.InConstantContext, E);
11709   }
11710 
11711   case Builtin::BI__builtin_ctz:
11712   case Builtin::BI__builtin_ctzl:
11713   case Builtin::BI__builtin_ctzll:
11714   case Builtin::BI__builtin_ctzs: {
11715     APSInt Val;
11716     if (!EvaluateInteger(E->getArg(0), Val, Info))
11717       return false;
11718     if (!Val)
11719       return Error(E);
11720 
11721     return Success(Val.countTrailingZeros(), E);
11722   }
11723 
11724   case Builtin::BI__builtin_eh_return_data_regno: {
11725     int Operand = E->getArg(0)->EvaluateKnownConstInt(Info.Ctx).getZExtValue();
11726     Operand = Info.Ctx.getTargetInfo().getEHDataRegisterNumber(Operand);
11727     return Success(Operand, E);
11728   }
11729 
11730   case Builtin::BI__builtin_expect:
11731   case Builtin::BI__builtin_expect_with_probability:
11732     return Visit(E->getArg(0));
11733 
11734   case Builtin::BI__builtin_ffs:
11735   case Builtin::BI__builtin_ffsl:
11736   case Builtin::BI__builtin_ffsll: {
11737     APSInt Val;
11738     if (!EvaluateInteger(E->getArg(0), Val, Info))
11739       return false;
11740 
11741     unsigned N = Val.countTrailingZeros();
11742     return Success(N == Val.getBitWidth() ? 0 : N + 1, E);
11743   }
11744 
11745   case Builtin::BI__builtin_fpclassify: {
11746     APFloat Val(0.0);
11747     if (!EvaluateFloat(E->getArg(5), Val, Info))
11748       return false;
11749     unsigned Arg;
11750     switch (Val.getCategory()) {
11751     case APFloat::fcNaN: Arg = 0; break;
11752     case APFloat::fcInfinity: Arg = 1; break;
11753     case APFloat::fcNormal: Arg = Val.isDenormal() ? 3 : 2; break;
11754     case APFloat::fcZero: Arg = 4; break;
11755     }
11756     return Visit(E->getArg(Arg));
11757   }
11758 
11759   case Builtin::BI__builtin_isinf_sign: {
11760     APFloat Val(0.0);
11761     return EvaluateFloat(E->getArg(0), Val, Info) &&
11762            Success(Val.isInfinity() ? (Val.isNegative() ? -1 : 1) : 0, E);
11763   }
11764 
11765   case Builtin::BI__builtin_isinf: {
11766     APFloat Val(0.0);
11767     return EvaluateFloat(E->getArg(0), Val, Info) &&
11768            Success(Val.isInfinity() ? 1 : 0, E);
11769   }
11770 
11771   case Builtin::BI__builtin_isfinite: {
11772     APFloat Val(0.0);
11773     return EvaluateFloat(E->getArg(0), Val, Info) &&
11774            Success(Val.isFinite() ? 1 : 0, E);
11775   }
11776 
11777   case Builtin::BI__builtin_isnan: {
11778     APFloat Val(0.0);
11779     return EvaluateFloat(E->getArg(0), Val, Info) &&
11780            Success(Val.isNaN() ? 1 : 0, E);
11781   }
11782 
11783   case Builtin::BI__builtin_isnormal: {
11784     APFloat Val(0.0);
11785     return EvaluateFloat(E->getArg(0), Val, Info) &&
11786            Success(Val.isNormal() ? 1 : 0, E);
11787   }
11788 
11789   case Builtin::BI__builtin_parity:
11790   case Builtin::BI__builtin_parityl:
11791   case Builtin::BI__builtin_parityll: {
11792     APSInt Val;
11793     if (!EvaluateInteger(E->getArg(0), Val, Info))
11794       return false;
11795 
11796     return Success(Val.countPopulation() % 2, E);
11797   }
11798 
11799   case Builtin::BI__builtin_popcount:
11800   case Builtin::BI__builtin_popcountl:
11801   case Builtin::BI__builtin_popcountll: {
11802     APSInt Val;
11803     if (!EvaluateInteger(E->getArg(0), Val, Info))
11804       return false;
11805 
11806     return Success(Val.countPopulation(), E);
11807   }
11808 
11809   case Builtin::BI__builtin_rotateleft8:
11810   case Builtin::BI__builtin_rotateleft16:
11811   case Builtin::BI__builtin_rotateleft32:
11812   case Builtin::BI__builtin_rotateleft64:
11813   case Builtin::BI_rotl8: // Microsoft variants of rotate right
11814   case Builtin::BI_rotl16:
11815   case Builtin::BI_rotl:
11816   case Builtin::BI_lrotl:
11817   case Builtin::BI_rotl64: {
11818     APSInt Val, Amt;
11819     if (!EvaluateInteger(E->getArg(0), Val, Info) ||
11820         !EvaluateInteger(E->getArg(1), Amt, Info))
11821       return false;
11822 
11823     return Success(Val.rotl(Amt.urem(Val.getBitWidth())), E);
11824   }
11825 
11826   case Builtin::BI__builtin_rotateright8:
11827   case Builtin::BI__builtin_rotateright16:
11828   case Builtin::BI__builtin_rotateright32:
11829   case Builtin::BI__builtin_rotateright64:
11830   case Builtin::BI_rotr8: // Microsoft variants of rotate right
11831   case Builtin::BI_rotr16:
11832   case Builtin::BI_rotr:
11833   case Builtin::BI_lrotr:
11834   case Builtin::BI_rotr64: {
11835     APSInt Val, Amt;
11836     if (!EvaluateInteger(E->getArg(0), Val, Info) ||
11837         !EvaluateInteger(E->getArg(1), Amt, Info))
11838       return false;
11839 
11840     return Success(Val.rotr(Amt.urem(Val.getBitWidth())), E);
11841   }
11842 
11843   case Builtin::BIstrlen:
11844   case Builtin::BIwcslen:
11845     // A call to strlen is not a constant expression.
11846     if (Info.getLangOpts().CPlusPlus11)
11847       Info.CCEDiag(E, diag::note_constexpr_invalid_function)
11848         << /*isConstexpr*/0 << /*isConstructor*/0
11849         << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'");
11850     else
11851       Info.CCEDiag(E, diag::note_invalid_subexpr_in_const_expr);
11852     LLVM_FALLTHROUGH;
11853   case Builtin::BI__builtin_strlen:
11854   case Builtin::BI__builtin_wcslen: {
11855     // As an extension, we support __builtin_strlen() as a constant expression,
11856     // and support folding strlen() to a constant.
11857     LValue String;
11858     if (!EvaluatePointer(E->getArg(0), String, Info))
11859       return false;
11860 
11861     QualType CharTy = E->getArg(0)->getType()->getPointeeType();
11862 
11863     // Fast path: if it's a string literal, search the string value.
11864     if (const StringLiteral *S = dyn_cast_or_null<StringLiteral>(
11865             String.getLValueBase().dyn_cast<const Expr *>())) {
11866       // The string literal may have embedded null characters. Find the first
11867       // one and truncate there.
11868       StringRef Str = S->getBytes();
11869       int64_t Off = String.Offset.getQuantity();
11870       if (Off >= 0 && (uint64_t)Off <= (uint64_t)Str.size() &&
11871           S->getCharByteWidth() == 1 &&
11872           // FIXME: Add fast-path for wchar_t too.
11873           Info.Ctx.hasSameUnqualifiedType(CharTy, Info.Ctx.CharTy)) {
11874         Str = Str.substr(Off);
11875 
11876         StringRef::size_type Pos = Str.find(0);
11877         if (Pos != StringRef::npos)
11878           Str = Str.substr(0, Pos);
11879 
11880         return Success(Str.size(), E);
11881       }
11882 
11883       // Fall through to slow path to issue appropriate diagnostic.
11884     }
11885 
11886     // Slow path: scan the bytes of the string looking for the terminating 0.
11887     for (uint64_t Strlen = 0; /**/; ++Strlen) {
11888       APValue Char;
11889       if (!handleLValueToRValueConversion(Info, E, CharTy, String, Char) ||
11890           !Char.isInt())
11891         return false;
11892       if (!Char.getInt())
11893         return Success(Strlen, E);
11894       if (!HandleLValueArrayAdjustment(Info, E, String, CharTy, 1))
11895         return false;
11896     }
11897   }
11898 
11899   case Builtin::BIstrcmp:
11900   case Builtin::BIwcscmp:
11901   case Builtin::BIstrncmp:
11902   case Builtin::BIwcsncmp:
11903   case Builtin::BImemcmp:
11904   case Builtin::BIbcmp:
11905   case Builtin::BIwmemcmp:
11906     // A call to strlen is not a constant expression.
11907     if (Info.getLangOpts().CPlusPlus11)
11908       Info.CCEDiag(E, diag::note_constexpr_invalid_function)
11909         << /*isConstexpr*/0 << /*isConstructor*/0
11910         << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'");
11911     else
11912       Info.CCEDiag(E, diag::note_invalid_subexpr_in_const_expr);
11913     LLVM_FALLTHROUGH;
11914   case Builtin::BI__builtin_strcmp:
11915   case Builtin::BI__builtin_wcscmp:
11916   case Builtin::BI__builtin_strncmp:
11917   case Builtin::BI__builtin_wcsncmp:
11918   case Builtin::BI__builtin_memcmp:
11919   case Builtin::BI__builtin_bcmp:
11920   case Builtin::BI__builtin_wmemcmp: {
11921     LValue String1, String2;
11922     if (!EvaluatePointer(E->getArg(0), String1, Info) ||
11923         !EvaluatePointer(E->getArg(1), String2, Info))
11924       return false;
11925 
11926     uint64_t MaxLength = uint64_t(-1);
11927     if (BuiltinOp != Builtin::BIstrcmp &&
11928         BuiltinOp != Builtin::BIwcscmp &&
11929         BuiltinOp != Builtin::BI__builtin_strcmp &&
11930         BuiltinOp != Builtin::BI__builtin_wcscmp) {
11931       APSInt N;
11932       if (!EvaluateInteger(E->getArg(2), N, Info))
11933         return false;
11934       MaxLength = N.getExtValue();
11935     }
11936 
11937     // Empty substrings compare equal by definition.
11938     if (MaxLength == 0u)
11939       return Success(0, E);
11940 
11941     if (!String1.checkNullPointerForFoldAccess(Info, E, AK_Read) ||
11942         !String2.checkNullPointerForFoldAccess(Info, E, AK_Read) ||
11943         String1.Designator.Invalid || String2.Designator.Invalid)
11944       return false;
11945 
11946     QualType CharTy1 = String1.Designator.getType(Info.Ctx);
11947     QualType CharTy2 = String2.Designator.getType(Info.Ctx);
11948 
11949     bool IsRawByte = BuiltinOp == Builtin::BImemcmp ||
11950                      BuiltinOp == Builtin::BIbcmp ||
11951                      BuiltinOp == Builtin::BI__builtin_memcmp ||
11952                      BuiltinOp == Builtin::BI__builtin_bcmp;
11953 
11954     assert(IsRawByte ||
11955            (Info.Ctx.hasSameUnqualifiedType(
11956                 CharTy1, E->getArg(0)->getType()->getPointeeType()) &&
11957             Info.Ctx.hasSameUnqualifiedType(CharTy1, CharTy2)));
11958 
11959     // For memcmp, allow comparing any arrays of '[[un]signed] char' or
11960     // 'char8_t', but no other types.
11961     if (IsRawByte &&
11962         !(isOneByteCharacterType(CharTy1) && isOneByteCharacterType(CharTy2))) {
11963       // FIXME: Consider using our bit_cast implementation to support this.
11964       Info.FFDiag(E, diag::note_constexpr_memcmp_unsupported)
11965           << (std::string("'") + Info.Ctx.BuiltinInfo.getName(BuiltinOp) + "'")
11966           << CharTy1 << CharTy2;
11967       return false;
11968     }
11969 
11970     const auto &ReadCurElems = [&](APValue &Char1, APValue &Char2) {
11971       return handleLValueToRValueConversion(Info, E, CharTy1, String1, Char1) &&
11972              handleLValueToRValueConversion(Info, E, CharTy2, String2, Char2) &&
11973              Char1.isInt() && Char2.isInt();
11974     };
11975     const auto &AdvanceElems = [&] {
11976       return HandleLValueArrayAdjustment(Info, E, String1, CharTy1, 1) &&
11977              HandleLValueArrayAdjustment(Info, E, String2, CharTy2, 1);
11978     };
11979 
11980     bool StopAtNull =
11981         (BuiltinOp != Builtin::BImemcmp && BuiltinOp != Builtin::BIbcmp &&
11982          BuiltinOp != Builtin::BIwmemcmp &&
11983          BuiltinOp != Builtin::BI__builtin_memcmp &&
11984          BuiltinOp != Builtin::BI__builtin_bcmp &&
11985          BuiltinOp != Builtin::BI__builtin_wmemcmp);
11986     bool IsWide = BuiltinOp == Builtin::BIwcscmp ||
11987                   BuiltinOp == Builtin::BIwcsncmp ||
11988                   BuiltinOp == Builtin::BIwmemcmp ||
11989                   BuiltinOp == Builtin::BI__builtin_wcscmp ||
11990                   BuiltinOp == Builtin::BI__builtin_wcsncmp ||
11991                   BuiltinOp == Builtin::BI__builtin_wmemcmp;
11992 
11993     for (; MaxLength; --MaxLength) {
11994       APValue Char1, Char2;
11995       if (!ReadCurElems(Char1, Char2))
11996         return false;
11997       if (Char1.getInt().ne(Char2.getInt())) {
11998         if (IsWide) // wmemcmp compares with wchar_t signedness.
11999           return Success(Char1.getInt() < Char2.getInt() ? -1 : 1, E);
12000         // memcmp always compares unsigned chars.
12001         return Success(Char1.getInt().ult(Char2.getInt()) ? -1 : 1, E);
12002       }
12003       if (StopAtNull && !Char1.getInt())
12004         return Success(0, E);
12005       assert(!(StopAtNull && !Char2.getInt()));
12006       if (!AdvanceElems())
12007         return false;
12008     }
12009     // We hit the strncmp / memcmp limit.
12010     return Success(0, E);
12011   }
12012 
12013   case Builtin::BI__atomic_always_lock_free:
12014   case Builtin::BI__atomic_is_lock_free:
12015   case Builtin::BI__c11_atomic_is_lock_free: {
12016     APSInt SizeVal;
12017     if (!EvaluateInteger(E->getArg(0), SizeVal, Info))
12018       return false;
12019 
12020     // For __atomic_is_lock_free(sizeof(_Atomic(T))), if the size is a power
12021     // of two less than or equal to the maximum inline atomic width, we know it
12022     // is lock-free.  If the size isn't a power of two, or greater than the
12023     // maximum alignment where we promote atomics, we know it is not lock-free
12024     // (at least not in the sense of atomic_is_lock_free).  Otherwise,
12025     // the answer can only be determined at runtime; for example, 16-byte
12026     // atomics have lock-free implementations on some, but not all,
12027     // x86-64 processors.
12028 
12029     // Check power-of-two.
12030     CharUnits Size = CharUnits::fromQuantity(SizeVal.getZExtValue());
12031     if (Size.isPowerOfTwo()) {
12032       // Check against inlining width.
12033       unsigned InlineWidthBits =
12034           Info.Ctx.getTargetInfo().getMaxAtomicInlineWidth();
12035       if (Size <= Info.Ctx.toCharUnitsFromBits(InlineWidthBits)) {
12036         if (BuiltinOp == Builtin::BI__c11_atomic_is_lock_free ||
12037             Size == CharUnits::One() ||
12038             E->getArg(1)->isNullPointerConstant(Info.Ctx,
12039                                                 Expr::NPC_NeverValueDependent))
12040           // OK, we will inline appropriately-aligned operations of this size,
12041           // and _Atomic(T) is appropriately-aligned.
12042           return Success(1, E);
12043 
12044         QualType PointeeType = E->getArg(1)->IgnoreImpCasts()->getType()->
12045           castAs<PointerType>()->getPointeeType();
12046         if (!PointeeType->isIncompleteType() &&
12047             Info.Ctx.getTypeAlignInChars(PointeeType) >= Size) {
12048           // OK, we will inline operations on this object.
12049           return Success(1, E);
12050         }
12051       }
12052     }
12053 
12054     return BuiltinOp == Builtin::BI__atomic_always_lock_free ?
12055         Success(0, E) : Error(E);
12056   }
12057   case Builtin::BI__builtin_add_overflow:
12058   case Builtin::BI__builtin_sub_overflow:
12059   case Builtin::BI__builtin_mul_overflow:
12060   case Builtin::BI__builtin_sadd_overflow:
12061   case Builtin::BI__builtin_uadd_overflow:
12062   case Builtin::BI__builtin_uaddl_overflow:
12063   case Builtin::BI__builtin_uaddll_overflow:
12064   case Builtin::BI__builtin_usub_overflow:
12065   case Builtin::BI__builtin_usubl_overflow:
12066   case Builtin::BI__builtin_usubll_overflow:
12067   case Builtin::BI__builtin_umul_overflow:
12068   case Builtin::BI__builtin_umull_overflow:
12069   case Builtin::BI__builtin_umulll_overflow:
12070   case Builtin::BI__builtin_saddl_overflow:
12071   case Builtin::BI__builtin_saddll_overflow:
12072   case Builtin::BI__builtin_ssub_overflow:
12073   case Builtin::BI__builtin_ssubl_overflow:
12074   case Builtin::BI__builtin_ssubll_overflow:
12075   case Builtin::BI__builtin_smul_overflow:
12076   case Builtin::BI__builtin_smull_overflow:
12077   case Builtin::BI__builtin_smulll_overflow: {
12078     LValue ResultLValue;
12079     APSInt LHS, RHS;
12080 
12081     QualType ResultType = E->getArg(2)->getType()->getPointeeType();
12082     if (!EvaluateInteger(E->getArg(0), LHS, Info) ||
12083         !EvaluateInteger(E->getArg(1), RHS, Info) ||
12084         !EvaluatePointer(E->getArg(2), ResultLValue, Info))
12085       return false;
12086 
12087     APSInt Result;
12088     bool DidOverflow = false;
12089 
12090     // If the types don't have to match, enlarge all 3 to the largest of them.
12091     if (BuiltinOp == Builtin::BI__builtin_add_overflow ||
12092         BuiltinOp == Builtin::BI__builtin_sub_overflow ||
12093         BuiltinOp == Builtin::BI__builtin_mul_overflow) {
12094       bool IsSigned = LHS.isSigned() || RHS.isSigned() ||
12095                       ResultType->isSignedIntegerOrEnumerationType();
12096       bool AllSigned = LHS.isSigned() && RHS.isSigned() &&
12097                       ResultType->isSignedIntegerOrEnumerationType();
12098       uint64_t LHSSize = LHS.getBitWidth();
12099       uint64_t RHSSize = RHS.getBitWidth();
12100       uint64_t ResultSize = Info.Ctx.getTypeSize(ResultType);
12101       uint64_t MaxBits = std::max(std::max(LHSSize, RHSSize), ResultSize);
12102 
12103       // Add an additional bit if the signedness isn't uniformly agreed to. We
12104       // could do this ONLY if there is a signed and an unsigned that both have
12105       // MaxBits, but the code to check that is pretty nasty.  The issue will be
12106       // caught in the shrink-to-result later anyway.
12107       if (IsSigned && !AllSigned)
12108         ++MaxBits;
12109 
12110       LHS = APSInt(LHS.extOrTrunc(MaxBits), !IsSigned);
12111       RHS = APSInt(RHS.extOrTrunc(MaxBits), !IsSigned);
12112       Result = APSInt(MaxBits, !IsSigned);
12113     }
12114 
12115     // Find largest int.
12116     switch (BuiltinOp) {
12117     default:
12118       llvm_unreachable("Invalid value for BuiltinOp");
12119     case Builtin::BI__builtin_add_overflow:
12120     case Builtin::BI__builtin_sadd_overflow:
12121     case Builtin::BI__builtin_saddl_overflow:
12122     case Builtin::BI__builtin_saddll_overflow:
12123     case Builtin::BI__builtin_uadd_overflow:
12124     case Builtin::BI__builtin_uaddl_overflow:
12125     case Builtin::BI__builtin_uaddll_overflow:
12126       Result = LHS.isSigned() ? LHS.sadd_ov(RHS, DidOverflow)
12127                               : LHS.uadd_ov(RHS, DidOverflow);
12128       break;
12129     case Builtin::BI__builtin_sub_overflow:
12130     case Builtin::BI__builtin_ssub_overflow:
12131     case Builtin::BI__builtin_ssubl_overflow:
12132     case Builtin::BI__builtin_ssubll_overflow:
12133     case Builtin::BI__builtin_usub_overflow:
12134     case Builtin::BI__builtin_usubl_overflow:
12135     case Builtin::BI__builtin_usubll_overflow:
12136       Result = LHS.isSigned() ? LHS.ssub_ov(RHS, DidOverflow)
12137                               : LHS.usub_ov(RHS, DidOverflow);
12138       break;
12139     case Builtin::BI__builtin_mul_overflow:
12140     case Builtin::BI__builtin_smul_overflow:
12141     case Builtin::BI__builtin_smull_overflow:
12142     case Builtin::BI__builtin_smulll_overflow:
12143     case Builtin::BI__builtin_umul_overflow:
12144     case Builtin::BI__builtin_umull_overflow:
12145     case Builtin::BI__builtin_umulll_overflow:
12146       Result = LHS.isSigned() ? LHS.smul_ov(RHS, DidOverflow)
12147                               : LHS.umul_ov(RHS, DidOverflow);
12148       break;
12149     }
12150 
12151     // In the case where multiple sizes are allowed, truncate and see if
12152     // the values are the same.
12153     if (BuiltinOp == Builtin::BI__builtin_add_overflow ||
12154         BuiltinOp == Builtin::BI__builtin_sub_overflow ||
12155         BuiltinOp == Builtin::BI__builtin_mul_overflow) {
12156       // APSInt doesn't have a TruncOrSelf, so we use extOrTrunc instead,
12157       // since it will give us the behavior of a TruncOrSelf in the case where
12158       // its parameter <= its size.  We previously set Result to be at least the
12159       // type-size of the result, so getTypeSize(ResultType) <= Result.BitWidth
12160       // will work exactly like TruncOrSelf.
12161       APSInt Temp = Result.extOrTrunc(Info.Ctx.getTypeSize(ResultType));
12162       Temp.setIsSigned(ResultType->isSignedIntegerOrEnumerationType());
12163 
12164       if (!APSInt::isSameValue(Temp, Result))
12165         DidOverflow = true;
12166       Result = Temp;
12167     }
12168 
12169     APValue APV{Result};
12170     if (!handleAssignment(Info, E, ResultLValue, ResultType, APV))
12171       return false;
12172     return Success(DidOverflow, E);
12173   }
12174   }
12175 }
12176 
12177 /// Determine whether this is a pointer past the end of the complete
12178 /// object referred to by the lvalue.
isOnePastTheEndOfCompleteObject(const ASTContext & Ctx,const LValue & LV)12179 static bool isOnePastTheEndOfCompleteObject(const ASTContext &Ctx,
12180                                             const LValue &LV) {
12181   // A null pointer can be viewed as being "past the end" but we don't
12182   // choose to look at it that way here.
12183   if (!LV.getLValueBase())
12184     return false;
12185 
12186   // If the designator is valid and refers to a subobject, we're not pointing
12187   // past the end.
12188   if (!LV.getLValueDesignator().Invalid &&
12189       !LV.getLValueDesignator().isOnePastTheEnd())
12190     return false;
12191 
12192   // A pointer to an incomplete type might be past-the-end if the type's size is
12193   // zero.  We cannot tell because the type is incomplete.
12194   QualType Ty = getType(LV.getLValueBase());
12195   if (Ty->isIncompleteType())
12196     return true;
12197 
12198   // We're a past-the-end pointer if we point to the byte after the object,
12199   // no matter what our type or path is.
12200   auto Size = Ctx.getTypeSizeInChars(Ty);
12201   return LV.getLValueOffset() == Size;
12202 }
12203 
12204 namespace {
12205 
12206 /// Data recursive integer evaluator of certain binary operators.
12207 ///
12208 /// We use a data recursive algorithm for binary operators so that we are able
12209 /// to handle extreme cases of chained binary operators without causing stack
12210 /// overflow.
12211 class DataRecursiveIntBinOpEvaluator {
12212   struct EvalResult {
12213     APValue Val;
12214     bool Failed;
12215 
EvalResult__anonb66d72d22811::DataRecursiveIntBinOpEvaluator::EvalResult12216     EvalResult() : Failed(false) { }
12217 
swap__anonb66d72d22811::DataRecursiveIntBinOpEvaluator::EvalResult12218     void swap(EvalResult &RHS) {
12219       Val.swap(RHS.Val);
12220       Failed = RHS.Failed;
12221       RHS.Failed = false;
12222     }
12223   };
12224 
12225   struct Job {
12226     const Expr *E;
12227     EvalResult LHSResult; // meaningful only for binary operator expression.
12228     enum { AnyExprKind, BinOpKind, BinOpVisitedLHSKind } Kind;
12229 
12230     Job() = default;
12231     Job(Job &&) = default;
12232 
startSpeculativeEval__anonb66d72d22811::DataRecursiveIntBinOpEvaluator::Job12233     void startSpeculativeEval(EvalInfo &Info) {
12234       SpecEvalRAII = SpeculativeEvaluationRAII(Info);
12235     }
12236 
12237   private:
12238     SpeculativeEvaluationRAII SpecEvalRAII;
12239   };
12240 
12241   SmallVector<Job, 16> Queue;
12242 
12243   IntExprEvaluator &IntEval;
12244   EvalInfo &Info;
12245   APValue &FinalResult;
12246 
12247 public:
DataRecursiveIntBinOpEvaluator(IntExprEvaluator & IntEval,APValue & Result)12248   DataRecursiveIntBinOpEvaluator(IntExprEvaluator &IntEval, APValue &Result)
12249     : IntEval(IntEval), Info(IntEval.getEvalInfo()), FinalResult(Result) { }
12250 
12251   /// True if \param E is a binary operator that we are going to handle
12252   /// data recursively.
12253   /// We handle binary operators that are comma, logical, or that have operands
12254   /// with integral or enumeration type.
shouldEnqueue(const BinaryOperator * E)12255   static bool shouldEnqueue(const BinaryOperator *E) {
12256     return E->getOpcode() == BO_Comma || E->isLogicalOp() ||
12257            (E->isPRValue() && E->getType()->isIntegralOrEnumerationType() &&
12258             E->getLHS()->getType()->isIntegralOrEnumerationType() &&
12259             E->getRHS()->getType()->isIntegralOrEnumerationType());
12260   }
12261 
Traverse(const BinaryOperator * E)12262   bool Traverse(const BinaryOperator *E) {
12263     enqueue(E);
12264     EvalResult PrevResult;
12265     while (!Queue.empty())
12266       process(PrevResult);
12267 
12268     if (PrevResult.Failed) return false;
12269 
12270     FinalResult.swap(PrevResult.Val);
12271     return true;
12272   }
12273 
12274 private:
Success(uint64_t Value,const Expr * E,APValue & Result)12275   bool Success(uint64_t Value, const Expr *E, APValue &Result) {
12276     return IntEval.Success(Value, E, Result);
12277   }
Success(const APSInt & Value,const Expr * E,APValue & Result)12278   bool Success(const APSInt &Value, const Expr *E, APValue &Result) {
12279     return IntEval.Success(Value, E, Result);
12280   }
Error(const Expr * E)12281   bool Error(const Expr *E) {
12282     return IntEval.Error(E);
12283   }
Error(const Expr * E,diag::kind D)12284   bool Error(const Expr *E, diag::kind D) {
12285     return IntEval.Error(E, D);
12286   }
12287 
CCEDiag(const Expr * E,diag::kind D)12288   OptionalDiagnostic CCEDiag(const Expr *E, diag::kind D) {
12289     return Info.CCEDiag(E, D);
12290   }
12291 
12292   // Returns true if visiting the RHS is necessary, false otherwise.
12293   bool VisitBinOpLHSOnly(EvalResult &LHSResult, const BinaryOperator *E,
12294                          bool &SuppressRHSDiags);
12295 
12296   bool VisitBinOp(const EvalResult &LHSResult, const EvalResult &RHSResult,
12297                   const BinaryOperator *E, APValue &Result);
12298 
EvaluateExpr(const Expr * E,EvalResult & Result)12299   void EvaluateExpr(const Expr *E, EvalResult &Result) {
12300     Result.Failed = !Evaluate(Result.Val, Info, E);
12301     if (Result.Failed)
12302       Result.Val = APValue();
12303   }
12304 
12305   void process(EvalResult &Result);
12306 
enqueue(const Expr * E)12307   void enqueue(const Expr *E) {
12308     E = E->IgnoreParens();
12309     Queue.resize(Queue.size()+1);
12310     Queue.back().E = E;
12311     Queue.back().Kind = Job::AnyExprKind;
12312   }
12313 };
12314 
12315 }
12316 
12317 bool DataRecursiveIntBinOpEvaluator::
VisitBinOpLHSOnly(EvalResult & LHSResult,const BinaryOperator * E,bool & SuppressRHSDiags)12318        VisitBinOpLHSOnly(EvalResult &LHSResult, const BinaryOperator *E,
12319                          bool &SuppressRHSDiags) {
12320   if (E->getOpcode() == BO_Comma) {
12321     // Ignore LHS but note if we could not evaluate it.
12322     if (LHSResult.Failed)
12323       return Info.noteSideEffect();
12324     return true;
12325   }
12326 
12327   if (E->isLogicalOp()) {
12328     bool LHSAsBool;
12329     if (!LHSResult.Failed && HandleConversionToBool(LHSResult.Val, LHSAsBool)) {
12330       // We were able to evaluate the LHS, see if we can get away with not
12331       // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
12332       if (LHSAsBool == (E->getOpcode() == BO_LOr)) {
12333         Success(LHSAsBool, E, LHSResult.Val);
12334         return false; // Ignore RHS
12335       }
12336     } else {
12337       LHSResult.Failed = true;
12338 
12339       // Since we weren't able to evaluate the left hand side, it
12340       // might have had side effects.
12341       if (!Info.noteSideEffect())
12342         return false;
12343 
12344       // We can't evaluate the LHS; however, sometimes the result
12345       // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
12346       // Don't ignore RHS and suppress diagnostics from this arm.
12347       SuppressRHSDiags = true;
12348     }
12349 
12350     return true;
12351   }
12352 
12353   assert(E->getLHS()->getType()->isIntegralOrEnumerationType() &&
12354          E->getRHS()->getType()->isIntegralOrEnumerationType());
12355 
12356   if (LHSResult.Failed && !Info.noteFailure())
12357     return false; // Ignore RHS;
12358 
12359   return true;
12360 }
12361 
addOrSubLValueAsInteger(APValue & LVal,const APSInt & Index,bool IsSub)12362 static void addOrSubLValueAsInteger(APValue &LVal, const APSInt &Index,
12363                                     bool IsSub) {
12364   // Compute the new offset in the appropriate width, wrapping at 64 bits.
12365   // FIXME: When compiling for a 32-bit target, we should use 32-bit
12366   // offsets.
12367   assert(!LVal.hasLValuePath() && "have designator for integer lvalue");
12368   CharUnits &Offset = LVal.getLValueOffset();
12369   uint64_t Offset64 = Offset.getQuantity();
12370   uint64_t Index64 = Index.extOrTrunc(64).getZExtValue();
12371   Offset = CharUnits::fromQuantity(IsSub ? Offset64 - Index64
12372                                          : Offset64 + Index64);
12373 }
12374 
12375 bool DataRecursiveIntBinOpEvaluator::
VisitBinOp(const EvalResult & LHSResult,const EvalResult & RHSResult,const BinaryOperator * E,APValue & Result)12376        VisitBinOp(const EvalResult &LHSResult, const EvalResult &RHSResult,
12377                   const BinaryOperator *E, APValue &Result) {
12378   if (E->getOpcode() == BO_Comma) {
12379     if (RHSResult.Failed)
12380       return false;
12381     Result = RHSResult.Val;
12382     return true;
12383   }
12384 
12385   if (E->isLogicalOp()) {
12386     bool lhsResult, rhsResult;
12387     bool LHSIsOK = HandleConversionToBool(LHSResult.Val, lhsResult);
12388     bool RHSIsOK = HandleConversionToBool(RHSResult.Val, rhsResult);
12389 
12390     if (LHSIsOK) {
12391       if (RHSIsOK) {
12392         if (E->getOpcode() == BO_LOr)
12393           return Success(lhsResult || rhsResult, E, Result);
12394         else
12395           return Success(lhsResult && rhsResult, E, Result);
12396       }
12397     } else {
12398       if (RHSIsOK) {
12399         // We can't evaluate the LHS; however, sometimes the result
12400         // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
12401         if (rhsResult == (E->getOpcode() == BO_LOr))
12402           return Success(rhsResult, E, Result);
12403       }
12404     }
12405 
12406     return false;
12407   }
12408 
12409   assert(E->getLHS()->getType()->isIntegralOrEnumerationType() &&
12410          E->getRHS()->getType()->isIntegralOrEnumerationType());
12411 
12412   if (LHSResult.Failed || RHSResult.Failed)
12413     return false;
12414 
12415   const APValue &LHSVal = LHSResult.Val;
12416   const APValue &RHSVal = RHSResult.Val;
12417 
12418   // Handle cases like (unsigned long)&a + 4.
12419   if (E->isAdditiveOp() && LHSVal.isLValue() && RHSVal.isInt()) {
12420     Result = LHSVal;
12421     addOrSubLValueAsInteger(Result, RHSVal.getInt(), E->getOpcode() == BO_Sub);
12422     return true;
12423   }
12424 
12425   // Handle cases like 4 + (unsigned long)&a
12426   if (E->getOpcode() == BO_Add &&
12427       RHSVal.isLValue() && LHSVal.isInt()) {
12428     Result = RHSVal;
12429     addOrSubLValueAsInteger(Result, LHSVal.getInt(), /*IsSub*/false);
12430     return true;
12431   }
12432 
12433   if (E->getOpcode() == BO_Sub && LHSVal.isLValue() && RHSVal.isLValue()) {
12434     // Handle (intptr_t)&&A - (intptr_t)&&B.
12435     if (!LHSVal.getLValueOffset().isZero() ||
12436         !RHSVal.getLValueOffset().isZero())
12437       return false;
12438     const Expr *LHSExpr = LHSVal.getLValueBase().dyn_cast<const Expr*>();
12439     const Expr *RHSExpr = RHSVal.getLValueBase().dyn_cast<const Expr*>();
12440     if (!LHSExpr || !RHSExpr)
12441       return false;
12442     const AddrLabelExpr *LHSAddrExpr = dyn_cast<AddrLabelExpr>(LHSExpr);
12443     const AddrLabelExpr *RHSAddrExpr = dyn_cast<AddrLabelExpr>(RHSExpr);
12444     if (!LHSAddrExpr || !RHSAddrExpr)
12445       return false;
12446     // Make sure both labels come from the same function.
12447     if (LHSAddrExpr->getLabel()->getDeclContext() !=
12448         RHSAddrExpr->getLabel()->getDeclContext())
12449       return false;
12450     Result = APValue(LHSAddrExpr, RHSAddrExpr);
12451     return true;
12452   }
12453 
12454   // All the remaining cases expect both operands to be an integer
12455   if (!LHSVal.isInt() || !RHSVal.isInt())
12456     return Error(E);
12457 
12458   // Set up the width and signedness manually, in case it can't be deduced
12459   // from the operation we're performing.
12460   // FIXME: Don't do this in the cases where we can deduce it.
12461   APSInt Value(Info.Ctx.getIntWidth(E->getType()),
12462                E->getType()->isUnsignedIntegerOrEnumerationType());
12463   if (!handleIntIntBinOp(Info, E, LHSVal.getInt(), E->getOpcode(),
12464                          RHSVal.getInt(), Value))
12465     return false;
12466   return Success(Value, E, Result);
12467 }
12468 
process(EvalResult & Result)12469 void DataRecursiveIntBinOpEvaluator::process(EvalResult &Result) {
12470   Job &job = Queue.back();
12471 
12472   switch (job.Kind) {
12473     case Job::AnyExprKind: {
12474       if (const BinaryOperator *Bop = dyn_cast<BinaryOperator>(job.E)) {
12475         if (shouldEnqueue(Bop)) {
12476           job.Kind = Job::BinOpKind;
12477           enqueue(Bop->getLHS());
12478           return;
12479         }
12480       }
12481 
12482       EvaluateExpr(job.E, Result);
12483       Queue.pop_back();
12484       return;
12485     }
12486 
12487     case Job::BinOpKind: {
12488       const BinaryOperator *Bop = cast<BinaryOperator>(job.E);
12489       bool SuppressRHSDiags = false;
12490       if (!VisitBinOpLHSOnly(Result, Bop, SuppressRHSDiags)) {
12491         Queue.pop_back();
12492         return;
12493       }
12494       if (SuppressRHSDiags)
12495         job.startSpeculativeEval(Info);
12496       job.LHSResult.swap(Result);
12497       job.Kind = Job::BinOpVisitedLHSKind;
12498       enqueue(Bop->getRHS());
12499       return;
12500     }
12501 
12502     case Job::BinOpVisitedLHSKind: {
12503       const BinaryOperator *Bop = cast<BinaryOperator>(job.E);
12504       EvalResult RHS;
12505       RHS.swap(Result);
12506       Result.Failed = !VisitBinOp(job.LHSResult, RHS, Bop, Result.Val);
12507       Queue.pop_back();
12508       return;
12509     }
12510   }
12511 
12512   llvm_unreachable("Invalid Job::Kind!");
12513 }
12514 
12515 namespace {
12516 enum class CmpResult {
12517   Unequal,
12518   Less,
12519   Equal,
12520   Greater,
12521   Unordered,
12522 };
12523 }
12524 
12525 template <class SuccessCB, class AfterCB>
12526 static bool
EvaluateComparisonBinaryOperator(EvalInfo & Info,const BinaryOperator * E,SuccessCB && Success,AfterCB && DoAfter)12527 EvaluateComparisonBinaryOperator(EvalInfo &Info, const BinaryOperator *E,
12528                                  SuccessCB &&Success, AfterCB &&DoAfter) {
12529   assert(!E->isValueDependent());
12530   assert(E->isComparisonOp() && "expected comparison operator");
12531   assert((E->getOpcode() == BO_Cmp ||
12532           E->getType()->isIntegralOrEnumerationType()) &&
12533          "unsupported binary expression evaluation");
12534   auto Error = [&](const Expr *E) {
12535     Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
12536     return false;
12537   };
12538 
12539   bool IsRelational = E->isRelationalOp() || E->getOpcode() == BO_Cmp;
12540   bool IsEquality = E->isEqualityOp();
12541 
12542   QualType LHSTy = E->getLHS()->getType();
12543   QualType RHSTy = E->getRHS()->getType();
12544 
12545   if (LHSTy->isIntegralOrEnumerationType() &&
12546       RHSTy->isIntegralOrEnumerationType()) {
12547     APSInt LHS, RHS;
12548     bool LHSOK = EvaluateInteger(E->getLHS(), LHS, Info);
12549     if (!LHSOK && !Info.noteFailure())
12550       return false;
12551     if (!EvaluateInteger(E->getRHS(), RHS, Info) || !LHSOK)
12552       return false;
12553     if (LHS < RHS)
12554       return Success(CmpResult::Less, E);
12555     if (LHS > RHS)
12556       return Success(CmpResult::Greater, E);
12557     return Success(CmpResult::Equal, E);
12558   }
12559 
12560   if (LHSTy->isFixedPointType() || RHSTy->isFixedPointType()) {
12561     APFixedPoint LHSFX(Info.Ctx.getFixedPointSemantics(LHSTy));
12562     APFixedPoint RHSFX(Info.Ctx.getFixedPointSemantics(RHSTy));
12563 
12564     bool LHSOK = EvaluateFixedPointOrInteger(E->getLHS(), LHSFX, Info);
12565     if (!LHSOK && !Info.noteFailure())
12566       return false;
12567     if (!EvaluateFixedPointOrInteger(E->getRHS(), RHSFX, Info) || !LHSOK)
12568       return false;
12569     if (LHSFX < RHSFX)
12570       return Success(CmpResult::Less, E);
12571     if (LHSFX > RHSFX)
12572       return Success(CmpResult::Greater, E);
12573     return Success(CmpResult::Equal, E);
12574   }
12575 
12576   if (LHSTy->isAnyComplexType() || RHSTy->isAnyComplexType()) {
12577     ComplexValue LHS, RHS;
12578     bool LHSOK;
12579     if (E->isAssignmentOp()) {
12580       LValue LV;
12581       EvaluateLValue(E->getLHS(), LV, Info);
12582       LHSOK = false;
12583     } else if (LHSTy->isRealFloatingType()) {
12584       LHSOK = EvaluateFloat(E->getLHS(), LHS.FloatReal, Info);
12585       if (LHSOK) {
12586         LHS.makeComplexFloat();
12587         LHS.FloatImag = APFloat(LHS.FloatReal.getSemantics());
12588       }
12589     } else {
12590       LHSOK = EvaluateComplex(E->getLHS(), LHS, Info);
12591     }
12592     if (!LHSOK && !Info.noteFailure())
12593       return false;
12594 
12595     if (E->getRHS()->getType()->isRealFloatingType()) {
12596       if (!EvaluateFloat(E->getRHS(), RHS.FloatReal, Info) || !LHSOK)
12597         return false;
12598       RHS.makeComplexFloat();
12599       RHS.FloatImag = APFloat(RHS.FloatReal.getSemantics());
12600     } else if (!EvaluateComplex(E->getRHS(), RHS, Info) || !LHSOK)
12601       return false;
12602 
12603     if (LHS.isComplexFloat()) {
12604       APFloat::cmpResult CR_r =
12605         LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
12606       APFloat::cmpResult CR_i =
12607         LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
12608       bool IsEqual = CR_r == APFloat::cmpEqual && CR_i == APFloat::cmpEqual;
12609       return Success(IsEqual ? CmpResult::Equal : CmpResult::Unequal, E);
12610     } else {
12611       assert(IsEquality && "invalid complex comparison");
12612       bool IsEqual = LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
12613                      LHS.getComplexIntImag() == RHS.getComplexIntImag();
12614       return Success(IsEqual ? CmpResult::Equal : CmpResult::Unequal, E);
12615     }
12616   }
12617 
12618   if (LHSTy->isRealFloatingType() &&
12619       RHSTy->isRealFloatingType()) {
12620     APFloat RHS(0.0), LHS(0.0);
12621 
12622     bool LHSOK = EvaluateFloat(E->getRHS(), RHS, Info);
12623     if (!LHSOK && !Info.noteFailure())
12624       return false;
12625 
12626     if (!EvaluateFloat(E->getLHS(), LHS, Info) || !LHSOK)
12627       return false;
12628 
12629     assert(E->isComparisonOp() && "Invalid binary operator!");
12630     llvm::APFloatBase::cmpResult APFloatCmpResult = LHS.compare(RHS);
12631     if (!Info.InConstantContext &&
12632         APFloatCmpResult == APFloat::cmpUnordered &&
12633         E->getFPFeaturesInEffect(Info.Ctx.getLangOpts()).isFPConstrained()) {
12634       // Note: Compares may raise invalid in some cases involving NaN or sNaN.
12635       Info.FFDiag(E, diag::note_constexpr_float_arithmetic_strict);
12636       return false;
12637     }
12638     auto GetCmpRes = [&]() {
12639       switch (APFloatCmpResult) {
12640       case APFloat::cmpEqual:
12641         return CmpResult::Equal;
12642       case APFloat::cmpLessThan:
12643         return CmpResult::Less;
12644       case APFloat::cmpGreaterThan:
12645         return CmpResult::Greater;
12646       case APFloat::cmpUnordered:
12647         return CmpResult::Unordered;
12648       }
12649       llvm_unreachable("Unrecognised APFloat::cmpResult enum");
12650     };
12651     return Success(GetCmpRes(), E);
12652   }
12653 
12654   if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
12655     LValue LHSValue, RHSValue;
12656 
12657     bool LHSOK = EvaluatePointer(E->getLHS(), LHSValue, Info);
12658     if (!LHSOK && !Info.noteFailure())
12659       return false;
12660 
12661     if (!EvaluatePointer(E->getRHS(), RHSValue, Info) || !LHSOK)
12662       return false;
12663 
12664     // Reject differing bases from the normal codepath; we special-case
12665     // comparisons to null.
12666     if (!HasSameBase(LHSValue, RHSValue)) {
12667       // Inequalities and subtractions between unrelated pointers have
12668       // unspecified or undefined behavior.
12669       if (!IsEquality) {
12670         Info.FFDiag(E, diag::note_constexpr_pointer_comparison_unspecified);
12671         return false;
12672       }
12673       // A constant address may compare equal to the address of a symbol.
12674       // The one exception is that address of an object cannot compare equal
12675       // to a null pointer constant.
12676       if ((!LHSValue.Base && !LHSValue.Offset.isZero()) ||
12677           (!RHSValue.Base && !RHSValue.Offset.isZero()))
12678         return Error(E);
12679       // It's implementation-defined whether distinct literals will have
12680       // distinct addresses. In clang, the result of such a comparison is
12681       // unspecified, so it is not a constant expression. However, we do know
12682       // that the address of a literal will be non-null.
12683       if ((IsLiteralLValue(LHSValue) || IsLiteralLValue(RHSValue)) &&
12684           LHSValue.Base && RHSValue.Base)
12685         return Error(E);
12686       // We can't tell whether weak symbols will end up pointing to the same
12687       // object.
12688       if (IsWeakLValue(LHSValue) || IsWeakLValue(RHSValue))
12689         return Error(E);
12690       // We can't compare the address of the start of one object with the
12691       // past-the-end address of another object, per C++ DR1652.
12692       if ((LHSValue.Base && LHSValue.Offset.isZero() &&
12693            isOnePastTheEndOfCompleteObject(Info.Ctx, RHSValue)) ||
12694           (RHSValue.Base && RHSValue.Offset.isZero() &&
12695            isOnePastTheEndOfCompleteObject(Info.Ctx, LHSValue)))
12696         return Error(E);
12697       // We can't tell whether an object is at the same address as another
12698       // zero sized object.
12699       if ((RHSValue.Base && isZeroSized(LHSValue)) ||
12700           (LHSValue.Base && isZeroSized(RHSValue)))
12701         return Error(E);
12702       return Success(CmpResult::Unequal, E);
12703     }
12704 
12705     const CharUnits &LHSOffset = LHSValue.getLValueOffset();
12706     const CharUnits &RHSOffset = RHSValue.getLValueOffset();
12707 
12708     SubobjectDesignator &LHSDesignator = LHSValue.getLValueDesignator();
12709     SubobjectDesignator &RHSDesignator = RHSValue.getLValueDesignator();
12710 
12711     // C++11 [expr.rel]p3:
12712     //   Pointers to void (after pointer conversions) can be compared, with a
12713     //   result defined as follows: If both pointers represent the same
12714     //   address or are both the null pointer value, the result is true if the
12715     //   operator is <= or >= and false otherwise; otherwise the result is
12716     //   unspecified.
12717     // We interpret this as applying to pointers to *cv* void.
12718     if (LHSTy->isVoidPointerType() && LHSOffset != RHSOffset && IsRelational)
12719       Info.CCEDiag(E, diag::note_constexpr_void_comparison);
12720 
12721     // C++11 [expr.rel]p2:
12722     // - If two pointers point to non-static data members of the same object,
12723     //   or to subobjects or array elements fo such members, recursively, the
12724     //   pointer to the later declared member compares greater provided the
12725     //   two members have the same access control and provided their class is
12726     //   not a union.
12727     //   [...]
12728     // - Otherwise pointer comparisons are unspecified.
12729     if (!LHSDesignator.Invalid && !RHSDesignator.Invalid && IsRelational) {
12730       bool WasArrayIndex;
12731       unsigned Mismatch = FindDesignatorMismatch(
12732           getType(LHSValue.Base), LHSDesignator, RHSDesignator, WasArrayIndex);
12733       // At the point where the designators diverge, the comparison has a
12734       // specified value if:
12735       //  - we are comparing array indices
12736       //  - we are comparing fields of a union, or fields with the same access
12737       // Otherwise, the result is unspecified and thus the comparison is not a
12738       // constant expression.
12739       if (!WasArrayIndex && Mismatch < LHSDesignator.Entries.size() &&
12740           Mismatch < RHSDesignator.Entries.size()) {
12741         const FieldDecl *LF = getAsField(LHSDesignator.Entries[Mismatch]);
12742         const FieldDecl *RF = getAsField(RHSDesignator.Entries[Mismatch]);
12743         if (!LF && !RF)
12744           Info.CCEDiag(E, diag::note_constexpr_pointer_comparison_base_classes);
12745         else if (!LF)
12746           Info.CCEDiag(E, diag::note_constexpr_pointer_comparison_base_field)
12747               << getAsBaseClass(LHSDesignator.Entries[Mismatch])
12748               << RF->getParent() << RF;
12749         else if (!RF)
12750           Info.CCEDiag(E, diag::note_constexpr_pointer_comparison_base_field)
12751               << getAsBaseClass(RHSDesignator.Entries[Mismatch])
12752               << LF->getParent() << LF;
12753         else if (!LF->getParent()->isUnion() &&
12754                  LF->getAccess() != RF->getAccess())
12755           Info.CCEDiag(E,
12756                        diag::note_constexpr_pointer_comparison_differing_access)
12757               << LF << LF->getAccess() << RF << RF->getAccess()
12758               << LF->getParent();
12759       }
12760     }
12761 
12762     // The comparison here must be unsigned, and performed with the same
12763     // width as the pointer.
12764     unsigned PtrSize = Info.Ctx.getTypeSize(LHSTy);
12765     uint64_t CompareLHS = LHSOffset.getQuantity();
12766     uint64_t CompareRHS = RHSOffset.getQuantity();
12767     assert(PtrSize <= 64 && "Unexpected pointer width");
12768     uint64_t Mask = ~0ULL >> (64 - PtrSize);
12769     CompareLHS &= Mask;
12770     CompareRHS &= Mask;
12771 
12772     // If there is a base and this is a relational operator, we can only
12773     // compare pointers within the object in question; otherwise, the result
12774     // depends on where the object is located in memory.
12775     if (!LHSValue.Base.isNull() && IsRelational) {
12776       QualType BaseTy = getType(LHSValue.Base);
12777       if (BaseTy->isIncompleteType())
12778         return Error(E);
12779       CharUnits Size = Info.Ctx.getTypeSizeInChars(BaseTy);
12780       uint64_t OffsetLimit = Size.getQuantity();
12781       if (CompareLHS > OffsetLimit || CompareRHS > OffsetLimit)
12782         return Error(E);
12783     }
12784 
12785     if (CompareLHS < CompareRHS)
12786       return Success(CmpResult::Less, E);
12787     if (CompareLHS > CompareRHS)
12788       return Success(CmpResult::Greater, E);
12789     return Success(CmpResult::Equal, E);
12790   }
12791 
12792   if (LHSTy->isMemberPointerType()) {
12793     assert(IsEquality && "unexpected member pointer operation");
12794     assert(RHSTy->isMemberPointerType() && "invalid comparison");
12795 
12796     MemberPtr LHSValue, RHSValue;
12797 
12798     bool LHSOK = EvaluateMemberPointer(E->getLHS(), LHSValue, Info);
12799     if (!LHSOK && !Info.noteFailure())
12800       return false;
12801 
12802     if (!EvaluateMemberPointer(E->getRHS(), RHSValue, Info) || !LHSOK)
12803       return false;
12804 
12805     // C++11 [expr.eq]p2:
12806     //   If both operands are null, they compare equal. Otherwise if only one is
12807     //   null, they compare unequal.
12808     if (!LHSValue.getDecl() || !RHSValue.getDecl()) {
12809       bool Equal = !LHSValue.getDecl() && !RHSValue.getDecl();
12810       return Success(Equal ? CmpResult::Equal : CmpResult::Unequal, E);
12811     }
12812 
12813     //   Otherwise if either is a pointer to a virtual member function, the
12814     //   result is unspecified.
12815     if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(LHSValue.getDecl()))
12816       if (MD->isVirtual())
12817         Info.CCEDiag(E, diag::note_constexpr_compare_virtual_mem_ptr) << MD;
12818     if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(RHSValue.getDecl()))
12819       if (MD->isVirtual())
12820         Info.CCEDiag(E, diag::note_constexpr_compare_virtual_mem_ptr) << MD;
12821 
12822     //   Otherwise they compare equal if and only if they would refer to the
12823     //   same member of the same most derived object or the same subobject if
12824     //   they were dereferenced with a hypothetical object of the associated
12825     //   class type.
12826     bool Equal = LHSValue == RHSValue;
12827     return Success(Equal ? CmpResult::Equal : CmpResult::Unequal, E);
12828   }
12829 
12830   if (LHSTy->isNullPtrType()) {
12831     assert(E->isComparisonOp() && "unexpected nullptr operation");
12832     assert(RHSTy->isNullPtrType() && "missing pointer conversion");
12833     // C++11 [expr.rel]p4, [expr.eq]p3: If two operands of type std::nullptr_t
12834     // are compared, the result is true of the operator is <=, >= or ==, and
12835     // false otherwise.
12836     return Success(CmpResult::Equal, E);
12837   }
12838 
12839   return DoAfter();
12840 }
12841 
VisitBinCmp(const BinaryOperator * E)12842 bool RecordExprEvaluator::VisitBinCmp(const BinaryOperator *E) {
12843   if (!CheckLiteralType(Info, E))
12844     return false;
12845 
12846   auto OnSuccess = [&](CmpResult CR, const BinaryOperator *E) {
12847     ComparisonCategoryResult CCR;
12848     switch (CR) {
12849     case CmpResult::Unequal:
12850       llvm_unreachable("should never produce Unequal for three-way comparison");
12851     case CmpResult::Less:
12852       CCR = ComparisonCategoryResult::Less;
12853       break;
12854     case CmpResult::Equal:
12855       CCR = ComparisonCategoryResult::Equal;
12856       break;
12857     case CmpResult::Greater:
12858       CCR = ComparisonCategoryResult::Greater;
12859       break;
12860     case CmpResult::Unordered:
12861       CCR = ComparisonCategoryResult::Unordered;
12862       break;
12863     }
12864     // Evaluation succeeded. Lookup the information for the comparison category
12865     // type and fetch the VarDecl for the result.
12866     const ComparisonCategoryInfo &CmpInfo =
12867         Info.Ctx.CompCategories.getInfoForType(E->getType());
12868     const VarDecl *VD = CmpInfo.getValueInfo(CmpInfo.makeWeakResult(CCR))->VD;
12869     // Check and evaluate the result as a constant expression.
12870     LValue LV;
12871     LV.set(VD);
12872     if (!handleLValueToRValueConversion(Info, E, E->getType(), LV, Result))
12873       return false;
12874     return CheckConstantExpression(Info, E->getExprLoc(), E->getType(), Result,
12875                                    ConstantExprKind::Normal);
12876   };
12877   return EvaluateComparisonBinaryOperator(Info, E, OnSuccess, [&]() {
12878     return ExprEvaluatorBaseTy::VisitBinCmp(E);
12879   });
12880 }
12881 
VisitBinaryOperator(const BinaryOperator * E)12882 bool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
12883   // We don't support assignment in C. C++ assignments don't get here because
12884   // assignment is an lvalue in C++.
12885   if (E->isAssignmentOp()) {
12886     Error(E);
12887     if (!Info.noteFailure())
12888       return false;
12889   }
12890 
12891   if (DataRecursiveIntBinOpEvaluator::shouldEnqueue(E))
12892     return DataRecursiveIntBinOpEvaluator(*this, Result).Traverse(E);
12893 
12894   assert((!E->getLHS()->getType()->isIntegralOrEnumerationType() ||
12895           !E->getRHS()->getType()->isIntegralOrEnumerationType()) &&
12896          "DataRecursiveIntBinOpEvaluator should have handled integral types");
12897 
12898   if (E->isComparisonOp()) {
12899     // Evaluate builtin binary comparisons by evaluating them as three-way
12900     // comparisons and then translating the result.
12901     auto OnSuccess = [&](CmpResult CR, const BinaryOperator *E) {
12902       assert((CR != CmpResult::Unequal || E->isEqualityOp()) &&
12903              "should only produce Unequal for equality comparisons");
12904       bool IsEqual   = CR == CmpResult::Equal,
12905            IsLess    = CR == CmpResult::Less,
12906            IsGreater = CR == CmpResult::Greater;
12907       auto Op = E->getOpcode();
12908       switch (Op) {
12909       default:
12910         llvm_unreachable("unsupported binary operator");
12911       case BO_EQ:
12912       case BO_NE:
12913         return Success(IsEqual == (Op == BO_EQ), E);
12914       case BO_LT:
12915         return Success(IsLess, E);
12916       case BO_GT:
12917         return Success(IsGreater, E);
12918       case BO_LE:
12919         return Success(IsEqual || IsLess, E);
12920       case BO_GE:
12921         return Success(IsEqual || IsGreater, E);
12922       }
12923     };
12924     return EvaluateComparisonBinaryOperator(Info, E, OnSuccess, [&]() {
12925       return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
12926     });
12927   }
12928 
12929   QualType LHSTy = E->getLHS()->getType();
12930   QualType RHSTy = E->getRHS()->getType();
12931 
12932   if (LHSTy->isPointerType() && RHSTy->isPointerType() &&
12933       E->getOpcode() == BO_Sub) {
12934     LValue LHSValue, RHSValue;
12935 
12936     bool LHSOK = EvaluatePointer(E->getLHS(), LHSValue, Info);
12937     if (!LHSOK && !Info.noteFailure())
12938       return false;
12939 
12940     if (!EvaluatePointer(E->getRHS(), RHSValue, Info) || !LHSOK)
12941       return false;
12942 
12943     // Reject differing bases from the normal codepath; we special-case
12944     // comparisons to null.
12945     if (!HasSameBase(LHSValue, RHSValue)) {
12946       // Handle &&A - &&B.
12947       if (!LHSValue.Offset.isZero() || !RHSValue.Offset.isZero())
12948         return Error(E);
12949       const Expr *LHSExpr = LHSValue.Base.dyn_cast<const Expr *>();
12950       const Expr *RHSExpr = RHSValue.Base.dyn_cast<const Expr *>();
12951       if (!LHSExpr || !RHSExpr)
12952         return Error(E);
12953       const AddrLabelExpr *LHSAddrExpr = dyn_cast<AddrLabelExpr>(LHSExpr);
12954       const AddrLabelExpr *RHSAddrExpr = dyn_cast<AddrLabelExpr>(RHSExpr);
12955       if (!LHSAddrExpr || !RHSAddrExpr)
12956         return Error(E);
12957       // Make sure both labels come from the same function.
12958       if (LHSAddrExpr->getLabel()->getDeclContext() !=
12959           RHSAddrExpr->getLabel()->getDeclContext())
12960         return Error(E);
12961       return Success(APValue(LHSAddrExpr, RHSAddrExpr), E);
12962     }
12963     const CharUnits &LHSOffset = LHSValue.getLValueOffset();
12964     const CharUnits &RHSOffset = RHSValue.getLValueOffset();
12965 
12966     SubobjectDesignator &LHSDesignator = LHSValue.getLValueDesignator();
12967     SubobjectDesignator &RHSDesignator = RHSValue.getLValueDesignator();
12968 
12969     // C++11 [expr.add]p6:
12970     //   Unless both pointers point to elements of the same array object, or
12971     //   one past the last element of the array object, the behavior is
12972     //   undefined.
12973     if (!LHSDesignator.Invalid && !RHSDesignator.Invalid &&
12974         !AreElementsOfSameArray(getType(LHSValue.Base), LHSDesignator,
12975                                 RHSDesignator))
12976       Info.CCEDiag(E, diag::note_constexpr_pointer_subtraction_not_same_array);
12977 
12978     QualType Type = E->getLHS()->getType();
12979     QualType ElementType = Type->castAs<PointerType>()->getPointeeType();
12980 
12981     CharUnits ElementSize;
12982     if (!HandleSizeof(Info, E->getExprLoc(), ElementType, ElementSize))
12983       return false;
12984 
12985     // As an extension, a type may have zero size (empty struct or union in
12986     // C, array of zero length). Pointer subtraction in such cases has
12987     // undefined behavior, so is not constant.
12988     if (ElementSize.isZero()) {
12989       Info.FFDiag(E, diag::note_constexpr_pointer_subtraction_zero_size)
12990           << ElementType;
12991       return false;
12992     }
12993 
12994     // FIXME: LLVM and GCC both compute LHSOffset - RHSOffset at runtime,
12995     // and produce incorrect results when it overflows. Such behavior
12996     // appears to be non-conforming, but is common, so perhaps we should
12997     // assume the standard intended for such cases to be undefined behavior
12998     // and check for them.
12999 
13000     // Compute (LHSOffset - RHSOffset) / Size carefully, checking for
13001     // overflow in the final conversion to ptrdiff_t.
13002     APSInt LHS(llvm::APInt(65, (int64_t)LHSOffset.getQuantity(), true), false);
13003     APSInt RHS(llvm::APInt(65, (int64_t)RHSOffset.getQuantity(), true), false);
13004     APSInt ElemSize(llvm::APInt(65, (int64_t)ElementSize.getQuantity(), true),
13005                     false);
13006     APSInt TrueResult = (LHS - RHS) / ElemSize;
13007     APSInt Result = TrueResult.trunc(Info.Ctx.getIntWidth(E->getType()));
13008 
13009     if (Result.extend(65) != TrueResult &&
13010         !HandleOverflow(Info, E, TrueResult, E->getType()))
13011       return false;
13012     return Success(Result, E);
13013   }
13014 
13015   return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
13016 }
13017 
13018 /// VisitUnaryExprOrTypeTraitExpr - Evaluate a sizeof, alignof or vec_step with
13019 /// a result as the expression's type.
VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr * E)13020 bool IntExprEvaluator::VisitUnaryExprOrTypeTraitExpr(
13021                                     const UnaryExprOrTypeTraitExpr *E) {
13022   switch(E->getKind()) {
13023   case UETT_PreferredAlignOf:
13024   case UETT_AlignOf: {
13025     if (E->isArgumentType())
13026       return Success(GetAlignOfType(Info, E->getArgumentType(), E->getKind()),
13027                      E);
13028     else
13029       return Success(GetAlignOfExpr(Info, E->getArgumentExpr(), E->getKind()),
13030                      E);
13031   }
13032 
13033   case UETT_VecStep: {
13034     QualType Ty = E->getTypeOfArgument();
13035 
13036     if (Ty->isVectorType()) {
13037       unsigned n = Ty->castAs<VectorType>()->getNumElements();
13038 
13039       // The vec_step built-in functions that take a 3-component
13040       // vector return 4. (OpenCL 1.1 spec 6.11.12)
13041       if (n == 3)
13042         n = 4;
13043 
13044       return Success(n, E);
13045     } else
13046       return Success(1, E);
13047   }
13048 
13049   case UETT_SizeOf: {
13050     QualType SrcTy = E->getTypeOfArgument();
13051     // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
13052     //   the result is the size of the referenced type."
13053     if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
13054       SrcTy = Ref->getPointeeType();
13055 
13056     CharUnits Sizeof;
13057     if (!HandleSizeof(Info, E->getExprLoc(), SrcTy, Sizeof))
13058       return false;
13059     return Success(Sizeof, E);
13060   }
13061   case UETT_OpenMPRequiredSimdAlign:
13062     assert(E->isArgumentType());
13063     return Success(
13064         Info.Ctx.toCharUnitsFromBits(
13065                     Info.Ctx.getOpenMPDefaultSimdAlign(E->getArgumentType()))
13066             .getQuantity(),
13067         E);
13068   }
13069 
13070   llvm_unreachable("unknown expr/type trait");
13071 }
13072 
VisitOffsetOfExpr(const OffsetOfExpr * OOE)13073 bool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *OOE) {
13074   CharUnits Result;
13075   unsigned n = OOE->getNumComponents();
13076   if (n == 0)
13077     return Error(OOE);
13078   QualType CurrentType = OOE->getTypeSourceInfo()->getType();
13079   for (unsigned i = 0; i != n; ++i) {
13080     OffsetOfNode ON = OOE->getComponent(i);
13081     switch (ON.getKind()) {
13082     case OffsetOfNode::Array: {
13083       const Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
13084       APSInt IdxResult;
13085       if (!EvaluateInteger(Idx, IdxResult, Info))
13086         return false;
13087       const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
13088       if (!AT)
13089         return Error(OOE);
13090       CurrentType = AT->getElementType();
13091       CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
13092       Result += IdxResult.getSExtValue() * ElementSize;
13093       break;
13094     }
13095 
13096     case OffsetOfNode::Field: {
13097       FieldDecl *MemberDecl = ON.getField();
13098       const RecordType *RT = CurrentType->getAs<RecordType>();
13099       if (!RT)
13100         return Error(OOE);
13101       RecordDecl *RD = RT->getDecl();
13102       if (RD->isInvalidDecl()) return false;
13103       const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
13104       unsigned i = MemberDecl->getFieldIndex();
13105       assert(i < RL.getFieldCount() && "offsetof field in wrong type");
13106       Result += Info.Ctx.toCharUnitsFromBits(RL.getFieldOffset(i));
13107       CurrentType = MemberDecl->getType().getNonReferenceType();
13108       break;
13109     }
13110 
13111     case OffsetOfNode::Identifier:
13112       llvm_unreachable("dependent __builtin_offsetof");
13113 
13114     case OffsetOfNode::Base: {
13115       CXXBaseSpecifier *BaseSpec = ON.getBase();
13116       if (BaseSpec->isVirtual())
13117         return Error(OOE);
13118 
13119       // Find the layout of the class whose base we are looking into.
13120       const RecordType *RT = CurrentType->getAs<RecordType>();
13121       if (!RT)
13122         return Error(OOE);
13123       RecordDecl *RD = RT->getDecl();
13124       if (RD->isInvalidDecl()) return false;
13125       const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
13126 
13127       // Find the base class itself.
13128       CurrentType = BaseSpec->getType();
13129       const RecordType *BaseRT = CurrentType->getAs<RecordType>();
13130       if (!BaseRT)
13131         return Error(OOE);
13132 
13133       // Add the offset to the base.
13134       Result += RL.getBaseClassOffset(cast<CXXRecordDecl>(BaseRT->getDecl()));
13135       break;
13136     }
13137     }
13138   }
13139   return Success(Result, OOE);
13140 }
13141 
VisitUnaryOperator(const UnaryOperator * E)13142 bool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
13143   switch (E->getOpcode()) {
13144   default:
13145     // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
13146     // See C99 6.6p3.
13147     return Error(E);
13148   case UO_Extension:
13149     // FIXME: Should extension allow i-c-e extension expressions in its scope?
13150     // If so, we could clear the diagnostic ID.
13151     return Visit(E->getSubExpr());
13152   case UO_Plus:
13153     // The result is just the value.
13154     return Visit(E->getSubExpr());
13155   case UO_Minus: {
13156     if (!Visit(E->getSubExpr()))
13157       return false;
13158     if (!Result.isInt()) return Error(E);
13159     const APSInt &Value = Result.getInt();
13160     if (Value.isSigned() && Value.isMinSignedValue() && E->canOverflow() &&
13161         !HandleOverflow(Info, E, -Value.extend(Value.getBitWidth() + 1),
13162                         E->getType()))
13163       return false;
13164     return Success(-Value, E);
13165   }
13166   case UO_Not: {
13167     if (!Visit(E->getSubExpr()))
13168       return false;
13169     if (!Result.isInt()) return Error(E);
13170     return Success(~Result.getInt(), E);
13171   }
13172   case UO_LNot: {
13173     bool bres;
13174     if (!EvaluateAsBooleanCondition(E->getSubExpr(), bres, Info))
13175       return false;
13176     return Success(!bres, E);
13177   }
13178   }
13179 }
13180 
13181 /// HandleCast - This is used to evaluate implicit or explicit casts where the
13182 /// result type is integer.
VisitCastExpr(const CastExpr * E)13183 bool IntExprEvaluator::VisitCastExpr(const CastExpr *E) {
13184   const Expr *SubExpr = E->getSubExpr();
13185   QualType DestType = E->getType();
13186   QualType SrcType = SubExpr->getType();
13187 
13188   switch (E->getCastKind()) {
13189   case CK_BaseToDerived:
13190   case CK_DerivedToBase:
13191   case CK_UncheckedDerivedToBase:
13192   case CK_Dynamic:
13193   case CK_ToUnion:
13194   case CK_ArrayToPointerDecay:
13195   case CK_FunctionToPointerDecay:
13196   case CK_NullToPointer:
13197   case CK_NullToMemberPointer:
13198   case CK_BaseToDerivedMemberPointer:
13199   case CK_DerivedToBaseMemberPointer:
13200   case CK_ReinterpretMemberPointer:
13201   case CK_ConstructorConversion:
13202   case CK_IntegralToPointer:
13203   case CK_ToVoid:
13204   case CK_VectorSplat:
13205   case CK_IntegralToFloating:
13206   case CK_FloatingCast:
13207   case CK_CPointerToObjCPointerCast:
13208   case CK_BlockPointerToObjCPointerCast:
13209   case CK_AnyPointerToBlockPointerCast:
13210   case CK_ObjCObjectLValueCast:
13211   case CK_FloatingRealToComplex:
13212   case CK_FloatingComplexToReal:
13213   case CK_FloatingComplexCast:
13214   case CK_FloatingComplexToIntegralComplex:
13215   case CK_IntegralRealToComplex:
13216   case CK_IntegralComplexCast:
13217   case CK_IntegralComplexToFloatingComplex:
13218   case CK_BuiltinFnToFnPtr:
13219   case CK_ZeroToOCLOpaqueType:
13220   case CK_NonAtomicToAtomic:
13221   case CK_AddressSpaceConversion:
13222   case CK_IntToOCLSampler:
13223   case CK_FloatingToFixedPoint:
13224   case CK_FixedPointToFloating:
13225   case CK_FixedPointCast:
13226   case CK_IntegralToFixedPoint:
13227   case CK_MatrixCast:
13228     llvm_unreachable("invalid cast kind for integral value");
13229 
13230   case CK_BitCast:
13231   case CK_Dependent:
13232   case CK_LValueBitCast:
13233   case CK_ARCProduceObject:
13234   case CK_ARCConsumeObject:
13235   case CK_ARCReclaimReturnedObject:
13236   case CK_ARCExtendBlockObject:
13237   case CK_CopyAndAutoreleaseBlockObject:
13238     return Error(E);
13239 
13240   case CK_UserDefinedConversion:
13241   case CK_LValueToRValue:
13242   case CK_AtomicToNonAtomic:
13243   case CK_NoOp:
13244   case CK_LValueToRValueBitCast:
13245     return ExprEvaluatorBaseTy::VisitCastExpr(E);
13246 
13247   case CK_MemberPointerToBoolean:
13248   case CK_PointerToBoolean:
13249   case CK_IntegralToBoolean:
13250   case CK_FloatingToBoolean:
13251   case CK_BooleanToSignedIntegral:
13252   case CK_FloatingComplexToBoolean:
13253   case CK_IntegralComplexToBoolean: {
13254     bool BoolResult;
13255     if (!EvaluateAsBooleanCondition(SubExpr, BoolResult, Info))
13256       return false;
13257     uint64_t IntResult = BoolResult;
13258     if (BoolResult && E->getCastKind() == CK_BooleanToSignedIntegral)
13259       IntResult = (uint64_t)-1;
13260     return Success(IntResult, E);
13261   }
13262 
13263   case CK_FixedPointToIntegral: {
13264     APFixedPoint Src(Info.Ctx.getFixedPointSemantics(SrcType));
13265     if (!EvaluateFixedPoint(SubExpr, Src, Info))
13266       return false;
13267     bool Overflowed;
13268     llvm::APSInt Result = Src.convertToInt(
13269         Info.Ctx.getIntWidth(DestType),
13270         DestType->isSignedIntegerOrEnumerationType(), &Overflowed);
13271     if (Overflowed && !HandleOverflow(Info, E, Result, DestType))
13272       return false;
13273     return Success(Result, E);
13274   }
13275 
13276   case CK_FixedPointToBoolean: {
13277     // Unsigned padding does not affect this.
13278     APValue Val;
13279     if (!Evaluate(Val, Info, SubExpr))
13280       return false;
13281     return Success(Val.getFixedPoint().getBoolValue(), E);
13282   }
13283 
13284   case CK_IntegralCast: {
13285     if (!Visit(SubExpr))
13286       return false;
13287 
13288     if (!Result.isInt()) {
13289       // Allow casts of address-of-label differences if they are no-ops
13290       // or narrowing.  (The narrowing case isn't actually guaranteed to
13291       // be constant-evaluatable except in some narrow cases which are hard
13292       // to detect here.  We let it through on the assumption the user knows
13293       // what they are doing.)
13294       if (Result.isAddrLabelDiff())
13295         return Info.Ctx.getTypeSize(DestType) <= Info.Ctx.getTypeSize(SrcType);
13296       // Only allow casts of lvalues if they are lossless.
13297       return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
13298     }
13299 
13300     return Success(HandleIntToIntCast(Info, E, DestType, SrcType,
13301                                       Result.getInt()), E);
13302   }
13303 
13304   case CK_PointerToIntegral: {
13305     CCEDiag(E, diag::note_constexpr_invalid_cast) << 2;
13306 
13307     LValue LV;
13308     if (!EvaluatePointer(SubExpr, LV, Info))
13309       return false;
13310 
13311     if (LV.getLValueBase()) {
13312       // Only allow based lvalue casts if they are lossless.
13313       // FIXME: Allow a larger integer size than the pointer size, and allow
13314       // narrowing back down to pointer width in subsequent integral casts.
13315       // FIXME: Check integer type's active bits, not its type size.
13316       if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
13317         return Error(E);
13318 
13319       LV.Designator.setInvalid();
13320       LV.moveInto(Result);
13321       return true;
13322     }
13323 
13324     APSInt AsInt;
13325     APValue V;
13326     LV.moveInto(V);
13327     if (!V.toIntegralConstant(AsInt, SrcType, Info.Ctx))
13328       llvm_unreachable("Can't cast this!");
13329 
13330     return Success(HandleIntToIntCast(Info, E, DestType, SrcType, AsInt), E);
13331   }
13332 
13333   case CK_IntegralComplexToReal: {
13334     ComplexValue C;
13335     if (!EvaluateComplex(SubExpr, C, Info))
13336       return false;
13337     return Success(C.getComplexIntReal(), E);
13338   }
13339 
13340   case CK_FloatingToIntegral: {
13341     APFloat F(0.0);
13342     if (!EvaluateFloat(SubExpr, F, Info))
13343       return false;
13344 
13345     APSInt Value;
13346     if (!HandleFloatToIntCast(Info, E, SrcType, F, DestType, Value))
13347       return false;
13348     return Success(Value, E);
13349   }
13350   }
13351 
13352   llvm_unreachable("unknown cast resulting in integral value");
13353 }
13354 
VisitUnaryReal(const UnaryOperator * E)13355 bool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
13356   if (E->getSubExpr()->getType()->isAnyComplexType()) {
13357     ComplexValue LV;
13358     if (!EvaluateComplex(E->getSubExpr(), LV, Info))
13359       return false;
13360     if (!LV.isComplexInt())
13361       return Error(E);
13362     return Success(LV.getComplexIntReal(), E);
13363   }
13364 
13365   return Visit(E->getSubExpr());
13366 }
13367 
VisitUnaryImag(const UnaryOperator * E)13368 bool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
13369   if (E->getSubExpr()->getType()->isComplexIntegerType()) {
13370     ComplexValue LV;
13371     if (!EvaluateComplex(E->getSubExpr(), LV, Info))
13372       return false;
13373     if (!LV.isComplexInt())
13374       return Error(E);
13375     return Success(LV.getComplexIntImag(), E);
13376   }
13377 
13378   VisitIgnoredValue(E->getSubExpr());
13379   return Success(0, E);
13380 }
13381 
VisitSizeOfPackExpr(const SizeOfPackExpr * E)13382 bool IntExprEvaluator::VisitSizeOfPackExpr(const SizeOfPackExpr *E) {
13383   return Success(E->getPackLength(), E);
13384 }
13385 
VisitCXXNoexceptExpr(const CXXNoexceptExpr * E)13386 bool IntExprEvaluator::VisitCXXNoexceptExpr(const CXXNoexceptExpr *E) {
13387   return Success(E->getValue(), E);
13388 }
13389 
VisitConceptSpecializationExpr(const ConceptSpecializationExpr * E)13390 bool IntExprEvaluator::VisitConceptSpecializationExpr(
13391        const ConceptSpecializationExpr *E) {
13392   return Success(E->isSatisfied(), E);
13393 }
13394 
VisitRequiresExpr(const RequiresExpr * E)13395 bool IntExprEvaluator::VisitRequiresExpr(const RequiresExpr *E) {
13396   return Success(E->isSatisfied(), E);
13397 }
13398 
VisitUnaryOperator(const UnaryOperator * E)13399 bool FixedPointExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
13400   switch (E->getOpcode()) {
13401     default:
13402       // Invalid unary operators
13403       return Error(E);
13404     case UO_Plus:
13405       // The result is just the value.
13406       return Visit(E->getSubExpr());
13407     case UO_Minus: {
13408       if (!Visit(E->getSubExpr())) return false;
13409       if (!Result.isFixedPoint())
13410         return Error(E);
13411       bool Overflowed;
13412       APFixedPoint Negated = Result.getFixedPoint().negate(&Overflowed);
13413       if (Overflowed && !HandleOverflow(Info, E, Negated, E->getType()))
13414         return false;
13415       return Success(Negated, E);
13416     }
13417     case UO_LNot: {
13418       bool bres;
13419       if (!EvaluateAsBooleanCondition(E->getSubExpr(), bres, Info))
13420         return false;
13421       return Success(!bres, E);
13422     }
13423   }
13424 }
13425 
VisitCastExpr(const CastExpr * E)13426 bool FixedPointExprEvaluator::VisitCastExpr(const CastExpr *E) {
13427   const Expr *SubExpr = E->getSubExpr();
13428   QualType DestType = E->getType();
13429   assert(DestType->isFixedPointType() &&
13430          "Expected destination type to be a fixed point type");
13431   auto DestFXSema = Info.Ctx.getFixedPointSemantics(DestType);
13432 
13433   switch (E->getCastKind()) {
13434   case CK_FixedPointCast: {
13435     APFixedPoint Src(Info.Ctx.getFixedPointSemantics(SubExpr->getType()));
13436     if (!EvaluateFixedPoint(SubExpr, Src, Info))
13437       return false;
13438     bool Overflowed;
13439     APFixedPoint Result = Src.convert(DestFXSema, &Overflowed);
13440     if (Overflowed) {
13441       if (Info.checkingForUndefinedBehavior())
13442         Info.Ctx.getDiagnostics().Report(E->getExprLoc(),
13443                                          diag::warn_fixedpoint_constant_overflow)
13444           << Result.toString() << E->getType();
13445       if (!HandleOverflow(Info, E, Result, E->getType()))
13446         return false;
13447     }
13448     return Success(Result, E);
13449   }
13450   case CK_IntegralToFixedPoint: {
13451     APSInt Src;
13452     if (!EvaluateInteger(SubExpr, Src, Info))
13453       return false;
13454 
13455     bool Overflowed;
13456     APFixedPoint IntResult = APFixedPoint::getFromIntValue(
13457         Src, Info.Ctx.getFixedPointSemantics(DestType), &Overflowed);
13458 
13459     if (Overflowed) {
13460       if (Info.checkingForUndefinedBehavior())
13461         Info.Ctx.getDiagnostics().Report(E->getExprLoc(),
13462                                          diag::warn_fixedpoint_constant_overflow)
13463           << IntResult.toString() << E->getType();
13464       if (!HandleOverflow(Info, E, IntResult, E->getType()))
13465         return false;
13466     }
13467 
13468     return Success(IntResult, E);
13469   }
13470   case CK_FloatingToFixedPoint: {
13471     APFloat Src(0.0);
13472     if (!EvaluateFloat(SubExpr, Src, Info))
13473       return false;
13474 
13475     bool Overflowed;
13476     APFixedPoint Result = APFixedPoint::getFromFloatValue(
13477         Src, Info.Ctx.getFixedPointSemantics(DestType), &Overflowed);
13478 
13479     if (Overflowed) {
13480       if (Info.checkingForUndefinedBehavior())
13481         Info.Ctx.getDiagnostics().Report(E->getExprLoc(),
13482                                          diag::warn_fixedpoint_constant_overflow)
13483           << Result.toString() << E->getType();
13484       if (!HandleOverflow(Info, E, Result, E->getType()))
13485         return false;
13486     }
13487 
13488     return Success(Result, E);
13489   }
13490   case CK_NoOp:
13491   case CK_LValueToRValue:
13492     return ExprEvaluatorBaseTy::VisitCastExpr(E);
13493   default:
13494     return Error(E);
13495   }
13496 }
13497 
VisitBinaryOperator(const BinaryOperator * E)13498 bool FixedPointExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
13499   if (E->isPtrMemOp() || E->isAssignmentOp() || E->getOpcode() == BO_Comma)
13500     return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
13501 
13502   const Expr *LHS = E->getLHS();
13503   const Expr *RHS = E->getRHS();
13504   FixedPointSemantics ResultFXSema =
13505       Info.Ctx.getFixedPointSemantics(E->getType());
13506 
13507   APFixedPoint LHSFX(Info.Ctx.getFixedPointSemantics(LHS->getType()));
13508   if (!EvaluateFixedPointOrInteger(LHS, LHSFX, Info))
13509     return false;
13510   APFixedPoint RHSFX(Info.Ctx.getFixedPointSemantics(RHS->getType()));
13511   if (!EvaluateFixedPointOrInteger(RHS, RHSFX, Info))
13512     return false;
13513 
13514   bool OpOverflow = false, ConversionOverflow = false;
13515   APFixedPoint Result(LHSFX.getSemantics());
13516   switch (E->getOpcode()) {
13517   case BO_Add: {
13518     Result = LHSFX.add(RHSFX, &OpOverflow)
13519                   .convert(ResultFXSema, &ConversionOverflow);
13520     break;
13521   }
13522   case BO_Sub: {
13523     Result = LHSFX.sub(RHSFX, &OpOverflow)
13524                   .convert(ResultFXSema, &ConversionOverflow);
13525     break;
13526   }
13527   case BO_Mul: {
13528     Result = LHSFX.mul(RHSFX, &OpOverflow)
13529                   .convert(ResultFXSema, &ConversionOverflow);
13530     break;
13531   }
13532   case BO_Div: {
13533     if (RHSFX.getValue() == 0) {
13534       Info.FFDiag(E, diag::note_expr_divide_by_zero);
13535       return false;
13536     }
13537     Result = LHSFX.div(RHSFX, &OpOverflow)
13538                   .convert(ResultFXSema, &ConversionOverflow);
13539     break;
13540   }
13541   case BO_Shl:
13542   case BO_Shr: {
13543     FixedPointSemantics LHSSema = LHSFX.getSemantics();
13544     llvm::APSInt RHSVal = RHSFX.getValue();
13545 
13546     unsigned ShiftBW =
13547         LHSSema.getWidth() - (unsigned)LHSSema.hasUnsignedPadding();
13548     unsigned Amt = RHSVal.getLimitedValue(ShiftBW - 1);
13549     // Embedded-C 4.1.6.2.2:
13550     //   The right operand must be nonnegative and less than the total number
13551     //   of (nonpadding) bits of the fixed-point operand ...
13552     if (RHSVal.isNegative())
13553       Info.CCEDiag(E, diag::note_constexpr_negative_shift) << RHSVal;
13554     else if (Amt != RHSVal)
13555       Info.CCEDiag(E, diag::note_constexpr_large_shift)
13556           << RHSVal << E->getType() << ShiftBW;
13557 
13558     if (E->getOpcode() == BO_Shl)
13559       Result = LHSFX.shl(Amt, &OpOverflow);
13560     else
13561       Result = LHSFX.shr(Amt, &OpOverflow);
13562     break;
13563   }
13564   default:
13565     return false;
13566   }
13567   if (OpOverflow || ConversionOverflow) {
13568     if (Info.checkingForUndefinedBehavior())
13569       Info.Ctx.getDiagnostics().Report(E->getExprLoc(),
13570                                        diag::warn_fixedpoint_constant_overflow)
13571         << Result.toString() << E->getType();
13572     if (!HandleOverflow(Info, E, Result, E->getType()))
13573       return false;
13574   }
13575   return Success(Result, E);
13576 }
13577 
13578 //===----------------------------------------------------------------------===//
13579 // Float Evaluation
13580 //===----------------------------------------------------------------------===//
13581 
13582 namespace {
13583 class FloatExprEvaluator
13584   : public ExprEvaluatorBase<FloatExprEvaluator> {
13585   APFloat &Result;
13586 public:
FloatExprEvaluator(EvalInfo & info,APFloat & result)13587   FloatExprEvaluator(EvalInfo &info, APFloat &result)
13588     : ExprEvaluatorBaseTy(info), Result(result) {}
13589 
Success(const APValue & V,const Expr * e)13590   bool Success(const APValue &V, const Expr *e) {
13591     Result = V.getFloat();
13592     return true;
13593   }
13594 
ZeroInitialization(const Expr * E)13595   bool ZeroInitialization(const Expr *E) {
13596     Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
13597     return true;
13598   }
13599 
13600   bool VisitCallExpr(const CallExpr *E);
13601 
13602   bool VisitUnaryOperator(const UnaryOperator *E);
13603   bool VisitBinaryOperator(const BinaryOperator *E);
13604   bool VisitFloatingLiteral(const FloatingLiteral *E);
13605   bool VisitCastExpr(const CastExpr *E);
13606 
13607   bool VisitUnaryReal(const UnaryOperator *E);
13608   bool VisitUnaryImag(const UnaryOperator *E);
13609 
13610   // FIXME: Missing: array subscript of vector, member of vector
13611 };
13612 } // end anonymous namespace
13613 
EvaluateFloat(const Expr * E,APFloat & Result,EvalInfo & Info)13614 static bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
13615   assert(!E->isValueDependent());
13616   assert(E->isPRValue() && E->getType()->isRealFloatingType());
13617   return FloatExprEvaluator(Info, Result).Visit(E);
13618 }
13619 
TryEvaluateBuiltinNaN(const ASTContext & Context,QualType ResultTy,const Expr * Arg,bool SNaN,llvm::APFloat & Result)13620 static bool TryEvaluateBuiltinNaN(const ASTContext &Context,
13621                                   QualType ResultTy,
13622                                   const Expr *Arg,
13623                                   bool SNaN,
13624                                   llvm::APFloat &Result) {
13625   const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
13626   if (!S) return false;
13627 
13628   const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
13629 
13630   llvm::APInt fill;
13631 
13632   // Treat empty strings as if they were zero.
13633   if (S->getString().empty())
13634     fill = llvm::APInt(32, 0);
13635   else if (S->getString().getAsInteger(0, fill))
13636     return false;
13637 
13638   if (Context.getTargetInfo().isNan2008()) {
13639     if (SNaN)
13640       Result = llvm::APFloat::getSNaN(Sem, false, &fill);
13641     else
13642       Result = llvm::APFloat::getQNaN(Sem, false, &fill);
13643   } else {
13644     // Prior to IEEE 754-2008, architectures were allowed to choose whether
13645     // the first bit of their significand was set for qNaN or sNaN. MIPS chose
13646     // a different encoding to what became a standard in 2008, and for pre-
13647     // 2008 revisions, MIPS interpreted sNaN-2008 as qNan and qNaN-2008 as
13648     // sNaN. This is now known as "legacy NaN" encoding.
13649     if (SNaN)
13650       Result = llvm::APFloat::getQNaN(Sem, false, &fill);
13651     else
13652       Result = llvm::APFloat::getSNaN(Sem, false, &fill);
13653   }
13654 
13655   return true;
13656 }
13657 
VisitCallExpr(const CallExpr * E)13658 bool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
13659   switch (E->getBuiltinCallee()) {
13660   default:
13661     return ExprEvaluatorBaseTy::VisitCallExpr(E);
13662 
13663   case Builtin::BI__builtin_huge_val:
13664   case Builtin::BI__builtin_huge_valf:
13665   case Builtin::BI__builtin_huge_vall:
13666   case Builtin::BI__builtin_huge_valf128:
13667   case Builtin::BI__builtin_inf:
13668   case Builtin::BI__builtin_inff:
13669   case Builtin::BI__builtin_infl:
13670   case Builtin::BI__builtin_inff128: {
13671     const llvm::fltSemantics &Sem =
13672       Info.Ctx.getFloatTypeSemantics(E->getType());
13673     Result = llvm::APFloat::getInf(Sem);
13674     return true;
13675   }
13676 
13677   case Builtin::BI__builtin_nans:
13678   case Builtin::BI__builtin_nansf:
13679   case Builtin::BI__builtin_nansl:
13680   case Builtin::BI__builtin_nansf128:
13681     if (!TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
13682                                true, Result))
13683       return Error(E);
13684     return true;
13685 
13686   case Builtin::BI__builtin_nan:
13687   case Builtin::BI__builtin_nanf:
13688   case Builtin::BI__builtin_nanl:
13689   case Builtin::BI__builtin_nanf128:
13690     // If this is __builtin_nan() turn this into a nan, otherwise we
13691     // can't constant fold it.
13692     if (!TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
13693                                false, Result))
13694       return Error(E);
13695     return true;
13696 
13697   case Builtin::BI__builtin_fabs:
13698   case Builtin::BI__builtin_fabsf:
13699   case Builtin::BI__builtin_fabsl:
13700   case Builtin::BI__builtin_fabsf128:
13701     // The C standard says "fabs raises no floating-point exceptions,
13702     // even if x is a signaling NaN. The returned value is independent of
13703     // the current rounding direction mode."  Therefore constant folding can
13704     // proceed without regard to the floating point settings.
13705     // Reference, WG14 N2478 F.10.4.3
13706     if (!EvaluateFloat(E->getArg(0), Result, Info))
13707       return false;
13708 
13709     if (Result.isNegative())
13710       Result.changeSign();
13711     return true;
13712 
13713   case Builtin::BI__arithmetic_fence:
13714     return EvaluateFloat(E->getArg(0), Result, Info);
13715 
13716   // FIXME: Builtin::BI__builtin_powi
13717   // FIXME: Builtin::BI__builtin_powif
13718   // FIXME: Builtin::BI__builtin_powil
13719 
13720   case Builtin::BI__builtin_copysign:
13721   case Builtin::BI__builtin_copysignf:
13722   case Builtin::BI__builtin_copysignl:
13723   case Builtin::BI__builtin_copysignf128: {
13724     APFloat RHS(0.);
13725     if (!EvaluateFloat(E->getArg(0), Result, Info) ||
13726         !EvaluateFloat(E->getArg(1), RHS, Info))
13727       return false;
13728     Result.copySign(RHS);
13729     return true;
13730   }
13731   }
13732 }
13733 
VisitUnaryReal(const UnaryOperator * E)13734 bool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
13735   if (E->getSubExpr()->getType()->isAnyComplexType()) {
13736     ComplexValue CV;
13737     if (!EvaluateComplex(E->getSubExpr(), CV, Info))
13738       return false;
13739     Result = CV.FloatReal;
13740     return true;
13741   }
13742 
13743   return Visit(E->getSubExpr());
13744 }
13745 
VisitUnaryImag(const UnaryOperator * E)13746 bool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
13747   if (E->getSubExpr()->getType()->isAnyComplexType()) {
13748     ComplexValue CV;
13749     if (!EvaluateComplex(E->getSubExpr(), CV, Info))
13750       return false;
13751     Result = CV.FloatImag;
13752     return true;
13753   }
13754 
13755   VisitIgnoredValue(E->getSubExpr());
13756   const llvm::fltSemantics &Sem = Info.Ctx.getFloatTypeSemantics(E->getType());
13757   Result = llvm::APFloat::getZero(Sem);
13758   return true;
13759 }
13760 
VisitUnaryOperator(const UnaryOperator * E)13761 bool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
13762   switch (E->getOpcode()) {
13763   default: return Error(E);
13764   case UO_Plus:
13765     return EvaluateFloat(E->getSubExpr(), Result, Info);
13766   case UO_Minus:
13767     // In C standard, WG14 N2478 F.3 p4
13768     // "the unary - raises no floating point exceptions,
13769     // even if the operand is signalling."
13770     if (!EvaluateFloat(E->getSubExpr(), Result, Info))
13771       return false;
13772     Result.changeSign();
13773     return true;
13774   }
13775 }
13776 
VisitBinaryOperator(const BinaryOperator * E)13777 bool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
13778   if (E->isPtrMemOp() || E->isAssignmentOp() || E->getOpcode() == BO_Comma)
13779     return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
13780 
13781   APFloat RHS(0.0);
13782   bool LHSOK = EvaluateFloat(E->getLHS(), Result, Info);
13783   if (!LHSOK && !Info.noteFailure())
13784     return false;
13785   return EvaluateFloat(E->getRHS(), RHS, Info) && LHSOK &&
13786          handleFloatFloatBinOp(Info, E, Result, E->getOpcode(), RHS);
13787 }
13788 
VisitFloatingLiteral(const FloatingLiteral * E)13789 bool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
13790   Result = E->getValue();
13791   return true;
13792 }
13793 
VisitCastExpr(const CastExpr * E)13794 bool FloatExprEvaluator::VisitCastExpr(const CastExpr *E) {
13795   const Expr* SubExpr = E->getSubExpr();
13796 
13797   switch (E->getCastKind()) {
13798   default:
13799     return ExprEvaluatorBaseTy::VisitCastExpr(E);
13800 
13801   case CK_IntegralToFloating: {
13802     APSInt IntResult;
13803     const FPOptions FPO = E->getFPFeaturesInEffect(
13804                                   Info.Ctx.getLangOpts());
13805     return EvaluateInteger(SubExpr, IntResult, Info) &&
13806            HandleIntToFloatCast(Info, E, FPO, SubExpr->getType(),
13807                                 IntResult, E->getType(), Result);
13808   }
13809 
13810   case CK_FixedPointToFloating: {
13811     APFixedPoint FixResult(Info.Ctx.getFixedPointSemantics(SubExpr->getType()));
13812     if (!EvaluateFixedPoint(SubExpr, FixResult, Info))
13813       return false;
13814     Result =
13815         FixResult.convertToFloat(Info.Ctx.getFloatTypeSemantics(E->getType()));
13816     return true;
13817   }
13818 
13819   case CK_FloatingCast: {
13820     if (!Visit(SubExpr))
13821       return false;
13822     return HandleFloatToFloatCast(Info, E, SubExpr->getType(), E->getType(),
13823                                   Result);
13824   }
13825 
13826   case CK_FloatingComplexToReal: {
13827     ComplexValue V;
13828     if (!EvaluateComplex(SubExpr, V, Info))
13829       return false;
13830     Result = V.getComplexFloatReal();
13831     return true;
13832   }
13833   }
13834 }
13835 
13836 //===----------------------------------------------------------------------===//
13837 // Complex Evaluation (for float and integer)
13838 //===----------------------------------------------------------------------===//
13839 
13840 namespace {
13841 class ComplexExprEvaluator
13842   : public ExprEvaluatorBase<ComplexExprEvaluator> {
13843   ComplexValue &Result;
13844 
13845 public:
ComplexExprEvaluator(EvalInfo & info,ComplexValue & Result)13846   ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
13847     : ExprEvaluatorBaseTy(info), Result(Result) {}
13848 
Success(const APValue & V,const Expr * e)13849   bool Success(const APValue &V, const Expr *e) {
13850     Result.setFrom(V);
13851     return true;
13852   }
13853 
13854   bool ZeroInitialization(const Expr *E);
13855 
13856   //===--------------------------------------------------------------------===//
13857   //                            Visitor Methods
13858   //===--------------------------------------------------------------------===//
13859 
13860   bool VisitImaginaryLiteral(const ImaginaryLiteral *E);
13861   bool VisitCastExpr(const CastExpr *E);
13862   bool VisitBinaryOperator(const BinaryOperator *E);
13863   bool VisitUnaryOperator(const UnaryOperator *E);
13864   bool VisitInitListExpr(const InitListExpr *E);
13865   bool VisitCallExpr(const CallExpr *E);
13866 };
13867 } // end anonymous namespace
13868 
EvaluateComplex(const Expr * E,ComplexValue & Result,EvalInfo & Info)13869 static bool EvaluateComplex(const Expr *E, ComplexValue &Result,
13870                             EvalInfo &Info) {
13871   assert(!E->isValueDependent());
13872   assert(E->isPRValue() && E->getType()->isAnyComplexType());
13873   return ComplexExprEvaluator(Info, Result).Visit(E);
13874 }
13875 
ZeroInitialization(const Expr * E)13876 bool ComplexExprEvaluator::ZeroInitialization(const Expr *E) {
13877   QualType ElemTy = E->getType()->castAs<ComplexType>()->getElementType();
13878   if (ElemTy->isRealFloatingType()) {
13879     Result.makeComplexFloat();
13880     APFloat Zero = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(ElemTy));
13881     Result.FloatReal = Zero;
13882     Result.FloatImag = Zero;
13883   } else {
13884     Result.makeComplexInt();
13885     APSInt Zero = Info.Ctx.MakeIntValue(0, ElemTy);
13886     Result.IntReal = Zero;
13887     Result.IntImag = Zero;
13888   }
13889   return true;
13890 }
13891 
VisitImaginaryLiteral(const ImaginaryLiteral * E)13892 bool ComplexExprEvaluator::VisitImaginaryLiteral(const ImaginaryLiteral *E) {
13893   const Expr* SubExpr = E->getSubExpr();
13894 
13895   if (SubExpr->getType()->isRealFloatingType()) {
13896     Result.makeComplexFloat();
13897     APFloat &Imag = Result.FloatImag;
13898     if (!EvaluateFloat(SubExpr, Imag, Info))
13899       return false;
13900 
13901     Result.FloatReal = APFloat(Imag.getSemantics());
13902     return true;
13903   } else {
13904     assert(SubExpr->getType()->isIntegerType() &&
13905            "Unexpected imaginary literal.");
13906 
13907     Result.makeComplexInt();
13908     APSInt &Imag = Result.IntImag;
13909     if (!EvaluateInteger(SubExpr, Imag, Info))
13910       return false;
13911 
13912     Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
13913     return true;
13914   }
13915 }
13916 
VisitCastExpr(const CastExpr * E)13917 bool ComplexExprEvaluator::VisitCastExpr(const CastExpr *E) {
13918 
13919   switch (E->getCastKind()) {
13920   case CK_BitCast:
13921   case CK_BaseToDerived:
13922   case CK_DerivedToBase:
13923   case CK_UncheckedDerivedToBase:
13924   case CK_Dynamic:
13925   case CK_ToUnion:
13926   case CK_ArrayToPointerDecay:
13927   case CK_FunctionToPointerDecay:
13928   case CK_NullToPointer:
13929   case CK_NullToMemberPointer:
13930   case CK_BaseToDerivedMemberPointer:
13931   case CK_DerivedToBaseMemberPointer:
13932   case CK_MemberPointerToBoolean:
13933   case CK_ReinterpretMemberPointer:
13934   case CK_ConstructorConversion:
13935   case CK_IntegralToPointer:
13936   case CK_PointerToIntegral:
13937   case CK_PointerToBoolean:
13938   case CK_ToVoid:
13939   case CK_VectorSplat:
13940   case CK_IntegralCast:
13941   case CK_BooleanToSignedIntegral:
13942   case CK_IntegralToBoolean:
13943   case CK_IntegralToFloating:
13944   case CK_FloatingToIntegral:
13945   case CK_FloatingToBoolean:
13946   case CK_FloatingCast:
13947   case CK_CPointerToObjCPointerCast:
13948   case CK_BlockPointerToObjCPointerCast:
13949   case CK_AnyPointerToBlockPointerCast:
13950   case CK_ObjCObjectLValueCast:
13951   case CK_FloatingComplexToReal:
13952   case CK_FloatingComplexToBoolean:
13953   case CK_IntegralComplexToReal:
13954   case CK_IntegralComplexToBoolean:
13955   case CK_ARCProduceObject:
13956   case CK_ARCConsumeObject:
13957   case CK_ARCReclaimReturnedObject:
13958   case CK_ARCExtendBlockObject:
13959   case CK_CopyAndAutoreleaseBlockObject:
13960   case CK_BuiltinFnToFnPtr:
13961   case CK_ZeroToOCLOpaqueType:
13962   case CK_NonAtomicToAtomic:
13963   case CK_AddressSpaceConversion:
13964   case CK_IntToOCLSampler:
13965   case CK_FloatingToFixedPoint:
13966   case CK_FixedPointToFloating:
13967   case CK_FixedPointCast:
13968   case CK_FixedPointToBoolean:
13969   case CK_FixedPointToIntegral:
13970   case CK_IntegralToFixedPoint:
13971   case CK_MatrixCast:
13972     llvm_unreachable("invalid cast kind for complex value");
13973 
13974   case CK_LValueToRValue:
13975   case CK_AtomicToNonAtomic:
13976   case CK_NoOp:
13977   case CK_LValueToRValueBitCast:
13978     return ExprEvaluatorBaseTy::VisitCastExpr(E);
13979 
13980   case CK_Dependent:
13981   case CK_LValueBitCast:
13982   case CK_UserDefinedConversion:
13983     return Error(E);
13984 
13985   case CK_FloatingRealToComplex: {
13986     APFloat &Real = Result.FloatReal;
13987     if (!EvaluateFloat(E->getSubExpr(), Real, Info))
13988       return false;
13989 
13990     Result.makeComplexFloat();
13991     Result.FloatImag = APFloat(Real.getSemantics());
13992     return true;
13993   }
13994 
13995   case CK_FloatingComplexCast: {
13996     if (!Visit(E->getSubExpr()))
13997       return false;
13998 
13999     QualType To = E->getType()->castAs<ComplexType>()->getElementType();
14000     QualType From
14001       = E->getSubExpr()->getType()->castAs<ComplexType>()->getElementType();
14002 
14003     return HandleFloatToFloatCast(Info, E, From, To, Result.FloatReal) &&
14004            HandleFloatToFloatCast(Info, E, From, To, Result.FloatImag);
14005   }
14006 
14007   case CK_FloatingComplexToIntegralComplex: {
14008     if (!Visit(E->getSubExpr()))
14009       return false;
14010 
14011     QualType To = E->getType()->castAs<ComplexType>()->getElementType();
14012     QualType From
14013       = E->getSubExpr()->getType()->castAs<ComplexType>()->getElementType();
14014     Result.makeComplexInt();
14015     return HandleFloatToIntCast(Info, E, From, Result.FloatReal,
14016                                 To, Result.IntReal) &&
14017            HandleFloatToIntCast(Info, E, From, Result.FloatImag,
14018                                 To, Result.IntImag);
14019   }
14020 
14021   case CK_IntegralRealToComplex: {
14022     APSInt &Real = Result.IntReal;
14023     if (!EvaluateInteger(E->getSubExpr(), Real, Info))
14024       return false;
14025 
14026     Result.makeComplexInt();
14027     Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
14028     return true;
14029   }
14030 
14031   case CK_IntegralComplexCast: {
14032     if (!Visit(E->getSubExpr()))
14033       return false;
14034 
14035     QualType To = E->getType()->castAs<ComplexType>()->getElementType();
14036     QualType From
14037       = E->getSubExpr()->getType()->castAs<ComplexType>()->getElementType();
14038 
14039     Result.IntReal = HandleIntToIntCast(Info, E, To, From, Result.IntReal);
14040     Result.IntImag = HandleIntToIntCast(Info, E, To, From, Result.IntImag);
14041     return true;
14042   }
14043 
14044   case CK_IntegralComplexToFloatingComplex: {
14045     if (!Visit(E->getSubExpr()))
14046       return false;
14047 
14048     const FPOptions FPO = E->getFPFeaturesInEffect(
14049                                   Info.Ctx.getLangOpts());
14050     QualType To = E->getType()->castAs<ComplexType>()->getElementType();
14051     QualType From
14052       = E->getSubExpr()->getType()->castAs<ComplexType>()->getElementType();
14053     Result.makeComplexFloat();
14054     return HandleIntToFloatCast(Info, E, FPO, From, Result.IntReal,
14055                                 To, Result.FloatReal) &&
14056            HandleIntToFloatCast(Info, E, FPO, From, Result.IntImag,
14057                                 To, Result.FloatImag);
14058   }
14059   }
14060 
14061   llvm_unreachable("unknown cast resulting in complex value");
14062 }
14063 
VisitBinaryOperator(const BinaryOperator * E)14064 bool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
14065   if (E->isPtrMemOp() || E->isAssignmentOp() || E->getOpcode() == BO_Comma)
14066     return ExprEvaluatorBaseTy::VisitBinaryOperator(E);
14067 
14068   // Track whether the LHS or RHS is real at the type system level. When this is
14069   // the case we can simplify our evaluation strategy.
14070   bool LHSReal = false, RHSReal = false;
14071 
14072   bool LHSOK;
14073   if (E->getLHS()->getType()->isRealFloatingType()) {
14074     LHSReal = true;
14075     APFloat &Real = Result.FloatReal;
14076     LHSOK = EvaluateFloat(E->getLHS(), Real, Info);
14077     if (LHSOK) {
14078       Result.makeComplexFloat();
14079       Result.FloatImag = APFloat(Real.getSemantics());
14080     }
14081   } else {
14082     LHSOK = Visit(E->getLHS());
14083   }
14084   if (!LHSOK && !Info.noteFailure())
14085     return false;
14086 
14087   ComplexValue RHS;
14088   if (E->getRHS()->getType()->isRealFloatingType()) {
14089     RHSReal = true;
14090     APFloat &Real = RHS.FloatReal;
14091     if (!EvaluateFloat(E->getRHS(), Real, Info) || !LHSOK)
14092       return false;
14093     RHS.makeComplexFloat();
14094     RHS.FloatImag = APFloat(Real.getSemantics());
14095   } else if (!EvaluateComplex(E->getRHS(), RHS, Info) || !LHSOK)
14096     return false;
14097 
14098   assert(!(LHSReal && RHSReal) &&
14099          "Cannot have both operands of a complex operation be real.");
14100   switch (E->getOpcode()) {
14101   default: return Error(E);
14102   case BO_Add:
14103     if (Result.isComplexFloat()) {
14104       Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
14105                                        APFloat::rmNearestTiesToEven);
14106       if (LHSReal)
14107         Result.getComplexFloatImag() = RHS.getComplexFloatImag();
14108       else if (!RHSReal)
14109         Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
14110                                          APFloat::rmNearestTiesToEven);
14111     } else {
14112       Result.getComplexIntReal() += RHS.getComplexIntReal();
14113       Result.getComplexIntImag() += RHS.getComplexIntImag();
14114     }
14115     break;
14116   case BO_Sub:
14117     if (Result.isComplexFloat()) {
14118       Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
14119                                             APFloat::rmNearestTiesToEven);
14120       if (LHSReal) {
14121         Result.getComplexFloatImag() = RHS.getComplexFloatImag();
14122         Result.getComplexFloatImag().changeSign();
14123       } else if (!RHSReal) {
14124         Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
14125                                               APFloat::rmNearestTiesToEven);
14126       }
14127     } else {
14128       Result.getComplexIntReal() -= RHS.getComplexIntReal();
14129       Result.getComplexIntImag() -= RHS.getComplexIntImag();
14130     }
14131     break;
14132   case BO_Mul:
14133     if (Result.isComplexFloat()) {
14134       // This is an implementation of complex multiplication according to the
14135       // constraints laid out in C11 Annex G. The implementation uses the
14136       // following naming scheme:
14137       //   (a + ib) * (c + id)
14138       ComplexValue LHS = Result;
14139       APFloat &A = LHS.getComplexFloatReal();
14140       APFloat &B = LHS.getComplexFloatImag();
14141       APFloat &C = RHS.getComplexFloatReal();
14142       APFloat &D = RHS.getComplexFloatImag();
14143       APFloat &ResR = Result.getComplexFloatReal();
14144       APFloat &ResI = Result.getComplexFloatImag();
14145       if (LHSReal) {
14146         assert(!RHSReal && "Cannot have two real operands for a complex op!");
14147         ResR = A * C;
14148         ResI = A * D;
14149       } else if (RHSReal) {
14150         ResR = C * A;
14151         ResI = C * B;
14152       } else {
14153         // In the fully general case, we need to handle NaNs and infinities
14154         // robustly.
14155         APFloat AC = A * C;
14156         APFloat BD = B * D;
14157         APFloat AD = A * D;
14158         APFloat BC = B * C;
14159         ResR = AC - BD;
14160         ResI = AD + BC;
14161         if (ResR.isNaN() && ResI.isNaN()) {
14162           bool Recalc = false;
14163           if (A.isInfinity() || B.isInfinity()) {
14164             A = APFloat::copySign(
14165                 APFloat(A.getSemantics(), A.isInfinity() ? 1 : 0), A);
14166             B = APFloat::copySign(
14167                 APFloat(B.getSemantics(), B.isInfinity() ? 1 : 0), B);
14168             if (C.isNaN())
14169               C = APFloat::copySign(APFloat(C.getSemantics()), C);
14170             if (D.isNaN())
14171               D = APFloat::copySign(APFloat(D.getSemantics()), D);
14172             Recalc = true;
14173           }
14174           if (C.isInfinity() || D.isInfinity()) {
14175             C = APFloat::copySign(
14176                 APFloat(C.getSemantics(), C.isInfinity() ? 1 : 0), C);
14177             D = APFloat::copySign(
14178                 APFloat(D.getSemantics(), D.isInfinity() ? 1 : 0), D);
14179             if (A.isNaN())
14180               A = APFloat::copySign(APFloat(A.getSemantics()), A);
14181             if (B.isNaN())
14182               B = APFloat::copySign(APFloat(B.getSemantics()), B);
14183             Recalc = true;
14184           }
14185           if (!Recalc && (AC.isInfinity() || BD.isInfinity() ||
14186                           AD.isInfinity() || BC.isInfinity())) {
14187             if (A.isNaN())
14188               A = APFloat::copySign(APFloat(A.getSemantics()), A);
14189             if (B.isNaN())
14190               B = APFloat::copySign(APFloat(B.getSemantics()), B);
14191             if (C.isNaN())
14192               C = APFloat::copySign(APFloat(C.getSemantics()), C);
14193             if (D.isNaN())
14194               D = APFloat::copySign(APFloat(D.getSemantics()), D);
14195             Recalc = true;
14196           }
14197           if (Recalc) {
14198             ResR = APFloat::getInf(A.getSemantics()) * (A * C - B * D);
14199             ResI = APFloat::getInf(A.getSemantics()) * (A * D + B * C);
14200           }
14201         }
14202       }
14203     } else {
14204       ComplexValue LHS = Result;
14205       Result.getComplexIntReal() =
14206         (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
14207          LHS.getComplexIntImag() * RHS.getComplexIntImag());
14208       Result.getComplexIntImag() =
14209         (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
14210          LHS.getComplexIntImag() * RHS.getComplexIntReal());
14211     }
14212     break;
14213   case BO_Div:
14214     if (Result.isComplexFloat()) {
14215       // This is an implementation of complex division according to the
14216       // constraints laid out in C11 Annex G. The implementation uses the
14217       // following naming scheme:
14218       //   (a + ib) / (c + id)
14219       ComplexValue LHS = Result;
14220       APFloat &A = LHS.getComplexFloatReal();
14221       APFloat &B = LHS.getComplexFloatImag();
14222       APFloat &C = RHS.getComplexFloatReal();
14223       APFloat &D = RHS.getComplexFloatImag();
14224       APFloat &ResR = Result.getComplexFloatReal();
14225       APFloat &ResI = Result.getComplexFloatImag();
14226       if (RHSReal) {
14227         ResR = A / C;
14228         ResI = B / C;
14229       } else {
14230         if (LHSReal) {
14231           // No real optimizations we can do here, stub out with zero.
14232           B = APFloat::getZero(A.getSemantics());
14233         }
14234         int DenomLogB = 0;
14235         APFloat MaxCD = maxnum(abs(C), abs(D));
14236         if (MaxCD.isFinite()) {
14237           DenomLogB = ilogb(MaxCD);
14238           C = scalbn(C, -DenomLogB, APFloat::rmNearestTiesToEven);
14239           D = scalbn(D, -DenomLogB, APFloat::rmNearestTiesToEven);
14240         }
14241         APFloat Denom = C * C + D * D;
14242         ResR = scalbn((A * C + B * D) / Denom, -DenomLogB,
14243                       APFloat::rmNearestTiesToEven);
14244         ResI = scalbn((B * C - A * D) / Denom, -DenomLogB,
14245                       APFloat::rmNearestTiesToEven);
14246         if (ResR.isNaN() && ResI.isNaN()) {
14247           if (Denom.isPosZero() && (!A.isNaN() || !B.isNaN())) {
14248             ResR = APFloat::getInf(ResR.getSemantics(), C.isNegative()) * A;
14249             ResI = APFloat::getInf(ResR.getSemantics(), C.isNegative()) * B;
14250           } else if ((A.isInfinity() || B.isInfinity()) && C.isFinite() &&
14251                      D.isFinite()) {
14252             A = APFloat::copySign(
14253                 APFloat(A.getSemantics(), A.isInfinity() ? 1 : 0), A);
14254             B = APFloat::copySign(
14255                 APFloat(B.getSemantics(), B.isInfinity() ? 1 : 0), B);
14256             ResR = APFloat::getInf(ResR.getSemantics()) * (A * C + B * D);
14257             ResI = APFloat::getInf(ResI.getSemantics()) * (B * C - A * D);
14258           } else if (MaxCD.isInfinity() && A.isFinite() && B.isFinite()) {
14259             C = APFloat::copySign(
14260                 APFloat(C.getSemantics(), C.isInfinity() ? 1 : 0), C);
14261             D = APFloat::copySign(
14262                 APFloat(D.getSemantics(), D.isInfinity() ? 1 : 0), D);
14263             ResR = APFloat::getZero(ResR.getSemantics()) * (A * C + B * D);
14264             ResI = APFloat::getZero(ResI.getSemantics()) * (B * C - A * D);
14265           }
14266         }
14267       }
14268     } else {
14269       if (RHS.getComplexIntReal() == 0 && RHS.getComplexIntImag() == 0)
14270         return Error(E, diag::note_expr_divide_by_zero);
14271 
14272       ComplexValue LHS = Result;
14273       APSInt Den = RHS.getComplexIntReal() * RHS.getComplexIntReal() +
14274         RHS.getComplexIntImag() * RHS.getComplexIntImag();
14275       Result.getComplexIntReal() =
14276         (LHS.getComplexIntReal() * RHS.getComplexIntReal() +
14277          LHS.getComplexIntImag() * RHS.getComplexIntImag()) / Den;
14278       Result.getComplexIntImag() =
14279         (LHS.getComplexIntImag() * RHS.getComplexIntReal() -
14280          LHS.getComplexIntReal() * RHS.getComplexIntImag()) / Den;
14281     }
14282     break;
14283   }
14284 
14285   return true;
14286 }
14287 
VisitUnaryOperator(const UnaryOperator * E)14288 bool ComplexExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
14289   // Get the operand value into 'Result'.
14290   if (!Visit(E->getSubExpr()))
14291     return false;
14292 
14293   switch (E->getOpcode()) {
14294   default:
14295     return Error(E);
14296   case UO_Extension:
14297     return true;
14298   case UO_Plus:
14299     // The result is always just the subexpr.
14300     return true;
14301   case UO_Minus:
14302     if (Result.isComplexFloat()) {
14303       Result.getComplexFloatReal().changeSign();
14304       Result.getComplexFloatImag().changeSign();
14305     }
14306     else {
14307       Result.getComplexIntReal() = -Result.getComplexIntReal();
14308       Result.getComplexIntImag() = -Result.getComplexIntImag();
14309     }
14310     return true;
14311   case UO_Not:
14312     if (Result.isComplexFloat())
14313       Result.getComplexFloatImag().changeSign();
14314     else
14315       Result.getComplexIntImag() = -Result.getComplexIntImag();
14316     return true;
14317   }
14318 }
14319 
VisitInitListExpr(const InitListExpr * E)14320 bool ComplexExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
14321   if (E->getNumInits() == 2) {
14322     if (E->getType()->isComplexType()) {
14323       Result.makeComplexFloat();
14324       if (!EvaluateFloat(E->getInit(0), Result.FloatReal, Info))
14325         return false;
14326       if (!EvaluateFloat(E->getInit(1), Result.FloatImag, Info))
14327         return false;
14328     } else {
14329       Result.makeComplexInt();
14330       if (!EvaluateInteger(E->getInit(0), Result.IntReal, Info))
14331         return false;
14332       if (!EvaluateInteger(E->getInit(1), Result.IntImag, Info))
14333         return false;
14334     }
14335     return true;
14336   }
14337   return ExprEvaluatorBaseTy::VisitInitListExpr(E);
14338 }
14339 
VisitCallExpr(const CallExpr * E)14340 bool ComplexExprEvaluator::VisitCallExpr(const CallExpr *E) {
14341   switch (E->getBuiltinCallee()) {
14342   case Builtin::BI__builtin_complex:
14343     Result.makeComplexFloat();
14344     if (!EvaluateFloat(E->getArg(0), Result.FloatReal, Info))
14345       return false;
14346     if (!EvaluateFloat(E->getArg(1), Result.FloatImag, Info))
14347       return false;
14348     return true;
14349 
14350   default:
14351     break;
14352   }
14353 
14354   return ExprEvaluatorBaseTy::VisitCallExpr(E);
14355 }
14356 
14357 //===----------------------------------------------------------------------===//
14358 // Atomic expression evaluation, essentially just handling the NonAtomicToAtomic
14359 // implicit conversion.
14360 //===----------------------------------------------------------------------===//
14361 
14362 namespace {
14363 class AtomicExprEvaluator :
14364     public ExprEvaluatorBase<AtomicExprEvaluator> {
14365   const LValue *This;
14366   APValue &Result;
14367 public:
AtomicExprEvaluator(EvalInfo & Info,const LValue * This,APValue & Result)14368   AtomicExprEvaluator(EvalInfo &Info, const LValue *This, APValue &Result)
14369       : ExprEvaluatorBaseTy(Info), This(This), Result(Result) {}
14370 
Success(const APValue & V,const Expr * E)14371   bool Success(const APValue &V, const Expr *E) {
14372     Result = V;
14373     return true;
14374   }
14375 
ZeroInitialization(const Expr * E)14376   bool ZeroInitialization(const Expr *E) {
14377     ImplicitValueInitExpr VIE(
14378         E->getType()->castAs<AtomicType>()->getValueType());
14379     // For atomic-qualified class (and array) types in C++, initialize the
14380     // _Atomic-wrapped subobject directly, in-place.
14381     return This ? EvaluateInPlace(Result, Info, *This, &VIE)
14382                 : Evaluate(Result, Info, &VIE);
14383   }
14384 
VisitCastExpr(const CastExpr * E)14385   bool VisitCastExpr(const CastExpr *E) {
14386     switch (E->getCastKind()) {
14387     default:
14388       return ExprEvaluatorBaseTy::VisitCastExpr(E);
14389     case CK_NonAtomicToAtomic:
14390       return This ? EvaluateInPlace(Result, Info, *This, E->getSubExpr())
14391                   : Evaluate(Result, Info, E->getSubExpr());
14392     }
14393   }
14394 };
14395 } // end anonymous namespace
14396 
EvaluateAtomic(const Expr * E,const LValue * This,APValue & Result,EvalInfo & Info)14397 static bool EvaluateAtomic(const Expr *E, const LValue *This, APValue &Result,
14398                            EvalInfo &Info) {
14399   assert(!E->isValueDependent());
14400   assert(E->isPRValue() && E->getType()->isAtomicType());
14401   return AtomicExprEvaluator(Info, This, Result).Visit(E);
14402 }
14403 
14404 //===----------------------------------------------------------------------===//
14405 // Void expression evaluation, primarily for a cast to void on the LHS of a
14406 // comma operator
14407 //===----------------------------------------------------------------------===//
14408 
14409 namespace {
14410 class VoidExprEvaluator
14411   : public ExprEvaluatorBase<VoidExprEvaluator> {
14412 public:
VoidExprEvaluator(EvalInfo & Info)14413   VoidExprEvaluator(EvalInfo &Info) : ExprEvaluatorBaseTy(Info) {}
14414 
Success(const APValue & V,const Expr * e)14415   bool Success(const APValue &V, const Expr *e) { return true; }
14416 
ZeroInitialization(const Expr * E)14417   bool ZeroInitialization(const Expr *E) { return true; }
14418 
VisitCastExpr(const CastExpr * E)14419   bool VisitCastExpr(const CastExpr *E) {
14420     switch (E->getCastKind()) {
14421     default:
14422       return ExprEvaluatorBaseTy::VisitCastExpr(E);
14423     case CK_ToVoid:
14424       VisitIgnoredValue(E->getSubExpr());
14425       return true;
14426     }
14427   }
14428 
VisitCallExpr(const CallExpr * E)14429   bool VisitCallExpr(const CallExpr *E) {
14430     switch (E->getBuiltinCallee()) {
14431     case Builtin::BI__assume:
14432     case Builtin::BI__builtin_assume:
14433       // The argument is not evaluated!
14434       return true;
14435 
14436     case Builtin::BI__builtin_operator_delete:
14437       return HandleOperatorDeleteCall(Info, E);
14438 
14439     default:
14440       break;
14441     }
14442 
14443     return ExprEvaluatorBaseTy::VisitCallExpr(E);
14444   }
14445 
14446   bool VisitCXXDeleteExpr(const CXXDeleteExpr *E);
14447 };
14448 } // end anonymous namespace
14449 
VisitCXXDeleteExpr(const CXXDeleteExpr * E)14450 bool VoidExprEvaluator::VisitCXXDeleteExpr(const CXXDeleteExpr *E) {
14451   // We cannot speculatively evaluate a delete expression.
14452   if (Info.SpeculativeEvaluationDepth)
14453     return false;
14454 
14455   FunctionDecl *OperatorDelete = E->getOperatorDelete();
14456   if (!OperatorDelete->isReplaceableGlobalAllocationFunction()) {
14457     Info.FFDiag(E, diag::note_constexpr_new_non_replaceable)
14458         << isa<CXXMethodDecl>(OperatorDelete) << OperatorDelete;
14459     return false;
14460   }
14461 
14462   const Expr *Arg = E->getArgument();
14463 
14464   LValue Pointer;
14465   if (!EvaluatePointer(Arg, Pointer, Info))
14466     return false;
14467   if (Pointer.Designator.Invalid)
14468     return false;
14469 
14470   // Deleting a null pointer has no effect.
14471   if (Pointer.isNullPointer()) {
14472     // This is the only case where we need to produce an extension warning:
14473     // the only other way we can succeed is if we find a dynamic allocation,
14474     // and we will have warned when we allocated it in that case.
14475     if (!Info.getLangOpts().CPlusPlus20)
14476       Info.CCEDiag(E, diag::note_constexpr_new);
14477     return true;
14478   }
14479 
14480   Optional<DynAlloc *> Alloc = CheckDeleteKind(
14481       Info, E, Pointer, E->isArrayForm() ? DynAlloc::ArrayNew : DynAlloc::New);
14482   if (!Alloc)
14483     return false;
14484   QualType AllocType = Pointer.Base.getDynamicAllocType();
14485 
14486   // For the non-array case, the designator must be empty if the static type
14487   // does not have a virtual destructor.
14488   if (!E->isArrayForm() && Pointer.Designator.Entries.size() != 0 &&
14489       !hasVirtualDestructor(Arg->getType()->getPointeeType())) {
14490     Info.FFDiag(E, diag::note_constexpr_delete_base_nonvirt_dtor)
14491         << Arg->getType()->getPointeeType() << AllocType;
14492     return false;
14493   }
14494 
14495   // For a class type with a virtual destructor, the selected operator delete
14496   // is the one looked up when building the destructor.
14497   if (!E->isArrayForm() && !E->isGlobalDelete()) {
14498     const FunctionDecl *VirtualDelete = getVirtualOperatorDelete(AllocType);
14499     if (VirtualDelete &&
14500         !VirtualDelete->isReplaceableGlobalAllocationFunction()) {
14501       Info.FFDiag(E, diag::note_constexpr_new_non_replaceable)
14502           << isa<CXXMethodDecl>(VirtualDelete) << VirtualDelete;
14503       return false;
14504     }
14505   }
14506 
14507   if (!HandleDestruction(Info, E->getExprLoc(), Pointer.getLValueBase(),
14508                          (*Alloc)->Value, AllocType))
14509     return false;
14510 
14511   if (!Info.HeapAllocs.erase(Pointer.Base.dyn_cast<DynamicAllocLValue>())) {
14512     // The element was already erased. This means the destructor call also
14513     // deleted the object.
14514     // FIXME: This probably results in undefined behavior before we get this
14515     // far, and should be diagnosed elsewhere first.
14516     Info.FFDiag(E, diag::note_constexpr_double_delete);
14517     return false;
14518   }
14519 
14520   return true;
14521 }
14522 
EvaluateVoid(const Expr * E,EvalInfo & Info)14523 static bool EvaluateVoid(const Expr *E, EvalInfo &Info) {
14524   assert(!E->isValueDependent());
14525   assert(E->isPRValue() && E->getType()->isVoidType());
14526   return VoidExprEvaluator(Info).Visit(E);
14527 }
14528 
14529 //===----------------------------------------------------------------------===//
14530 // Top level Expr::EvaluateAsRValue method.
14531 //===----------------------------------------------------------------------===//
14532 
Evaluate(APValue & Result,EvalInfo & Info,const Expr * E)14533 static bool Evaluate(APValue &Result, EvalInfo &Info, const Expr *E) {
14534   assert(!E->isValueDependent());
14535   // In C, function designators are not lvalues, but we evaluate them as if they
14536   // are.
14537   QualType T = E->getType();
14538   if (E->isGLValue() || T->isFunctionType()) {
14539     LValue LV;
14540     if (!EvaluateLValue(E, LV, Info))
14541       return false;
14542     LV.moveInto(Result);
14543   } else if (T->isVectorType()) {
14544     if (!EvaluateVector(E, Result, Info))
14545       return false;
14546   } else if (T->isIntegralOrEnumerationType()) {
14547     if (!IntExprEvaluator(Info, Result).Visit(E))
14548       return false;
14549   } else if (T->hasPointerRepresentation()) {
14550     LValue LV;
14551     if (!EvaluatePointer(E, LV, Info))
14552       return false;
14553     LV.moveInto(Result);
14554   } else if (T->isRealFloatingType()) {
14555     llvm::APFloat F(0.0);
14556     if (!EvaluateFloat(E, F, Info))
14557       return false;
14558     Result = APValue(F);
14559   } else if (T->isAnyComplexType()) {
14560     ComplexValue C;
14561     if (!EvaluateComplex(E, C, Info))
14562       return false;
14563     C.moveInto(Result);
14564   } else if (T->isFixedPointType()) {
14565     if (!FixedPointExprEvaluator(Info, Result).Visit(E)) return false;
14566   } else if (T->isMemberPointerType()) {
14567     MemberPtr P;
14568     if (!EvaluateMemberPointer(E, P, Info))
14569       return false;
14570     P.moveInto(Result);
14571     return true;
14572   } else if (T->isArrayType()) {
14573     LValue LV;
14574     APValue &Value =
14575         Info.CurrentCall->createTemporary(E, T, ScopeKind::FullExpression, LV);
14576     if (!EvaluateArray(E, LV, Value, Info))
14577       return false;
14578     Result = Value;
14579   } else if (T->isRecordType()) {
14580     LValue LV;
14581     APValue &Value =
14582         Info.CurrentCall->createTemporary(E, T, ScopeKind::FullExpression, LV);
14583     if (!EvaluateRecord(E, LV, Value, Info))
14584       return false;
14585     Result = Value;
14586   } else if (T->isVoidType()) {
14587     if (!Info.getLangOpts().CPlusPlus11)
14588       Info.CCEDiag(E, diag::note_constexpr_nonliteral)
14589         << E->getType();
14590     if (!EvaluateVoid(E, Info))
14591       return false;
14592   } else if (T->isAtomicType()) {
14593     QualType Unqual = T.getAtomicUnqualifiedType();
14594     if (Unqual->isArrayType() || Unqual->isRecordType()) {
14595       LValue LV;
14596       APValue &Value = Info.CurrentCall->createTemporary(
14597           E, Unqual, ScopeKind::FullExpression, LV);
14598       if (!EvaluateAtomic(E, &LV, Value, Info))
14599         return false;
14600     } else {
14601       if (!EvaluateAtomic(E, nullptr, Result, Info))
14602         return false;
14603     }
14604   } else if (Info.getLangOpts().CPlusPlus11) {
14605     Info.FFDiag(E, diag::note_constexpr_nonliteral) << E->getType();
14606     return false;
14607   } else {
14608     Info.FFDiag(E, diag::note_invalid_subexpr_in_const_expr);
14609     return false;
14610   }
14611 
14612   return true;
14613 }
14614 
14615 /// EvaluateInPlace - Evaluate an expression in-place in an APValue. In some
14616 /// cases, the in-place evaluation is essential, since later initializers for
14617 /// an object can indirectly refer to subobjects which were initialized earlier.
EvaluateInPlace(APValue & Result,EvalInfo & Info,const LValue & This,const Expr * E,bool AllowNonLiteralTypes)14618 static bool EvaluateInPlace(APValue &Result, EvalInfo &Info, const LValue &This,
14619                             const Expr *E, bool AllowNonLiteralTypes) {
14620   assert(!E->isValueDependent());
14621 
14622   if (!AllowNonLiteralTypes && !CheckLiteralType(Info, E, &This))
14623     return false;
14624 
14625   if (E->isPRValue()) {
14626     // Evaluate arrays and record types in-place, so that later initializers can
14627     // refer to earlier-initialized members of the object.
14628     QualType T = E->getType();
14629     if (T->isArrayType())
14630       return EvaluateArray(E, This, Result, Info);
14631     else if (T->isRecordType())
14632       return EvaluateRecord(E, This, Result, Info);
14633     else if (T->isAtomicType()) {
14634       QualType Unqual = T.getAtomicUnqualifiedType();
14635       if (Unqual->isArrayType() || Unqual->isRecordType())
14636         return EvaluateAtomic(E, &This, Result, Info);
14637     }
14638   }
14639 
14640   // For any other type, in-place evaluation is unimportant.
14641   return Evaluate(Result, Info, E);
14642 }
14643 
14644 /// EvaluateAsRValue - Try to evaluate this expression, performing an implicit
14645 /// lvalue-to-rvalue cast if it is an lvalue.
EvaluateAsRValue(EvalInfo & Info,const Expr * E,APValue & Result)14646 static bool EvaluateAsRValue(EvalInfo &Info, const Expr *E, APValue &Result) {
14647   assert(!E->isValueDependent());
14648   if (Info.EnableNewConstInterp) {
14649     if (!Info.Ctx.getInterpContext().evaluateAsRValue(Info, E, Result))
14650       return false;
14651   } else {
14652     if (E->getType().isNull())
14653       return false;
14654 
14655     if (!CheckLiteralType(Info, E))
14656       return false;
14657 
14658     if (!::Evaluate(Result, Info, E))
14659       return false;
14660 
14661     if (E->isGLValue()) {
14662       LValue LV;
14663       LV.setFrom(Info.Ctx, Result);
14664       if (!handleLValueToRValueConversion(Info, E, E->getType(), LV, Result))
14665         return false;
14666     }
14667   }
14668 
14669   // Check this core constant expression is a constant expression.
14670   return CheckConstantExpression(Info, E->getExprLoc(), E->getType(), Result,
14671                                  ConstantExprKind::Normal) &&
14672          CheckMemoryLeaks(Info);
14673 }
14674 
FastEvaluateAsRValue(const Expr * Exp,Expr::EvalResult & Result,const ASTContext & Ctx,bool & IsConst)14675 static bool FastEvaluateAsRValue(const Expr *Exp, Expr::EvalResult &Result,
14676                                  const ASTContext &Ctx, bool &IsConst) {
14677   // Fast-path evaluations of integer literals, since we sometimes see files
14678   // containing vast quantities of these.
14679   if (const IntegerLiteral *L = dyn_cast<IntegerLiteral>(Exp)) {
14680     Result.Val = APValue(APSInt(L->getValue(),
14681                                 L->getType()->isUnsignedIntegerType()));
14682     IsConst = true;
14683     return true;
14684   }
14685 
14686   // This case should be rare, but we need to check it before we check on
14687   // the type below.
14688   if (Exp->getType().isNull()) {
14689     IsConst = false;
14690     return true;
14691   }
14692 
14693   // FIXME: Evaluating values of large array and record types can cause
14694   // performance problems. Only do so in C++11 for now.
14695   if (Exp->isPRValue() &&
14696       (Exp->getType()->isArrayType() || Exp->getType()->isRecordType()) &&
14697       !Ctx.getLangOpts().CPlusPlus11) {
14698     IsConst = false;
14699     return true;
14700   }
14701   return false;
14702 }
14703 
hasUnacceptableSideEffect(Expr::EvalStatus & Result,Expr::SideEffectsKind SEK)14704 static bool hasUnacceptableSideEffect(Expr::EvalStatus &Result,
14705                                       Expr::SideEffectsKind SEK) {
14706   return (SEK < Expr::SE_AllowSideEffects && Result.HasSideEffects) ||
14707          (SEK < Expr::SE_AllowUndefinedBehavior && Result.HasUndefinedBehavior);
14708 }
14709 
EvaluateAsRValue(const Expr * E,Expr::EvalResult & Result,const ASTContext & Ctx,EvalInfo & Info)14710 static bool EvaluateAsRValue(const Expr *E, Expr::EvalResult &Result,
14711                              const ASTContext &Ctx, EvalInfo &Info) {
14712   assert(!E->isValueDependent());
14713   bool IsConst;
14714   if (FastEvaluateAsRValue(E, Result, Ctx, IsConst))
14715     return IsConst;
14716 
14717   return EvaluateAsRValue(Info, E, Result.Val);
14718 }
14719 
EvaluateAsInt(const Expr * E,Expr::EvalResult & ExprResult,const ASTContext & Ctx,Expr::SideEffectsKind AllowSideEffects,EvalInfo & Info)14720 static bool EvaluateAsInt(const Expr *E, Expr::EvalResult &ExprResult,
14721                           const ASTContext &Ctx,
14722                           Expr::SideEffectsKind AllowSideEffects,
14723                           EvalInfo &Info) {
14724   assert(!E->isValueDependent());
14725   if (!E->getType()->isIntegralOrEnumerationType())
14726     return false;
14727 
14728   if (!::EvaluateAsRValue(E, ExprResult, Ctx, Info) ||
14729       !ExprResult.Val.isInt() ||
14730       hasUnacceptableSideEffect(ExprResult, AllowSideEffects))
14731     return false;
14732 
14733   return true;
14734 }
14735 
EvaluateAsFixedPoint(const Expr * E,Expr::EvalResult & ExprResult,const ASTContext & Ctx,Expr::SideEffectsKind AllowSideEffects,EvalInfo & Info)14736 static bool EvaluateAsFixedPoint(const Expr *E, Expr::EvalResult &ExprResult,
14737                                  const ASTContext &Ctx,
14738                                  Expr::SideEffectsKind AllowSideEffects,
14739                                  EvalInfo &Info) {
14740   assert(!E->isValueDependent());
14741   if (!E->getType()->isFixedPointType())
14742     return false;
14743 
14744   if (!::EvaluateAsRValue(E, ExprResult, Ctx, Info))
14745     return false;
14746 
14747   if (!ExprResult.Val.isFixedPoint() ||
14748       hasUnacceptableSideEffect(ExprResult, AllowSideEffects))
14749     return false;
14750 
14751   return true;
14752 }
14753 
14754 /// EvaluateAsRValue - Return true if this is a constant which we can fold using
14755 /// any crazy technique (that has nothing to do with language standards) that
14756 /// we want to.  If this function returns true, it returns the folded constant
14757 /// in Result. If this expression is a glvalue, an lvalue-to-rvalue conversion
14758 /// will be applied to the result.
EvaluateAsRValue(EvalResult & Result,const ASTContext & Ctx,bool InConstantContext) const14759 bool Expr::EvaluateAsRValue(EvalResult &Result, const ASTContext &Ctx,
14760                             bool InConstantContext) const {
14761   assert(!isValueDependent() &&
14762          "Expression evaluator can't be called on a dependent expression.");
14763   EvalInfo Info(Ctx, Result, EvalInfo::EM_IgnoreSideEffects);
14764   Info.InConstantContext = InConstantContext;
14765   return ::EvaluateAsRValue(this, Result, Ctx, Info);
14766 }
14767 
EvaluateAsBooleanCondition(bool & Result,const ASTContext & Ctx,bool InConstantContext) const14768 bool Expr::EvaluateAsBooleanCondition(bool &Result, const ASTContext &Ctx,
14769                                       bool InConstantContext) const {
14770   assert(!isValueDependent() &&
14771          "Expression evaluator can't be called on a dependent expression.");
14772   EvalResult Scratch;
14773   return EvaluateAsRValue(Scratch, Ctx, InConstantContext) &&
14774          HandleConversionToBool(Scratch.Val, Result);
14775 }
14776 
EvaluateAsInt(EvalResult & Result,const ASTContext & Ctx,SideEffectsKind AllowSideEffects,bool InConstantContext) const14777 bool Expr::EvaluateAsInt(EvalResult &Result, const ASTContext &Ctx,
14778                          SideEffectsKind AllowSideEffects,
14779                          bool InConstantContext) const {
14780   assert(!isValueDependent() &&
14781          "Expression evaluator can't be called on a dependent expression.");
14782   EvalInfo Info(Ctx, Result, EvalInfo::EM_IgnoreSideEffects);
14783   Info.InConstantContext = InConstantContext;
14784   return ::EvaluateAsInt(this, Result, Ctx, AllowSideEffects, Info);
14785 }
14786 
EvaluateAsFixedPoint(EvalResult & Result,const ASTContext & Ctx,SideEffectsKind AllowSideEffects,bool InConstantContext) const14787 bool Expr::EvaluateAsFixedPoint(EvalResult &Result, const ASTContext &Ctx,
14788                                 SideEffectsKind AllowSideEffects,
14789                                 bool InConstantContext) const {
14790   assert(!isValueDependent() &&
14791          "Expression evaluator can't be called on a dependent expression.");
14792   EvalInfo Info(Ctx, Result, EvalInfo::EM_IgnoreSideEffects);
14793   Info.InConstantContext = InConstantContext;
14794   return ::EvaluateAsFixedPoint(this, Result, Ctx, AllowSideEffects, Info);
14795 }
14796 
EvaluateAsFloat(APFloat & Result,const ASTContext & Ctx,SideEffectsKind AllowSideEffects,bool InConstantContext) const14797 bool Expr::EvaluateAsFloat(APFloat &Result, const ASTContext &Ctx,
14798                            SideEffectsKind AllowSideEffects,
14799                            bool InConstantContext) const {
14800   assert(!isValueDependent() &&
14801          "Expression evaluator can't be called on a dependent expression.");
14802 
14803   if (!getType()->isRealFloatingType())
14804     return false;
14805 
14806   EvalResult ExprResult;
14807   if (!EvaluateAsRValue(ExprResult, Ctx, InConstantContext) ||
14808       !ExprResult.Val.isFloat() ||
14809       hasUnacceptableSideEffect(ExprResult, AllowSideEffects))
14810     return false;
14811 
14812   Result = ExprResult.Val.getFloat();
14813   return true;
14814 }
14815 
EvaluateAsLValue(EvalResult & Result,const ASTContext & Ctx,bool InConstantContext) const14816 bool Expr::EvaluateAsLValue(EvalResult &Result, const ASTContext &Ctx,
14817                             bool InConstantContext) const {
14818   assert(!isValueDependent() &&
14819          "Expression evaluator can't be called on a dependent expression.");
14820 
14821   EvalInfo Info(Ctx, Result, EvalInfo::EM_ConstantFold);
14822   Info.InConstantContext = InConstantContext;
14823   LValue LV;
14824   CheckedTemporaries CheckedTemps;
14825   if (!EvaluateLValue(this, LV, Info) || !Info.discardCleanups() ||
14826       Result.HasSideEffects ||
14827       !CheckLValueConstantExpression(Info, getExprLoc(),
14828                                      Ctx.getLValueReferenceType(getType()), LV,
14829                                      ConstantExprKind::Normal, CheckedTemps))
14830     return false;
14831 
14832   LV.moveInto(Result.Val);
14833   return true;
14834 }
14835 
EvaluateDestruction(const ASTContext & Ctx,APValue::LValueBase Base,APValue DestroyedValue,QualType Type,SourceLocation Loc,Expr::EvalStatus & EStatus,bool IsConstantDestruction)14836 static bool EvaluateDestruction(const ASTContext &Ctx, APValue::LValueBase Base,
14837                                 APValue DestroyedValue, QualType Type,
14838                                 SourceLocation Loc, Expr::EvalStatus &EStatus,
14839                                 bool IsConstantDestruction) {
14840   EvalInfo Info(Ctx, EStatus,
14841                 IsConstantDestruction ? EvalInfo::EM_ConstantExpression
14842                                       : EvalInfo::EM_ConstantFold);
14843   Info.setEvaluatingDecl(Base, DestroyedValue,
14844                          EvalInfo::EvaluatingDeclKind::Dtor);
14845   Info.InConstantContext = IsConstantDestruction;
14846 
14847   LValue LVal;
14848   LVal.set(Base);
14849 
14850   if (!HandleDestruction(Info, Loc, Base, DestroyedValue, Type) ||
14851       EStatus.HasSideEffects)
14852     return false;
14853 
14854   if (!Info.discardCleanups())
14855     llvm_unreachable("Unhandled cleanup; missing full expression marker?");
14856 
14857   return true;
14858 }
14859 
EvaluateAsConstantExpr(EvalResult & Result,const ASTContext & Ctx,ConstantExprKind Kind) const14860 bool Expr::EvaluateAsConstantExpr(EvalResult &Result, const ASTContext &Ctx,
14861                                   ConstantExprKind Kind) const {
14862   assert(!isValueDependent() &&
14863          "Expression evaluator can't be called on a dependent expression.");
14864 
14865   EvalInfo::EvaluationMode EM = EvalInfo::EM_ConstantExpression;
14866   EvalInfo Info(Ctx, Result, EM);
14867   Info.InConstantContext = true;
14868 
14869   // The type of the object we're initializing is 'const T' for a class NTTP.
14870   QualType T = getType();
14871   if (Kind == ConstantExprKind::ClassTemplateArgument)
14872     T.addConst();
14873 
14874   // If we're evaluating a prvalue, fake up a MaterializeTemporaryExpr to
14875   // represent the result of the evaluation. CheckConstantExpression ensures
14876   // this doesn't escape.
14877   MaterializeTemporaryExpr BaseMTE(T, const_cast<Expr*>(this), true);
14878   APValue::LValueBase Base(&BaseMTE);
14879 
14880   Info.setEvaluatingDecl(Base, Result.Val);
14881   LValue LVal;
14882   LVal.set(Base);
14883 
14884   if (!::EvaluateInPlace(Result.Val, Info, LVal, this) || Result.HasSideEffects)
14885     return false;
14886 
14887   if (!Info.discardCleanups())
14888     llvm_unreachable("Unhandled cleanup; missing full expression marker?");
14889 
14890   if (!CheckConstantExpression(Info, getExprLoc(), getStorageType(Ctx, this),
14891                                Result.Val, Kind))
14892     return false;
14893   if (!CheckMemoryLeaks(Info))
14894     return false;
14895 
14896   // If this is a class template argument, it's required to have constant
14897   // destruction too.
14898   if (Kind == ConstantExprKind::ClassTemplateArgument &&
14899       (!EvaluateDestruction(Ctx, Base, Result.Val, T, getBeginLoc(), Result,
14900                             true) ||
14901        Result.HasSideEffects)) {
14902     // FIXME: Prefix a note to indicate that the problem is lack of constant
14903     // destruction.
14904     return false;
14905   }
14906 
14907   return true;
14908 }
14909 
EvaluateAsInitializer(APValue & Value,const ASTContext & Ctx,const VarDecl * VD,SmallVectorImpl<PartialDiagnosticAt> & Notes) const14910 bool Expr::EvaluateAsInitializer(APValue &Value, const ASTContext &Ctx,
14911                                  const VarDecl *VD,
14912                             SmallVectorImpl<PartialDiagnosticAt> &Notes) const {
14913   assert(!isValueDependent() &&
14914          "Expression evaluator can't be called on a dependent expression.");
14915 
14916   // FIXME: Evaluating initializers for large array and record types can cause
14917   // performance problems. Only do so in C++11 for now.
14918   if (isPRValue() && (getType()->isArrayType() || getType()->isRecordType()) &&
14919       !Ctx.getLangOpts().CPlusPlus11)
14920     return false;
14921 
14922   Expr::EvalStatus EStatus;
14923   EStatus.Diag = &Notes;
14924 
14925   EvalInfo Info(Ctx, EStatus, VD->isConstexpr()
14926                                       ? EvalInfo::EM_ConstantExpression
14927                                       : EvalInfo::EM_ConstantFold);
14928   Info.setEvaluatingDecl(VD, Value);
14929   Info.InConstantContext = true;
14930 
14931   SourceLocation DeclLoc = VD->getLocation();
14932   QualType DeclTy = VD->getType();
14933 
14934   if (Info.EnableNewConstInterp) {
14935     auto &InterpCtx = const_cast<ASTContext &>(Ctx).getInterpContext();
14936     if (!InterpCtx.evaluateAsInitializer(Info, VD, Value))
14937       return false;
14938   } else {
14939     LValue LVal;
14940     LVal.set(VD);
14941 
14942     if (!EvaluateInPlace(Value, Info, LVal, this,
14943                          /*AllowNonLiteralTypes=*/true) ||
14944         EStatus.HasSideEffects)
14945       return false;
14946 
14947     // At this point, any lifetime-extended temporaries are completely
14948     // initialized.
14949     Info.performLifetimeExtension();
14950 
14951     if (!Info.discardCleanups())
14952       llvm_unreachable("Unhandled cleanup; missing full expression marker?");
14953   }
14954   return CheckConstantExpression(Info, DeclLoc, DeclTy, Value,
14955                                  ConstantExprKind::Normal) &&
14956          CheckMemoryLeaks(Info);
14957 }
14958 
evaluateDestruction(SmallVectorImpl<PartialDiagnosticAt> & Notes) const14959 bool VarDecl::evaluateDestruction(
14960     SmallVectorImpl<PartialDiagnosticAt> &Notes) const {
14961   Expr::EvalStatus EStatus;
14962   EStatus.Diag = &Notes;
14963 
14964   // Only treat the destruction as constant destruction if we formally have
14965   // constant initialization (or are usable in a constant expression).
14966   bool IsConstantDestruction = hasConstantInitialization();
14967 
14968   // Make a copy of the value for the destructor to mutate, if we know it.
14969   // Otherwise, treat the value as default-initialized; if the destructor works
14970   // anyway, then the destruction is constant (and must be essentially empty).
14971   APValue DestroyedValue;
14972   if (getEvaluatedValue() && !getEvaluatedValue()->isAbsent())
14973     DestroyedValue = *getEvaluatedValue();
14974   else if (!getDefaultInitValue(getType(), DestroyedValue))
14975     return false;
14976 
14977   if (!EvaluateDestruction(getASTContext(), this, std::move(DestroyedValue),
14978                            getType(), getLocation(), EStatus,
14979                            IsConstantDestruction) ||
14980       EStatus.HasSideEffects)
14981     return false;
14982 
14983   ensureEvaluatedStmt()->HasConstantDestruction = true;
14984   return true;
14985 }
14986 
14987 /// isEvaluatable - Call EvaluateAsRValue to see if this expression can be
14988 /// constant folded, but discard the result.
isEvaluatable(const ASTContext & Ctx,SideEffectsKind SEK) const14989 bool Expr::isEvaluatable(const ASTContext &Ctx, SideEffectsKind SEK) const {
14990   assert(!isValueDependent() &&
14991          "Expression evaluator can't be called on a dependent expression.");
14992 
14993   EvalResult Result;
14994   return EvaluateAsRValue(Result, Ctx, /* in constant context */ true) &&
14995          !hasUnacceptableSideEffect(Result, SEK);
14996 }
14997 
EvaluateKnownConstInt(const ASTContext & Ctx,SmallVectorImpl<PartialDiagnosticAt> * Diag) const14998 APSInt Expr::EvaluateKnownConstInt(const ASTContext &Ctx,
14999                     SmallVectorImpl<PartialDiagnosticAt> *Diag) const {
15000   assert(!isValueDependent() &&
15001          "Expression evaluator can't be called on a dependent expression.");
15002 
15003   EvalResult EVResult;
15004   EVResult.Diag = Diag;
15005   EvalInfo Info(Ctx, EVResult, EvalInfo::EM_IgnoreSideEffects);
15006   Info.InConstantContext = true;
15007 
15008   bool Result = ::EvaluateAsRValue(this, EVResult, Ctx, Info);
15009   (void)Result;
15010   assert(Result && "Could not evaluate expression");
15011   assert(EVResult.Val.isInt() && "Expression did not evaluate to integer");
15012 
15013   return EVResult.Val.getInt();
15014 }
15015 
EvaluateKnownConstIntCheckOverflow(const ASTContext & Ctx,SmallVectorImpl<PartialDiagnosticAt> * Diag) const15016 APSInt Expr::EvaluateKnownConstIntCheckOverflow(
15017     const ASTContext &Ctx, SmallVectorImpl<PartialDiagnosticAt> *Diag) const {
15018   assert(!isValueDependent() &&
15019          "Expression evaluator can't be called on a dependent expression.");
15020 
15021   EvalResult EVResult;
15022   EVResult.Diag = Diag;
15023   EvalInfo Info(Ctx, EVResult, EvalInfo::EM_IgnoreSideEffects);
15024   Info.InConstantContext = true;
15025   Info.CheckingForUndefinedBehavior = true;
15026 
15027   bool Result = ::EvaluateAsRValue(Info, this, EVResult.Val);
15028   (void)Result;
15029   assert(Result && "Could not evaluate expression");
15030   assert(EVResult.Val.isInt() && "Expression did not evaluate to integer");
15031 
15032   return EVResult.Val.getInt();
15033 }
15034 
EvaluateForOverflow(const ASTContext & Ctx) const15035 void Expr::EvaluateForOverflow(const ASTContext &Ctx) const {
15036   assert(!isValueDependent() &&
15037          "Expression evaluator can't be called on a dependent expression.");
15038 
15039   bool IsConst;
15040   EvalResult EVResult;
15041   if (!FastEvaluateAsRValue(this, EVResult, Ctx, IsConst)) {
15042     EvalInfo Info(Ctx, EVResult, EvalInfo::EM_IgnoreSideEffects);
15043     Info.CheckingForUndefinedBehavior = true;
15044     (void)::EvaluateAsRValue(Info, this, EVResult.Val);
15045   }
15046 }
15047 
isGlobalLValue() const15048 bool Expr::EvalResult::isGlobalLValue() const {
15049   assert(Val.isLValue());
15050   return IsGlobalLValue(Val.getLValueBase());
15051 }
15052 
15053 /// isIntegerConstantExpr - this recursive routine will test if an expression is
15054 /// an integer constant expression.
15055 
15056 /// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
15057 /// comma, etc
15058 
15059 // CheckICE - This function does the fundamental ICE checking: the returned
15060 // ICEDiag contains an ICEKind indicating whether the expression is an ICE,
15061 // and a (possibly null) SourceLocation indicating the location of the problem.
15062 //
15063 // Note that to reduce code duplication, this helper does no evaluation
15064 // itself; the caller checks whether the expression is evaluatable, and
15065 // in the rare cases where CheckICE actually cares about the evaluated
15066 // value, it calls into Evaluate.
15067 
15068 namespace {
15069 
15070 enum ICEKind {
15071   /// This expression is an ICE.
15072   IK_ICE,
15073   /// This expression is not an ICE, but if it isn't evaluated, it's
15074   /// a legal subexpression for an ICE. This return value is used to handle
15075   /// the comma operator in C99 mode, and non-constant subexpressions.
15076   IK_ICEIfUnevaluated,
15077   /// This expression is not an ICE, and is not a legal subexpression for one.
15078   IK_NotICE
15079 };
15080 
15081 struct ICEDiag {
15082   ICEKind Kind;
15083   SourceLocation Loc;
15084 
ICEDiag__anonb66d72d23511::ICEDiag15085   ICEDiag(ICEKind IK, SourceLocation l) : Kind(IK), Loc(l) {}
15086 };
15087 
15088 }
15089 
NoDiag()15090 static ICEDiag NoDiag() { return ICEDiag(IK_ICE, SourceLocation()); }
15091 
Worst(ICEDiag A,ICEDiag B)15092 static ICEDiag Worst(ICEDiag A, ICEDiag B) { return A.Kind >= B.Kind ? A : B; }
15093 
CheckEvalInICE(const Expr * E,const ASTContext & Ctx)15094 static ICEDiag CheckEvalInICE(const Expr* E, const ASTContext &Ctx) {
15095   Expr::EvalResult EVResult;
15096   Expr::EvalStatus Status;
15097   EvalInfo Info(Ctx, Status, EvalInfo::EM_ConstantExpression);
15098 
15099   Info.InConstantContext = true;
15100   if (!::EvaluateAsRValue(E, EVResult, Ctx, Info) || EVResult.HasSideEffects ||
15101       !EVResult.Val.isInt())
15102     return ICEDiag(IK_NotICE, E->getBeginLoc());
15103 
15104   return NoDiag();
15105 }
15106 
CheckICE(const Expr * E,const ASTContext & Ctx)15107 static ICEDiag CheckICE(const Expr* E, const ASTContext &Ctx) {
15108   assert(!E->isValueDependent() && "Should not see value dependent exprs!");
15109   if (!E->getType()->isIntegralOrEnumerationType())
15110     return ICEDiag(IK_NotICE, E->getBeginLoc());
15111 
15112   switch (E->getStmtClass()) {
15113 #define ABSTRACT_STMT(Node)
15114 #define STMT(Node, Base) case Expr::Node##Class:
15115 #define EXPR(Node, Base)
15116 #include "clang/AST/StmtNodes.inc"
15117   case Expr::PredefinedExprClass:
15118   case Expr::FloatingLiteralClass:
15119   case Expr::ImaginaryLiteralClass:
15120   case Expr::StringLiteralClass:
15121   case Expr::ArraySubscriptExprClass:
15122   case Expr::MatrixSubscriptExprClass:
15123   case Expr::OMPArraySectionExprClass:
15124   case Expr::OMPArrayShapingExprClass:
15125   case Expr::OMPIteratorExprClass:
15126   case Expr::MemberExprClass:
15127   case Expr::CompoundAssignOperatorClass:
15128   case Expr::CompoundLiteralExprClass:
15129   case Expr::ExtVectorElementExprClass:
15130   case Expr::DesignatedInitExprClass:
15131   case Expr::ArrayInitLoopExprClass:
15132   case Expr::ArrayInitIndexExprClass:
15133   case Expr::NoInitExprClass:
15134   case Expr::DesignatedInitUpdateExprClass:
15135   case Expr::ImplicitValueInitExprClass:
15136   case Expr::ParenListExprClass:
15137   case Expr::VAArgExprClass:
15138   case Expr::AddrLabelExprClass:
15139   case Expr::StmtExprClass:
15140   case Expr::CXXMemberCallExprClass:
15141   case Expr::CUDAKernelCallExprClass:
15142   case Expr::CXXAddrspaceCastExprClass:
15143   case Expr::CXXDynamicCastExprClass:
15144   case Expr::CXXTypeidExprClass:
15145   case Expr::CXXUuidofExprClass:
15146   case Expr::MSPropertyRefExprClass:
15147   case Expr::MSPropertySubscriptExprClass:
15148   case Expr::CXXNullPtrLiteralExprClass:
15149   case Expr::UserDefinedLiteralClass:
15150   case Expr::CXXThisExprClass:
15151   case Expr::CXXThrowExprClass:
15152   case Expr::CXXNewExprClass:
15153   case Expr::CXXDeleteExprClass:
15154   case Expr::CXXPseudoDestructorExprClass:
15155   case Expr::UnresolvedLookupExprClass:
15156   case Expr::TypoExprClass:
15157   case Expr::RecoveryExprClass:
15158   case Expr::DependentScopeDeclRefExprClass:
15159   case Expr::CXXConstructExprClass:
15160   case Expr::CXXInheritedCtorInitExprClass:
15161   case Expr::CXXStdInitializerListExprClass:
15162   case Expr::CXXBindTemporaryExprClass:
15163   case Expr::ExprWithCleanupsClass:
15164   case Expr::CXXTemporaryObjectExprClass:
15165   case Expr::CXXUnresolvedConstructExprClass:
15166   case Expr::CXXDependentScopeMemberExprClass:
15167   case Expr::UnresolvedMemberExprClass:
15168   case Expr::ObjCStringLiteralClass:
15169   case Expr::ObjCBoxedExprClass:
15170   case Expr::ObjCArrayLiteralClass:
15171   case Expr::ObjCDictionaryLiteralClass:
15172   case Expr::ObjCEncodeExprClass:
15173   case Expr::ObjCMessageExprClass:
15174   case Expr::ObjCSelectorExprClass:
15175   case Expr::ObjCProtocolExprClass:
15176   case Expr::ObjCIvarRefExprClass:
15177   case Expr::ObjCPropertyRefExprClass:
15178   case Expr::ObjCSubscriptRefExprClass:
15179   case Expr::ObjCIsaExprClass:
15180   case Expr::ObjCAvailabilityCheckExprClass:
15181   case Expr::ShuffleVectorExprClass:
15182   case Expr::ConvertVectorExprClass:
15183   case Expr::BlockExprClass:
15184   case Expr::NoStmtClass:
15185   case Expr::OpaqueValueExprClass:
15186   case Expr::PackExpansionExprClass:
15187   case Expr::SubstNonTypeTemplateParmPackExprClass:
15188   case Expr::FunctionParmPackExprClass:
15189   case Expr::AsTypeExprClass:
15190   case Expr::ObjCIndirectCopyRestoreExprClass:
15191   case Expr::MaterializeTemporaryExprClass:
15192   case Expr::PseudoObjectExprClass:
15193   case Expr::AtomicExprClass:
15194   case Expr::LambdaExprClass:
15195   case Expr::CXXFoldExprClass:
15196   case Expr::CoawaitExprClass:
15197   case Expr::DependentCoawaitExprClass:
15198   case Expr::CoyieldExprClass:
15199   case Expr::SYCLUniqueStableNameExprClass:
15200     return ICEDiag(IK_NotICE, E->getBeginLoc());
15201 
15202   case Expr::InitListExprClass: {
15203     // C++03 [dcl.init]p13: If T is a scalar type, then a declaration of the
15204     // form "T x = { a };" is equivalent to "T x = a;".
15205     // Unless we're initializing a reference, T is a scalar as it is known to be
15206     // of integral or enumeration type.
15207     if (E->isPRValue())
15208       if (cast<InitListExpr>(E)->getNumInits() == 1)
15209         return CheckICE(cast<InitListExpr>(E)->getInit(0), Ctx);
15210     return ICEDiag(IK_NotICE, E->getBeginLoc());
15211   }
15212 
15213   case Expr::SizeOfPackExprClass:
15214   case Expr::GNUNullExprClass:
15215   case Expr::SourceLocExprClass:
15216     return NoDiag();
15217 
15218   case Expr::SubstNonTypeTemplateParmExprClass:
15219     return
15220       CheckICE(cast<SubstNonTypeTemplateParmExpr>(E)->getReplacement(), Ctx);
15221 
15222   case Expr::ConstantExprClass:
15223     return CheckICE(cast<ConstantExpr>(E)->getSubExpr(), Ctx);
15224 
15225   case Expr::ParenExprClass:
15226     return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
15227   case Expr::GenericSelectionExprClass:
15228     return CheckICE(cast<GenericSelectionExpr>(E)->getResultExpr(), Ctx);
15229   case Expr::IntegerLiteralClass:
15230   case Expr::FixedPointLiteralClass:
15231   case Expr::CharacterLiteralClass:
15232   case Expr::ObjCBoolLiteralExprClass:
15233   case Expr::CXXBoolLiteralExprClass:
15234   case Expr::CXXScalarValueInitExprClass:
15235   case Expr::TypeTraitExprClass:
15236   case Expr::ConceptSpecializationExprClass:
15237   case Expr::RequiresExprClass:
15238   case Expr::ArrayTypeTraitExprClass:
15239   case Expr::ExpressionTraitExprClass:
15240   case Expr::CXXNoexceptExprClass:
15241     return NoDiag();
15242   case Expr::CallExprClass:
15243   case Expr::CXXOperatorCallExprClass: {
15244     // C99 6.6/3 allows function calls within unevaluated subexpressions of
15245     // constant expressions, but they can never be ICEs because an ICE cannot
15246     // contain an operand of (pointer to) function type.
15247     const CallExpr *CE = cast<CallExpr>(E);
15248     if (CE->getBuiltinCallee())
15249       return CheckEvalInICE(E, Ctx);
15250     return ICEDiag(IK_NotICE, E->getBeginLoc());
15251   }
15252   case Expr::CXXRewrittenBinaryOperatorClass:
15253     return CheckICE(cast<CXXRewrittenBinaryOperator>(E)->getSemanticForm(),
15254                     Ctx);
15255   case Expr::DeclRefExprClass: {
15256     const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
15257     if (isa<EnumConstantDecl>(D))
15258       return NoDiag();
15259 
15260     // C++ and OpenCL (FIXME: spec reference?) allow reading const-qualified
15261     // integer variables in constant expressions:
15262     //
15263     // C++ 7.1.5.1p2
15264     //   A variable of non-volatile const-qualified integral or enumeration
15265     //   type initialized by an ICE can be used in ICEs.
15266     //
15267     // We sometimes use CheckICE to check the C++98 rules in C++11 mode. In
15268     // that mode, use of reference variables should not be allowed.
15269     const VarDecl *VD = dyn_cast<VarDecl>(D);
15270     if (VD && VD->isUsableInConstantExpressions(Ctx) &&
15271         !VD->getType()->isReferenceType())
15272       return NoDiag();
15273 
15274     return ICEDiag(IK_NotICE, E->getBeginLoc());
15275   }
15276   case Expr::UnaryOperatorClass: {
15277     const UnaryOperator *Exp = cast<UnaryOperator>(E);
15278     switch (Exp->getOpcode()) {
15279     case UO_PostInc:
15280     case UO_PostDec:
15281     case UO_PreInc:
15282     case UO_PreDec:
15283     case UO_AddrOf:
15284     case UO_Deref:
15285     case UO_Coawait:
15286       // C99 6.6/3 allows increment and decrement within unevaluated
15287       // subexpressions of constant expressions, but they can never be ICEs
15288       // because an ICE cannot contain an lvalue operand.
15289       return ICEDiag(IK_NotICE, E->getBeginLoc());
15290     case UO_Extension:
15291     case UO_LNot:
15292     case UO_Plus:
15293     case UO_Minus:
15294     case UO_Not:
15295     case UO_Real:
15296     case UO_Imag:
15297       return CheckICE(Exp->getSubExpr(), Ctx);
15298     }
15299     llvm_unreachable("invalid unary operator class");
15300   }
15301   case Expr::OffsetOfExprClass: {
15302     // Note that per C99, offsetof must be an ICE. And AFAIK, using
15303     // EvaluateAsRValue matches the proposed gcc behavior for cases like
15304     // "offsetof(struct s{int x[4];}, x[1.0])".  This doesn't affect
15305     // compliance: we should warn earlier for offsetof expressions with
15306     // array subscripts that aren't ICEs, and if the array subscripts
15307     // are ICEs, the value of the offsetof must be an integer constant.
15308     return CheckEvalInICE(E, Ctx);
15309   }
15310   case Expr::UnaryExprOrTypeTraitExprClass: {
15311     const UnaryExprOrTypeTraitExpr *Exp = cast<UnaryExprOrTypeTraitExpr>(E);
15312     if ((Exp->getKind() ==  UETT_SizeOf) &&
15313         Exp->getTypeOfArgument()->isVariableArrayType())
15314       return ICEDiag(IK_NotICE, E->getBeginLoc());
15315     return NoDiag();
15316   }
15317   case Expr::BinaryOperatorClass: {
15318     const BinaryOperator *Exp = cast<BinaryOperator>(E);
15319     switch (Exp->getOpcode()) {
15320     case BO_PtrMemD:
15321     case BO_PtrMemI:
15322     case BO_Assign:
15323     case BO_MulAssign:
15324     case BO_DivAssign:
15325     case BO_RemAssign:
15326     case BO_AddAssign:
15327     case BO_SubAssign:
15328     case BO_ShlAssign:
15329     case BO_ShrAssign:
15330     case BO_AndAssign:
15331     case BO_XorAssign:
15332     case BO_OrAssign:
15333       // C99 6.6/3 allows assignments within unevaluated subexpressions of
15334       // constant expressions, but they can never be ICEs because an ICE cannot
15335       // contain an lvalue operand.
15336       return ICEDiag(IK_NotICE, E->getBeginLoc());
15337 
15338     case BO_Mul:
15339     case BO_Div:
15340     case BO_Rem:
15341     case BO_Add:
15342     case BO_Sub:
15343     case BO_Shl:
15344     case BO_Shr:
15345     case BO_LT:
15346     case BO_GT:
15347     case BO_LE:
15348     case BO_GE:
15349     case BO_EQ:
15350     case BO_NE:
15351     case BO_And:
15352     case BO_Xor:
15353     case BO_Or:
15354     case BO_Comma:
15355     case BO_Cmp: {
15356       ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
15357       ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
15358       if (Exp->getOpcode() == BO_Div ||
15359           Exp->getOpcode() == BO_Rem) {
15360         // EvaluateAsRValue gives an error for undefined Div/Rem, so make sure
15361         // we don't evaluate one.
15362         if (LHSResult.Kind == IK_ICE && RHSResult.Kind == IK_ICE) {
15363           llvm::APSInt REval = Exp->getRHS()->EvaluateKnownConstInt(Ctx);
15364           if (REval == 0)
15365             return ICEDiag(IK_ICEIfUnevaluated, E->getBeginLoc());
15366           if (REval.isSigned() && REval.isAllOnesValue()) {
15367             llvm::APSInt LEval = Exp->getLHS()->EvaluateKnownConstInt(Ctx);
15368             if (LEval.isMinSignedValue())
15369               return ICEDiag(IK_ICEIfUnevaluated, E->getBeginLoc());
15370           }
15371         }
15372       }
15373       if (Exp->getOpcode() == BO_Comma) {
15374         if (Ctx.getLangOpts().C99) {
15375           // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
15376           // if it isn't evaluated.
15377           if (LHSResult.Kind == IK_ICE && RHSResult.Kind == IK_ICE)
15378             return ICEDiag(IK_ICEIfUnevaluated, E->getBeginLoc());
15379         } else {
15380           // In both C89 and C++, commas in ICEs are illegal.
15381           return ICEDiag(IK_NotICE, E->getBeginLoc());
15382         }
15383       }
15384       return Worst(LHSResult, RHSResult);
15385     }
15386     case BO_LAnd:
15387     case BO_LOr: {
15388       ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
15389       ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
15390       if (LHSResult.Kind == IK_ICE && RHSResult.Kind == IK_ICEIfUnevaluated) {
15391         // Rare case where the RHS has a comma "side-effect"; we need
15392         // to actually check the condition to see whether the side
15393         // with the comma is evaluated.
15394         if ((Exp->getOpcode() == BO_LAnd) !=
15395             (Exp->getLHS()->EvaluateKnownConstInt(Ctx) == 0))
15396           return RHSResult;
15397         return NoDiag();
15398       }
15399 
15400       return Worst(LHSResult, RHSResult);
15401     }
15402     }
15403     llvm_unreachable("invalid binary operator kind");
15404   }
15405   case Expr::ImplicitCastExprClass:
15406   case Expr::CStyleCastExprClass:
15407   case Expr::CXXFunctionalCastExprClass:
15408   case Expr::CXXStaticCastExprClass:
15409   case Expr::CXXReinterpretCastExprClass:
15410   case Expr::CXXConstCastExprClass:
15411   case Expr::ObjCBridgedCastExprClass: {
15412     const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
15413     if (isa<ExplicitCastExpr>(E)) {
15414       if (const FloatingLiteral *FL
15415             = dyn_cast<FloatingLiteral>(SubExpr->IgnoreParenImpCasts())) {
15416         unsigned DestWidth = Ctx.getIntWidth(E->getType());
15417         bool DestSigned = E->getType()->isSignedIntegerOrEnumerationType();
15418         APSInt IgnoredVal(DestWidth, !DestSigned);
15419         bool Ignored;
15420         // If the value does not fit in the destination type, the behavior is
15421         // undefined, so we are not required to treat it as a constant
15422         // expression.
15423         if (FL->getValue().convertToInteger(IgnoredVal,
15424                                             llvm::APFloat::rmTowardZero,
15425                                             &Ignored) & APFloat::opInvalidOp)
15426           return ICEDiag(IK_NotICE, E->getBeginLoc());
15427         return NoDiag();
15428       }
15429     }
15430     switch (cast<CastExpr>(E)->getCastKind()) {
15431     case CK_LValueToRValue:
15432     case CK_AtomicToNonAtomic:
15433     case CK_NonAtomicToAtomic:
15434     case CK_NoOp:
15435     case CK_IntegralToBoolean:
15436     case CK_IntegralCast:
15437       return CheckICE(SubExpr, Ctx);
15438     default:
15439       return ICEDiag(IK_NotICE, E->getBeginLoc());
15440     }
15441   }
15442   case Expr::BinaryConditionalOperatorClass: {
15443     const BinaryConditionalOperator *Exp = cast<BinaryConditionalOperator>(E);
15444     ICEDiag CommonResult = CheckICE(Exp->getCommon(), Ctx);
15445     if (CommonResult.Kind == IK_NotICE) return CommonResult;
15446     ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
15447     if (FalseResult.Kind == IK_NotICE) return FalseResult;
15448     if (CommonResult.Kind == IK_ICEIfUnevaluated) return CommonResult;
15449     if (FalseResult.Kind == IK_ICEIfUnevaluated &&
15450         Exp->getCommon()->EvaluateKnownConstInt(Ctx) != 0) return NoDiag();
15451     return FalseResult;
15452   }
15453   case Expr::ConditionalOperatorClass: {
15454     const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
15455     // If the condition (ignoring parens) is a __builtin_constant_p call,
15456     // then only the true side is actually considered in an integer constant
15457     // expression, and it is fully evaluated.  This is an important GNU
15458     // extension.  See GCC PR38377 for discussion.
15459     if (const CallExpr *CallCE
15460         = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
15461       if (CallCE->getBuiltinCallee() == Builtin::BI__builtin_constant_p)
15462         return CheckEvalInICE(E, Ctx);
15463     ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
15464     if (CondResult.Kind == IK_NotICE)
15465       return CondResult;
15466 
15467     ICEDiag TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
15468     ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
15469 
15470     if (TrueResult.Kind == IK_NotICE)
15471       return TrueResult;
15472     if (FalseResult.Kind == IK_NotICE)
15473       return FalseResult;
15474     if (CondResult.Kind == IK_ICEIfUnevaluated)
15475       return CondResult;
15476     if (TrueResult.Kind == IK_ICE && FalseResult.Kind == IK_ICE)
15477       return NoDiag();
15478     // Rare case where the diagnostics depend on which side is evaluated
15479     // Note that if we get here, CondResult is 0, and at least one of
15480     // TrueResult and FalseResult is non-zero.
15481     if (Exp->getCond()->EvaluateKnownConstInt(Ctx) == 0)
15482       return FalseResult;
15483     return TrueResult;
15484   }
15485   case Expr::CXXDefaultArgExprClass:
15486     return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
15487   case Expr::CXXDefaultInitExprClass:
15488     return CheckICE(cast<CXXDefaultInitExpr>(E)->getExpr(), Ctx);
15489   case Expr::ChooseExprClass: {
15490     return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(), Ctx);
15491   }
15492   case Expr::BuiltinBitCastExprClass: {
15493     if (!checkBitCastConstexprEligibility(nullptr, Ctx, cast<CastExpr>(E)))
15494       return ICEDiag(IK_NotICE, E->getBeginLoc());
15495     return CheckICE(cast<CastExpr>(E)->getSubExpr(), Ctx);
15496   }
15497   }
15498 
15499   llvm_unreachable("Invalid StmtClass!");
15500 }
15501 
15502 /// Evaluate an expression as a C++11 integral constant expression.
EvaluateCPlusPlus11IntegralConstantExpr(const ASTContext & Ctx,const Expr * E,llvm::APSInt * Value,SourceLocation * Loc)15503 static bool EvaluateCPlusPlus11IntegralConstantExpr(const ASTContext &Ctx,
15504                                                     const Expr *E,
15505                                                     llvm::APSInt *Value,
15506                                                     SourceLocation *Loc) {
15507   if (!E->getType()->isIntegralOrUnscopedEnumerationType()) {
15508     if (Loc) *Loc = E->getExprLoc();
15509     return false;
15510   }
15511 
15512   APValue Result;
15513   if (!E->isCXX11ConstantExpr(Ctx, &Result, Loc))
15514     return false;
15515 
15516   if (!Result.isInt()) {
15517     if (Loc) *Loc = E->getExprLoc();
15518     return false;
15519   }
15520 
15521   if (Value) *Value = Result.getInt();
15522   return true;
15523 }
15524 
isIntegerConstantExpr(const ASTContext & Ctx,SourceLocation * Loc) const15525 bool Expr::isIntegerConstantExpr(const ASTContext &Ctx,
15526                                  SourceLocation *Loc) const {
15527   assert(!isValueDependent() &&
15528          "Expression evaluator can't be called on a dependent expression.");
15529 
15530   if (Ctx.getLangOpts().CPlusPlus11)
15531     return EvaluateCPlusPlus11IntegralConstantExpr(Ctx, this, nullptr, Loc);
15532 
15533   ICEDiag D = CheckICE(this, Ctx);
15534   if (D.Kind != IK_ICE) {
15535     if (Loc) *Loc = D.Loc;
15536     return false;
15537   }
15538   return true;
15539 }
15540 
getIntegerConstantExpr(const ASTContext & Ctx,SourceLocation * Loc,bool isEvaluated) const15541 Optional<llvm::APSInt> Expr::getIntegerConstantExpr(const ASTContext &Ctx,
15542                                                     SourceLocation *Loc,
15543                                                     bool isEvaluated) const {
15544   assert(!isValueDependent() &&
15545          "Expression evaluator can't be called on a dependent expression.");
15546 
15547   APSInt Value;
15548 
15549   if (Ctx.getLangOpts().CPlusPlus11) {
15550     if (EvaluateCPlusPlus11IntegralConstantExpr(Ctx, this, &Value, Loc))
15551       return Value;
15552     return None;
15553   }
15554 
15555   if (!isIntegerConstantExpr(Ctx, Loc))
15556     return None;
15557 
15558   // The only possible side-effects here are due to UB discovered in the
15559   // evaluation (for instance, INT_MAX + 1). In such a case, we are still
15560   // required to treat the expression as an ICE, so we produce the folded
15561   // value.
15562   EvalResult ExprResult;
15563   Expr::EvalStatus Status;
15564   EvalInfo Info(Ctx, Status, EvalInfo::EM_IgnoreSideEffects);
15565   Info.InConstantContext = true;
15566 
15567   if (!::EvaluateAsInt(this, ExprResult, Ctx, SE_AllowSideEffects, Info))
15568     llvm_unreachable("ICE cannot be evaluated!");
15569 
15570   return ExprResult.Val.getInt();
15571 }
15572 
isCXX98IntegralConstantExpr(const ASTContext & Ctx) const15573 bool Expr::isCXX98IntegralConstantExpr(const ASTContext &Ctx) const {
15574   assert(!isValueDependent() &&
15575          "Expression evaluator can't be called on a dependent expression.");
15576 
15577   return CheckICE(this, Ctx).Kind == IK_ICE;
15578 }
15579 
isCXX11ConstantExpr(const ASTContext & Ctx,APValue * Result,SourceLocation * Loc) const15580 bool Expr::isCXX11ConstantExpr(const ASTContext &Ctx, APValue *Result,
15581                                SourceLocation *Loc) const {
15582   assert(!isValueDependent() &&
15583          "Expression evaluator can't be called on a dependent expression.");
15584 
15585   // We support this checking in C++98 mode in order to diagnose compatibility
15586   // issues.
15587   assert(Ctx.getLangOpts().CPlusPlus);
15588 
15589   // Build evaluation settings.
15590   Expr::EvalStatus Status;
15591   SmallVector<PartialDiagnosticAt, 8> Diags;
15592   Status.Diag = &Diags;
15593   EvalInfo Info(Ctx, Status, EvalInfo::EM_ConstantExpression);
15594 
15595   APValue Scratch;
15596   bool IsConstExpr =
15597       ::EvaluateAsRValue(Info, this, Result ? *Result : Scratch) &&
15598       // FIXME: We don't produce a diagnostic for this, but the callers that
15599       // call us on arbitrary full-expressions should generally not care.
15600       Info.discardCleanups() && !Status.HasSideEffects;
15601 
15602   if (!Diags.empty()) {
15603     IsConstExpr = false;
15604     if (Loc) *Loc = Diags[0].first;
15605   } else if (!IsConstExpr) {
15606     // FIXME: This shouldn't happen.
15607     if (Loc) *Loc = getExprLoc();
15608   }
15609 
15610   return IsConstExpr;
15611 }
15612 
EvaluateWithSubstitution(APValue & Value,ASTContext & Ctx,const FunctionDecl * Callee,ArrayRef<const Expr * > Args,const Expr * This) const15613 bool Expr::EvaluateWithSubstitution(APValue &Value, ASTContext &Ctx,
15614                                     const FunctionDecl *Callee,
15615                                     ArrayRef<const Expr*> Args,
15616                                     const Expr *This) const {
15617   assert(!isValueDependent() &&
15618          "Expression evaluator can't be called on a dependent expression.");
15619 
15620   Expr::EvalStatus Status;
15621   EvalInfo Info(Ctx, Status, EvalInfo::EM_ConstantExpressionUnevaluated);
15622   Info.InConstantContext = true;
15623 
15624   LValue ThisVal;
15625   const LValue *ThisPtr = nullptr;
15626   if (This) {
15627 #ifndef NDEBUG
15628     auto *MD = dyn_cast<CXXMethodDecl>(Callee);
15629     assert(MD && "Don't provide `this` for non-methods.");
15630     assert(!MD->isStatic() && "Don't provide `this` for static methods.");
15631 #endif
15632     if (!This->isValueDependent() &&
15633         EvaluateObjectArgument(Info, This, ThisVal) &&
15634         !Info.EvalStatus.HasSideEffects)
15635       ThisPtr = &ThisVal;
15636 
15637     // Ignore any side-effects from a failed evaluation. This is safe because
15638     // they can't interfere with any other argument evaluation.
15639     Info.EvalStatus.HasSideEffects = false;
15640   }
15641 
15642   CallRef Call = Info.CurrentCall->createCall(Callee);
15643   for (ArrayRef<const Expr*>::iterator I = Args.begin(), E = Args.end();
15644        I != E; ++I) {
15645     unsigned Idx = I - Args.begin();
15646     if (Idx >= Callee->getNumParams())
15647       break;
15648     const ParmVarDecl *PVD = Callee->getParamDecl(Idx);
15649     if ((*I)->isValueDependent() ||
15650         !EvaluateCallArg(PVD, *I, Call, Info) ||
15651         Info.EvalStatus.HasSideEffects) {
15652       // If evaluation fails, throw away the argument entirely.
15653       if (APValue *Slot = Info.getParamSlot(Call, PVD))
15654         *Slot = APValue();
15655     }
15656 
15657     // Ignore any side-effects from a failed evaluation. This is safe because
15658     // they can't interfere with any other argument evaluation.
15659     Info.EvalStatus.HasSideEffects = false;
15660   }
15661 
15662   // Parameter cleanups happen in the caller and are not part of this
15663   // evaluation.
15664   Info.discardCleanups();
15665   Info.EvalStatus.HasSideEffects = false;
15666 
15667   // Build fake call to Callee.
15668   CallStackFrame Frame(Info, Callee->getLocation(), Callee, ThisPtr, Call);
15669   // FIXME: Missing ExprWithCleanups in enable_if conditions?
15670   FullExpressionRAII Scope(Info);
15671   return Evaluate(Value, Info, this) && Scope.destroy() &&
15672          !Info.EvalStatus.HasSideEffects;
15673 }
15674 
isPotentialConstantExpr(const FunctionDecl * FD,SmallVectorImpl<PartialDiagnosticAt> & Diags)15675 bool Expr::isPotentialConstantExpr(const FunctionDecl *FD,
15676                                    SmallVectorImpl<
15677                                      PartialDiagnosticAt> &Diags) {
15678   // FIXME: It would be useful to check constexpr function templates, but at the
15679   // moment the constant expression evaluator cannot cope with the non-rigorous
15680   // ASTs which we build for dependent expressions.
15681   if (FD->isDependentContext())
15682     return true;
15683 
15684   Expr::EvalStatus Status;
15685   Status.Diag = &Diags;
15686 
15687   EvalInfo Info(FD->getASTContext(), Status, EvalInfo::EM_ConstantExpression);
15688   Info.InConstantContext = true;
15689   Info.CheckingPotentialConstantExpression = true;
15690 
15691   // The constexpr VM attempts to compile all methods to bytecode here.
15692   if (Info.EnableNewConstInterp) {
15693     Info.Ctx.getInterpContext().isPotentialConstantExpr(Info, FD);
15694     return Diags.empty();
15695   }
15696 
15697   const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD);
15698   const CXXRecordDecl *RD = MD ? MD->getParent()->getCanonicalDecl() : nullptr;
15699 
15700   // Fabricate an arbitrary expression on the stack and pretend that it
15701   // is a temporary being used as the 'this' pointer.
15702   LValue This;
15703   ImplicitValueInitExpr VIE(RD ? Info.Ctx.getRecordType(RD) : Info.Ctx.IntTy);
15704   This.set({&VIE, Info.CurrentCall->Index});
15705 
15706   ArrayRef<const Expr*> Args;
15707 
15708   APValue Scratch;
15709   if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(FD)) {
15710     // Evaluate the call as a constant initializer, to allow the construction
15711     // of objects of non-literal types.
15712     Info.setEvaluatingDecl(This.getLValueBase(), Scratch);
15713     HandleConstructorCall(&VIE, This, Args, CD, Info, Scratch);
15714   } else {
15715     SourceLocation Loc = FD->getLocation();
15716     HandleFunctionCall(Loc, FD, (MD && MD->isInstance()) ? &This : nullptr,
15717                        Args, CallRef(), FD->getBody(), Info, Scratch, nullptr);
15718   }
15719 
15720   return Diags.empty();
15721 }
15722 
isPotentialConstantExprUnevaluated(Expr * E,const FunctionDecl * FD,SmallVectorImpl<PartialDiagnosticAt> & Diags)15723 bool Expr::isPotentialConstantExprUnevaluated(Expr *E,
15724                                               const FunctionDecl *FD,
15725                                               SmallVectorImpl<
15726                                                 PartialDiagnosticAt> &Diags) {
15727   assert(!E->isValueDependent() &&
15728          "Expression evaluator can't be called on a dependent expression.");
15729 
15730   Expr::EvalStatus Status;
15731   Status.Diag = &Diags;
15732 
15733   EvalInfo Info(FD->getASTContext(), Status,
15734                 EvalInfo::EM_ConstantExpressionUnevaluated);
15735   Info.InConstantContext = true;
15736   Info.CheckingPotentialConstantExpression = true;
15737 
15738   // Fabricate a call stack frame to give the arguments a plausible cover story.
15739   CallStackFrame Frame(Info, SourceLocation(), FD, /*This*/ nullptr, CallRef());
15740 
15741   APValue ResultScratch;
15742   Evaluate(ResultScratch, Info, E);
15743   return Diags.empty();
15744 }
15745 
tryEvaluateObjectSize(uint64_t & Result,ASTContext & Ctx,unsigned Type) const15746 bool Expr::tryEvaluateObjectSize(uint64_t &Result, ASTContext &Ctx,
15747                                  unsigned Type) const {
15748   if (!getType()->isPointerType())
15749     return false;
15750 
15751   Expr::EvalStatus Status;
15752   EvalInfo Info(Ctx, Status, EvalInfo::EM_ConstantFold);
15753   return tryEvaluateBuiltinObjectSize(this, Type, Info, Result);
15754 }
15755