blob: 97786053be1833120360896b1d9620f0ccfb6daa [file] [log] [blame]
Chris Lattnere13042c2008-07-11 19:10:17 +00001//===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
Anders Carlsson7a241ba2008-07-03 04:20:39 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Expr constant evaluator.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/APValue.h"
15#include "clang/AST/ASTContext.h"
Ken Dyck40775002010-01-11 17:06:35 +000016#include "clang/AST/CharUnits.h"
Anders Carlsson15b73de2009-07-18 19:43:29 +000017#include "clang/AST/RecordLayout.h"
Seo Sanghyeon1904f442008-07-08 07:23:12 +000018#include "clang/AST/StmtVisitor.h"
Douglas Gregor882211c2010-04-28 22:16:22 +000019#include "clang/AST/TypeLoc.h"
Chris Lattner60f36222009-01-29 05:15:15 +000020#include "clang/AST/ASTDiagnostic.h"
Douglas Gregor882211c2010-04-28 22:16:22 +000021#include "clang/AST/Expr.h"
Chris Lattner15ba9492009-06-14 01:54:56 +000022#include "clang/Basic/Builtins.h"
Anders Carlsson374b93d2008-07-08 05:49:43 +000023#include "clang/Basic/TargetInfo.h"
Mike Stumpb807c9c2009-05-30 14:43:18 +000024#include "llvm/ADT/SmallString.h"
Mike Stump2346cd22009-05-30 03:56:50 +000025#include <cstring>
26
Anders Carlsson7a241ba2008-07-03 04:20:39 +000027using namespace clang;
Chris Lattner05706e882008-07-11 18:11:29 +000028using llvm::APSInt;
Eli Friedman24c01542008-08-22 00:06:13 +000029using llvm::APFloat;
Anders Carlsson7a241ba2008-07-03 04:20:39 +000030
Chris Lattnercdf34e72008-07-11 22:52:41 +000031/// EvalInfo - This is a private struct used by the evaluator to capture
32/// information about a subexpression as it is folded. It retains information
33/// about the AST context, but also maintains information about the folded
34/// expression.
35///
36/// If an expression could be evaluated, it is still possible it is not a C
37/// "integer constant expression" or constant expression. If not, this struct
38/// captures information about how and why not.
39///
40/// One bit of information passed *into* the request for constant folding
41/// indicates whether the subexpression is "evaluated" or not according to C
42/// rules. For example, the RHS of (0 && foo()) is not evaluated. We can
43/// evaluate the expression regardless of what the RHS is, but C only allows
44/// certain things in certain situations.
John McCall93d91dc2010-05-07 17:22:02 +000045namespace {
Benjamin Kramer024e6192011-03-04 13:12:48 +000046 struct EvalInfo {
47 const ASTContext &Ctx;
48
49 /// EvalResult - Contains information about the evaluation.
50 Expr::EvalResult &EvalResult;
51
52 typedef llvm::DenseMap<const OpaqueValueExpr*, APValue> MapTy;
53 MapTy OpaqueValues;
54 const APValue *getOpaqueValue(const OpaqueValueExpr *e) const {
55 MapTy::const_iterator i = OpaqueValues.find(e);
56 if (i == OpaqueValues.end()) return 0;
57 return &i->second;
58 }
59
60 EvalInfo(const ASTContext &ctx, Expr::EvalResult &evalresult)
61 : Ctx(ctx), EvalResult(evalresult) {}
62 };
63
John McCall93d91dc2010-05-07 17:22:02 +000064 struct ComplexValue {
65 private:
66 bool IsInt;
67
68 public:
69 APSInt IntReal, IntImag;
70 APFloat FloatReal, FloatImag;
71
72 ComplexValue() : FloatReal(APFloat::Bogus), FloatImag(APFloat::Bogus) {}
73
74 void makeComplexFloat() { IsInt = false; }
75 bool isComplexFloat() const { return !IsInt; }
76 APFloat &getComplexFloatReal() { return FloatReal; }
77 APFloat &getComplexFloatImag() { return FloatImag; }
78
79 void makeComplexInt() { IsInt = true; }
80 bool isComplexInt() const { return IsInt; }
81 APSInt &getComplexIntReal() { return IntReal; }
82 APSInt &getComplexIntImag() { return IntImag; }
83
John McCallc07a0c72011-02-17 10:25:35 +000084 void moveInto(APValue &v) const {
John McCall93d91dc2010-05-07 17:22:02 +000085 if (isComplexFloat())
86 v = APValue(FloatReal, FloatImag);
87 else
88 v = APValue(IntReal, IntImag);
89 }
John McCallc07a0c72011-02-17 10:25:35 +000090 void setFrom(const APValue &v) {
91 assert(v.isComplexFloat() || v.isComplexInt());
92 if (v.isComplexFloat()) {
93 makeComplexFloat();
94 FloatReal = v.getComplexFloatReal();
95 FloatImag = v.getComplexFloatImag();
96 } else {
97 makeComplexInt();
98 IntReal = v.getComplexIntReal();
99 IntImag = v.getComplexIntImag();
100 }
101 }
John McCall93d91dc2010-05-07 17:22:02 +0000102 };
John McCall45d55e42010-05-07 21:00:08 +0000103
104 struct LValue {
Peter Collingbournee9200682011-05-13 03:29:01 +0000105 const Expr *Base;
John McCall45d55e42010-05-07 21:00:08 +0000106 CharUnits Offset;
107
Peter Collingbournee9200682011-05-13 03:29:01 +0000108 const Expr *getLValueBase() { return Base; }
John McCall45d55e42010-05-07 21:00:08 +0000109 CharUnits getLValueOffset() { return Offset; }
110
John McCallc07a0c72011-02-17 10:25:35 +0000111 void moveInto(APValue &v) const {
John McCall45d55e42010-05-07 21:00:08 +0000112 v = APValue(Base, Offset);
113 }
John McCallc07a0c72011-02-17 10:25:35 +0000114 void setFrom(const APValue &v) {
115 assert(v.isLValue());
116 Base = v.getLValueBase();
117 Offset = v.getLValueOffset();
118 }
John McCall45d55e42010-05-07 21:00:08 +0000119 };
John McCall93d91dc2010-05-07 17:22:02 +0000120}
Chris Lattnercdf34e72008-07-11 22:52:41 +0000121
John McCallc07a0c72011-02-17 10:25:35 +0000122static bool Evaluate(EvalInfo &info, const Expr *E);
John McCall45d55e42010-05-07 21:00:08 +0000123static bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info);
124static bool EvaluatePointer(const Expr *E, LValue &Result, EvalInfo &Info);
Chris Lattnercdf34e72008-07-11 22:52:41 +0000125static bool EvaluateInteger(const Expr *E, APSInt &Result, EvalInfo &Info);
Chris Lattner6c4d2552009-10-28 23:59:40 +0000126static bool EvaluateIntegerOrLValue(const Expr *E, APValue &Result,
127 EvalInfo &Info);
Eli Friedman24c01542008-08-22 00:06:13 +0000128static bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
John McCall93d91dc2010-05-07 17:22:02 +0000129static bool EvaluateComplex(const Expr *E, ComplexValue &Res, EvalInfo &Info);
Chris Lattner05706e882008-07-11 18:11:29 +0000130
131//===----------------------------------------------------------------------===//
Eli Friedman9a156e52008-11-12 09:44:48 +0000132// Misc utilities
133//===----------------------------------------------------------------------===//
134
Abramo Bagnaraf8199452010-05-14 17:07:14 +0000135static bool IsGlobalLValue(const Expr* E) {
John McCall95007602010-05-10 23:27:23 +0000136 if (!E) return true;
137
138 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
139 if (isa<FunctionDecl>(DRE->getDecl()))
140 return true;
141 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
142 return VD->hasGlobalStorage();
143 return false;
144 }
145
146 if (const CompoundLiteralExpr *CLE = dyn_cast<CompoundLiteralExpr>(E))
147 return CLE->isFileScope();
148
149 return true;
150}
151
John McCall45d55e42010-05-07 21:00:08 +0000152static bool EvalPointerValueAsBool(LValue& Value, bool& Result) {
153 const Expr* Base = Value.Base;
Rafael Espindolaa1f9cc12010-05-07 15:18:43 +0000154
John McCalleb3e4f32010-05-07 21:34:32 +0000155 // A null base expression indicates a null pointer. These are always
156 // evaluatable, and they are false unless the offset is zero.
157 if (!Base) {
158 Result = !Value.Offset.isZero();
159 return true;
160 }
Rafael Espindolaa1f9cc12010-05-07 15:18:43 +0000161
John McCall95007602010-05-10 23:27:23 +0000162 // Require the base expression to be a global l-value.
Abramo Bagnaraf8199452010-05-14 17:07:14 +0000163 if (!IsGlobalLValue(Base)) return false;
John McCall95007602010-05-10 23:27:23 +0000164
John McCalleb3e4f32010-05-07 21:34:32 +0000165 // We have a non-null base expression. These are generally known to
166 // be true, but if it'a decl-ref to a weak symbol it can be null at
167 // runtime.
John McCalleb3e4f32010-05-07 21:34:32 +0000168 Result = true;
169
170 const DeclRefExpr* DeclRef = dyn_cast<DeclRefExpr>(Base);
Rafael Espindolaa1f9cc12010-05-07 15:18:43 +0000171 if (!DeclRef)
172 return true;
173
John McCalleb3e4f32010-05-07 21:34:32 +0000174 // If it's a weak symbol, it isn't constant-evaluable.
Rafael Espindolaa1f9cc12010-05-07 15:18:43 +0000175 const ValueDecl* Decl = DeclRef->getDecl();
176 if (Decl->hasAttr<WeakAttr>() ||
177 Decl->hasAttr<WeakRefAttr>() ||
Douglas Gregor20b2ebd2011-03-23 00:50:03 +0000178 Decl->isWeakImported())
Rafael Espindolaa1f9cc12010-05-07 15:18:43 +0000179 return false;
180
Eli Friedman334046a2009-06-14 02:17:33 +0000181 return true;
182}
183
John McCall1be1c632010-01-05 23:42:56 +0000184static bool HandleConversionToBool(const Expr* E, bool& Result,
185 EvalInfo &Info) {
Douglas Gregorb90df602010-06-16 00:17:44 +0000186 if (E->getType()->isIntegralOrEnumerationType()) {
Eli Friedman9a156e52008-11-12 09:44:48 +0000187 APSInt IntResult;
188 if (!EvaluateInteger(E, IntResult, Info))
189 return false;
190 Result = IntResult != 0;
191 return true;
192 } else if (E->getType()->isRealFloatingType()) {
193 APFloat FloatResult(0.0);
194 if (!EvaluateFloat(E, FloatResult, Info))
195 return false;
196 Result = !FloatResult.isZero();
197 return true;
Eli Friedman64004332009-03-23 04:38:34 +0000198 } else if (E->getType()->hasPointerRepresentation()) {
John McCall45d55e42010-05-07 21:00:08 +0000199 LValue PointerResult;
Eli Friedman9a156e52008-11-12 09:44:48 +0000200 if (!EvaluatePointer(E, PointerResult, Info))
201 return false;
Eli Friedman334046a2009-06-14 02:17:33 +0000202 return EvalPointerValueAsBool(PointerResult, Result);
Eli Friedman64004332009-03-23 04:38:34 +0000203 } else if (E->getType()->isAnyComplexType()) {
John McCall93d91dc2010-05-07 17:22:02 +0000204 ComplexValue ComplexResult;
Eli Friedman64004332009-03-23 04:38:34 +0000205 if (!EvaluateComplex(E, ComplexResult, Info))
206 return false;
207 if (ComplexResult.isComplexFloat()) {
208 Result = !ComplexResult.getComplexFloatReal().isZero() ||
209 !ComplexResult.getComplexFloatImag().isZero();
210 } else {
211 Result = ComplexResult.getComplexIntReal().getBoolValue() ||
212 ComplexResult.getComplexIntImag().getBoolValue();
213 }
214 return true;
Eli Friedman9a156e52008-11-12 09:44:48 +0000215 }
216
217 return false;
218}
219
Mike Stump11289f42009-09-09 15:08:12 +0000220static APSInt HandleFloatToIntCast(QualType DestType, QualType SrcType,
Jay Foad39c79802011-01-12 09:06:06 +0000221 APFloat &Value, const ASTContext &Ctx) {
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000222 unsigned DestWidth = Ctx.getIntWidth(DestType);
223 // Determine whether we are converting to unsigned or signed.
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000224 bool DestSigned = DestType->isSignedIntegerOrEnumerationType();
Mike Stump11289f42009-09-09 15:08:12 +0000225
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000226 // FIXME: Warning for overflow.
Jeffrey Yasskind0f079d2011-07-15 17:03:07 +0000227 APSInt Result(DestWidth, !DestSigned);
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000228 bool ignored;
Jeffrey Yasskind0f079d2011-07-15 17:03:07 +0000229 (void)Value.convertToInteger(Result, llvm::APFloat::rmTowardZero, &ignored);
230 return Result;
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000231}
232
Mike Stump11289f42009-09-09 15:08:12 +0000233static APFloat HandleFloatToFloatCast(QualType DestType, QualType SrcType,
Jay Foad39c79802011-01-12 09:06:06 +0000234 APFloat &Value, const ASTContext &Ctx) {
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000235 bool ignored;
236 APFloat Result = Value;
Mike Stump11289f42009-09-09 15:08:12 +0000237 Result.convert(Ctx.getFloatTypeSemantics(DestType),
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000238 APFloat::rmNearestTiesToEven, &ignored);
239 return Result;
240}
241
Mike Stump11289f42009-09-09 15:08:12 +0000242static APSInt HandleIntToIntCast(QualType DestType, QualType SrcType,
Jay Foad39c79802011-01-12 09:06:06 +0000243 APSInt &Value, const ASTContext &Ctx) {
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000244 unsigned DestWidth = Ctx.getIntWidth(DestType);
245 APSInt Result = Value;
246 // Figure out if this is a truncate, extend or noop cast.
247 // If the input is signed, do a sign extend, noop, or truncate.
Jay Foad6d4db0c2010-12-07 08:25:34 +0000248 Result = Result.extOrTrunc(DestWidth);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000249 Result.setIsUnsigned(DestType->isUnsignedIntegerOrEnumerationType());
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000250 return Result;
251}
252
Mike Stump11289f42009-09-09 15:08:12 +0000253static APFloat HandleIntToFloatCast(QualType DestType, QualType SrcType,
Jay Foad39c79802011-01-12 09:06:06 +0000254 APSInt &Value, const ASTContext &Ctx) {
Daniel Dunbarb6f953e2009-01-29 06:16:07 +0000255
256 APFloat Result(Ctx.getFloatTypeSemantics(DestType), 1);
257 Result.convertFromAPInt(Value, Value.isSigned(),
258 APFloat::rmNearestTiesToEven);
259 return Result;
260}
261
Mike Stump876387b2009-10-27 22:09:17 +0000262namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +0000263class HasSideEffect
Peter Collingbournee9200682011-05-13 03:29:01 +0000264 : public ConstStmtVisitor<HasSideEffect, bool> {
Mike Stump876387b2009-10-27 22:09:17 +0000265 EvalInfo &Info;
266public:
267
268 HasSideEffect(EvalInfo &info) : Info(info) {}
269
270 // Unhandled nodes conservatively default to having side effects.
Peter Collingbournee9200682011-05-13 03:29:01 +0000271 bool VisitStmt(const Stmt *S) {
Mike Stump876387b2009-10-27 22:09:17 +0000272 return true;
273 }
274
Peter Collingbournee9200682011-05-13 03:29:01 +0000275 bool VisitParenExpr(const ParenExpr *E) { return Visit(E->getSubExpr()); }
276 bool VisitGenericSelectionExpr(const GenericSelectionExpr *E) {
Peter Collingbourne91147592011-04-15 00:35:48 +0000277 return Visit(E->getResultExpr());
278 }
Peter Collingbournee9200682011-05-13 03:29:01 +0000279 bool VisitDeclRefExpr(const DeclRefExpr *E) {
Mike Stump53f9ded2009-11-03 23:25:48 +0000280 if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
Mike Stump876387b2009-10-27 22:09:17 +0000281 return true;
282 return false;
283 }
John McCall31168b02011-06-15 23:02:42 +0000284 bool VisitObjCIvarRefExpr(const ObjCIvarRefExpr *E) {
285 if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
286 return true;
287 return false;
288 }
289 bool VisitBlockDeclRefExpr (const BlockDeclRefExpr *E) {
290 if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
291 return true;
292 return false;
293 }
294
Mike Stump876387b2009-10-27 22:09:17 +0000295 // We don't want to evaluate BlockExprs multiple times, as they generate
296 // a ton of code.
Peter Collingbournee9200682011-05-13 03:29:01 +0000297 bool VisitBlockExpr(const BlockExpr *E) { return true; }
298 bool VisitPredefinedExpr(const PredefinedExpr *E) { return false; }
299 bool VisitCompoundLiteralExpr(const CompoundLiteralExpr *E)
Mike Stump876387b2009-10-27 22:09:17 +0000300 { return Visit(E->getInitializer()); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000301 bool VisitMemberExpr(const MemberExpr *E) { return Visit(E->getBase()); }
302 bool VisitIntegerLiteral(const IntegerLiteral *E) { return false; }
303 bool VisitFloatingLiteral(const FloatingLiteral *E) { return false; }
304 bool VisitStringLiteral(const StringLiteral *E) { return false; }
305 bool VisitCharacterLiteral(const CharacterLiteral *E) { return false; }
306 bool VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E)
Peter Collingbournee190dee2011-03-11 19:24:49 +0000307 { return false; }
Peter Collingbournee9200682011-05-13 03:29:01 +0000308 bool VisitArraySubscriptExpr(const ArraySubscriptExpr *E)
Mike Stumpfa502902009-10-29 20:48:09 +0000309 { return Visit(E->getLHS()) || Visit(E->getRHS()); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000310 bool VisitChooseExpr(const ChooseExpr *E)
Mike Stump876387b2009-10-27 22:09:17 +0000311 { return Visit(E->getChosenSubExpr(Info.Ctx)); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000312 bool VisitCastExpr(const CastExpr *E) { return Visit(E->getSubExpr()); }
313 bool VisitBinAssign(const BinaryOperator *E) { return true; }
314 bool VisitCompoundAssignOperator(const BinaryOperator *E) { return true; }
315 bool VisitBinaryOperator(const BinaryOperator *E)
Mike Stumpfa502902009-10-29 20:48:09 +0000316 { return Visit(E->getLHS()) || Visit(E->getRHS()); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000317 bool VisitUnaryPreInc(const UnaryOperator *E) { return true; }
318 bool VisitUnaryPostInc(const UnaryOperator *E) { return true; }
319 bool VisitUnaryPreDec(const UnaryOperator *E) { return true; }
320 bool VisitUnaryPostDec(const UnaryOperator *E) { return true; }
321 bool VisitUnaryDeref(const UnaryOperator *E) {
Mike Stump53f9ded2009-11-03 23:25:48 +0000322 if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
Mike Stump876387b2009-10-27 22:09:17 +0000323 return true;
Mike Stumpfa502902009-10-29 20:48:09 +0000324 return Visit(E->getSubExpr());
Mike Stump876387b2009-10-27 22:09:17 +0000325 }
Peter Collingbournee9200682011-05-13 03:29:01 +0000326 bool VisitUnaryOperator(const UnaryOperator *E) { return Visit(E->getSubExpr()); }
Chris Lattnera0679422010-04-13 17:34:23 +0000327
328 // Has side effects if any element does.
Peter Collingbournee9200682011-05-13 03:29:01 +0000329 bool VisitInitListExpr(const InitListExpr *E) {
Chris Lattnera0679422010-04-13 17:34:23 +0000330 for (unsigned i = 0, e = E->getNumInits(); i != e; ++i)
331 if (Visit(E->getInit(i))) return true;
Peter Collingbournee9200682011-05-13 03:29:01 +0000332 if (const Expr *filler = E->getArrayFiller())
Argyrios Kyrtzidisb2ed28e2011-04-21 00:27:41 +0000333 return Visit(filler);
Chris Lattnera0679422010-04-13 17:34:23 +0000334 return false;
335 }
Douglas Gregor820ba7b2011-01-04 17:33:58 +0000336
Peter Collingbournee9200682011-05-13 03:29:01 +0000337 bool VisitSizeOfPackExpr(const SizeOfPackExpr *) { return false; }
Mike Stump876387b2009-10-27 22:09:17 +0000338};
339
John McCallc07a0c72011-02-17 10:25:35 +0000340class OpaqueValueEvaluation {
341 EvalInfo &info;
342 OpaqueValueExpr *opaqueValue;
343
344public:
345 OpaqueValueEvaluation(EvalInfo &info, OpaqueValueExpr *opaqueValue,
346 Expr *value)
347 : info(info), opaqueValue(opaqueValue) {
348
349 // If evaluation fails, fail immediately.
350 if (!Evaluate(info, value)) {
351 this->opaqueValue = 0;
352 return;
353 }
354 info.OpaqueValues[opaqueValue] = info.EvalResult.Val;
355 }
356
357 bool hasError() const { return opaqueValue == 0; }
358
359 ~OpaqueValueEvaluation() {
360 if (opaqueValue) info.OpaqueValues.erase(opaqueValue);
361 }
362};
363
Mike Stump876387b2009-10-27 22:09:17 +0000364} // end anonymous namespace
365
Eli Friedman9a156e52008-11-12 09:44:48 +0000366//===----------------------------------------------------------------------===//
Peter Collingbournee9200682011-05-13 03:29:01 +0000367// Generic Evaluation
368//===----------------------------------------------------------------------===//
369namespace {
370
371template <class Derived, typename RetTy=void>
372class ExprEvaluatorBase
373 : public ConstStmtVisitor<Derived, RetTy> {
374private:
375 RetTy DerivedSuccess(const APValue &V, const Expr *E) {
376 return static_cast<Derived*>(this)->Success(V, E);
377 }
378 RetTy DerivedError(const Expr *E) {
379 return static_cast<Derived*>(this)->Error(E);
380 }
381
382protected:
383 EvalInfo &Info;
384 typedef ConstStmtVisitor<Derived, RetTy> StmtVisitorTy;
385 typedef ExprEvaluatorBase ExprEvaluatorBaseTy;
386
387public:
388 ExprEvaluatorBase(EvalInfo &Info) : Info(Info) {}
389
390 RetTy VisitStmt(const Stmt *) {
391 assert(0 && "Expression evaluator should not be called on stmts");
392 return DerivedError(0);
393 }
394 RetTy VisitExpr(const Expr *E) {
395 return DerivedError(E);
396 }
397
398 RetTy VisitParenExpr(const ParenExpr *E)
399 { return StmtVisitorTy::Visit(E->getSubExpr()); }
400 RetTy VisitUnaryExtension(const UnaryOperator *E)
401 { return StmtVisitorTy::Visit(E->getSubExpr()); }
402 RetTy VisitUnaryPlus(const UnaryOperator *E)
403 { return StmtVisitorTy::Visit(E->getSubExpr()); }
404 RetTy VisitChooseExpr(const ChooseExpr *E)
405 { return StmtVisitorTy::Visit(E->getChosenSubExpr(Info.Ctx)); }
406 RetTy VisitGenericSelectionExpr(const GenericSelectionExpr *E)
407 { return StmtVisitorTy::Visit(E->getResultExpr()); }
John McCall7c454bb2011-07-15 05:09:51 +0000408 RetTy VisitSubstNonTypeTemplateParmExpr(const SubstNonTypeTemplateParmExpr *E)
409 { return StmtVisitorTy::Visit(E->getReplacement()); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000410
411 RetTy VisitBinaryConditionalOperator(const BinaryConditionalOperator *E) {
412 OpaqueValueEvaluation opaque(Info, E->getOpaqueValue(), E->getCommon());
413 if (opaque.hasError())
414 return DerivedError(E);
415
416 bool cond;
417 if (!HandleConversionToBool(E->getCond(), cond, Info))
418 return DerivedError(E);
419
420 return StmtVisitorTy::Visit(cond ? E->getTrueExpr() : E->getFalseExpr());
421 }
422
423 RetTy VisitConditionalOperator(const ConditionalOperator *E) {
424 bool BoolResult;
425 if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
426 return DerivedError(E);
427
428 Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
429 return StmtVisitorTy::Visit(EvalExpr);
430 }
431
432 RetTy VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
433 const APValue *value = Info.getOpaqueValue(E);
434 if (!value)
435 return (E->getSourceExpr() ? StmtVisitorTy::Visit(E->getSourceExpr())
436 : DerivedError(E));
437 return DerivedSuccess(*value, E);
438 }
439};
440
441}
442
443//===----------------------------------------------------------------------===//
Eli Friedman9a156e52008-11-12 09:44:48 +0000444// LValue Evaluation
445//===----------------------------------------------------------------------===//
446namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +0000447class LValueExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +0000448 : public ExprEvaluatorBase<LValueExprEvaluator, bool> {
John McCall45d55e42010-05-07 21:00:08 +0000449 LValue &Result;
Chandler Carruth41c6dcc2011-08-22 17:24:56 +0000450 const Decl *PrevDecl;
John McCall45d55e42010-05-07 21:00:08 +0000451
Peter Collingbournee9200682011-05-13 03:29:01 +0000452 bool Success(const Expr *E) {
John McCall45d55e42010-05-07 21:00:08 +0000453 Result.Base = E;
454 Result.Offset = CharUnits::Zero();
455 return true;
456 }
Eli Friedman9a156e52008-11-12 09:44:48 +0000457public:
Mike Stump11289f42009-09-09 15:08:12 +0000458
John McCall45d55e42010-05-07 21:00:08 +0000459 LValueExprEvaluator(EvalInfo &info, LValue &Result) :
Chandler Carruth41c6dcc2011-08-22 17:24:56 +0000460 ExprEvaluatorBaseTy(info), Result(Result), PrevDecl(0) {}
Eli Friedman9a156e52008-11-12 09:44:48 +0000461
Peter Collingbournee9200682011-05-13 03:29:01 +0000462 bool Success(const APValue &V, const Expr *E) {
463 Result.setFrom(V);
464 return true;
465 }
466 bool Error(const Expr *E) {
John McCall45d55e42010-05-07 21:00:08 +0000467 return false;
Eli Friedman9a156e52008-11-12 09:44:48 +0000468 }
Douglas Gregor882211c2010-04-28 22:16:22 +0000469
Peter Collingbournee9200682011-05-13 03:29:01 +0000470 bool VisitDeclRefExpr(const DeclRefExpr *E);
471 bool VisitPredefinedExpr(const PredefinedExpr *E) { return Success(E); }
472 bool VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
473 bool VisitMemberExpr(const MemberExpr *E);
474 bool VisitStringLiteral(const StringLiteral *E) { return Success(E); }
475 bool VisitObjCEncodeExpr(const ObjCEncodeExpr *E) { return Success(E); }
476 bool VisitArraySubscriptExpr(const ArraySubscriptExpr *E);
477 bool VisitUnaryDeref(const UnaryOperator *E);
Anders Carlssonde55f642009-10-03 16:30:22 +0000478
Peter Collingbournee9200682011-05-13 03:29:01 +0000479 bool VisitCastExpr(const CastExpr *E) {
Anders Carlssonde55f642009-10-03 16:30:22 +0000480 switch (E->getCastKind()) {
481 default:
John McCall45d55e42010-05-07 21:00:08 +0000482 return false;
Anders Carlssonde55f642009-10-03 16:30:22 +0000483
John McCalle3027922010-08-25 11:45:40 +0000484 case CK_NoOp:
Anders Carlssonde55f642009-10-03 16:30:22 +0000485 return Visit(E->getSubExpr());
486 }
487 }
Eli Friedman449fe542009-03-23 04:56:01 +0000488 // FIXME: Missing: __real__, __imag__
Peter Collingbournee9200682011-05-13 03:29:01 +0000489
Eli Friedman9a156e52008-11-12 09:44:48 +0000490};
491} // end anonymous namespace
492
John McCall45d55e42010-05-07 21:00:08 +0000493static bool EvaluateLValue(const Expr* E, LValue& Result, EvalInfo &Info) {
Peter Collingbournee9200682011-05-13 03:29:01 +0000494 return LValueExprEvaluator(Info, Result).Visit(E);
Eli Friedman9a156e52008-11-12 09:44:48 +0000495}
496
Peter Collingbournee9200682011-05-13 03:29:01 +0000497bool LValueExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
Eli Friedman751aa72b72009-05-27 06:04:58 +0000498 if (isa<FunctionDecl>(E->getDecl())) {
John McCall45d55e42010-05-07 21:00:08 +0000499 return Success(E);
Peter Collingbournee9200682011-05-13 03:29:01 +0000500 } else if (const VarDecl* VD = dyn_cast<VarDecl>(E->getDecl())) {
Eli Friedman751aa72b72009-05-27 06:04:58 +0000501 if (!VD->getType()->isReferenceType())
John McCall45d55e42010-05-07 21:00:08 +0000502 return Success(E);
Chandler Carruthe299ba62010-05-16 09:32:51 +0000503 // Reference parameters can refer to anything even if they have an
504 // "initializer" in the form of a default argument.
Chandler Carruth41c6dcc2011-08-22 17:24:56 +0000505 if (!isa<ParmVarDecl>(VD)) {
Peter Collingbournee9200682011-05-13 03:29:01 +0000506 // FIXME: Check whether VD might be overridden!
Chandler Carruth41c6dcc2011-08-22 17:24:56 +0000507
508 // Check for recursive initializers of references.
509 if (PrevDecl == VD)
510 return Error(E);
511 PrevDecl = VD;
Peter Collingbournee9200682011-05-13 03:29:01 +0000512 if (const Expr *Init = VD->getAnyInitializer())
513 return Visit(Init);
Chandler Carruth41c6dcc2011-08-22 17:24:56 +0000514 }
Eli Friedman751aa72b72009-05-27 06:04:58 +0000515 }
516
Peter Collingbournee9200682011-05-13 03:29:01 +0000517 return ExprEvaluatorBaseTy::VisitDeclRefExpr(E);
Anders Carlssona42ee442008-11-24 04:41:22 +0000518}
519
Peter Collingbournee9200682011-05-13 03:29:01 +0000520bool
521LValueExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
John McCall45d55e42010-05-07 21:00:08 +0000522 return Success(E);
Eli Friedman9a156e52008-11-12 09:44:48 +0000523}
524
Peter Collingbournee9200682011-05-13 03:29:01 +0000525bool LValueExprEvaluator::VisitMemberExpr(const MemberExpr *E) {
Eli Friedman9a156e52008-11-12 09:44:48 +0000526 QualType Ty;
527 if (E->isArrow()) {
John McCall45d55e42010-05-07 21:00:08 +0000528 if (!EvaluatePointer(E->getBase(), Result, Info))
529 return false;
Ted Kremenekc23c7e62009-07-29 21:53:49 +0000530 Ty = E->getBase()->getType()->getAs<PointerType>()->getPointeeType();
Eli Friedman9a156e52008-11-12 09:44:48 +0000531 } else {
John McCall45d55e42010-05-07 21:00:08 +0000532 if (!Visit(E->getBase()))
533 return false;
Eli Friedman9a156e52008-11-12 09:44:48 +0000534 Ty = E->getBase()->getType();
535 }
536
Peter Collingbournee9200682011-05-13 03:29:01 +0000537 const RecordDecl *RD = Ty->getAs<RecordType>()->getDecl();
Eli Friedman9a156e52008-11-12 09:44:48 +0000538 const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
Douglas Gregor2eedc3a2008-12-20 23:49:58 +0000539
Peter Collingbournee9200682011-05-13 03:29:01 +0000540 const FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
Douglas Gregor2eedc3a2008-12-20 23:49:58 +0000541 if (!FD) // FIXME: deal with other kinds of member expressions
John McCall45d55e42010-05-07 21:00:08 +0000542 return false;
Eli Friedmanf7f9f682009-05-30 21:09:44 +0000543
544 if (FD->getType()->isReferenceType())
John McCall45d55e42010-05-07 21:00:08 +0000545 return false;
Eli Friedmanf7f9f682009-05-30 21:09:44 +0000546
Eli Friedmana3c122d2011-07-07 01:54:01 +0000547 unsigned i = FD->getFieldIndex();
Ken Dyck86a7fcc2011-01-18 01:56:16 +0000548 Result.Offset += Info.Ctx.toCharUnitsFromBits(RL.getFieldOffset(i));
John McCall45d55e42010-05-07 21:00:08 +0000549 return true;
Eli Friedman9a156e52008-11-12 09:44:48 +0000550}
551
Peter Collingbournee9200682011-05-13 03:29:01 +0000552bool LValueExprEvaluator::VisitArraySubscriptExpr(const ArraySubscriptExpr *E) {
Anders Carlsson9f9e4242008-11-16 19:01:22 +0000553 if (!EvaluatePointer(E->getBase(), Result, Info))
John McCall45d55e42010-05-07 21:00:08 +0000554 return false;
Mike Stump11289f42009-09-09 15:08:12 +0000555
Anders Carlsson9f9e4242008-11-16 19:01:22 +0000556 APSInt Index;
557 if (!EvaluateInteger(E->getIdx(), Index, Info))
John McCall45d55e42010-05-07 21:00:08 +0000558 return false;
Anders Carlsson9f9e4242008-11-16 19:01:22 +0000559
Ken Dyck40775002010-01-11 17:06:35 +0000560 CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(E->getType());
John McCall45d55e42010-05-07 21:00:08 +0000561 Result.Offset += Index.getSExtValue() * ElementSize;
562 return true;
Anders Carlsson9f9e4242008-11-16 19:01:22 +0000563}
Eli Friedman9a156e52008-11-12 09:44:48 +0000564
Peter Collingbournee9200682011-05-13 03:29:01 +0000565bool LValueExprEvaluator::VisitUnaryDeref(const UnaryOperator *E) {
John McCall45d55e42010-05-07 21:00:08 +0000566 return EvaluatePointer(E->getSubExpr(), Result, Info);
Eli Friedman0b8337c2009-02-20 01:57:15 +0000567}
568
Eli Friedman9a156e52008-11-12 09:44:48 +0000569//===----------------------------------------------------------------------===//
Chris Lattner05706e882008-07-11 18:11:29 +0000570// Pointer Evaluation
571//===----------------------------------------------------------------------===//
572
Anders Carlsson0a1707c2008-07-08 05:13:58 +0000573namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +0000574class PointerExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +0000575 : public ExprEvaluatorBase<PointerExprEvaluator, bool> {
John McCall45d55e42010-05-07 21:00:08 +0000576 LValue &Result;
577
Peter Collingbournee9200682011-05-13 03:29:01 +0000578 bool Success(const Expr *E) {
John McCall45d55e42010-05-07 21:00:08 +0000579 Result.Base = E;
580 Result.Offset = CharUnits::Zero();
581 return true;
582 }
Anders Carlssonb5ad0212008-07-08 14:30:00 +0000583public:
Mike Stump11289f42009-09-09 15:08:12 +0000584
John McCall45d55e42010-05-07 21:00:08 +0000585 PointerExprEvaluator(EvalInfo &info, LValue &Result)
Peter Collingbournee9200682011-05-13 03:29:01 +0000586 : ExprEvaluatorBaseTy(info), Result(Result) {}
Chris Lattner05706e882008-07-11 18:11:29 +0000587
Peter Collingbournee9200682011-05-13 03:29:01 +0000588 bool Success(const APValue &V, const Expr *E) {
589 Result.setFrom(V);
590 return true;
591 }
592 bool Error(const Stmt *S) {
John McCall45d55e42010-05-07 21:00:08 +0000593 return false;
Anders Carlssonb5ad0212008-07-08 14:30:00 +0000594 }
595
John McCall45d55e42010-05-07 21:00:08 +0000596 bool VisitBinaryOperator(const BinaryOperator *E);
Peter Collingbournee9200682011-05-13 03:29:01 +0000597 bool VisitCastExpr(const CastExpr* E);
John McCall45d55e42010-05-07 21:00:08 +0000598 bool VisitUnaryAddrOf(const UnaryOperator *E);
Peter Collingbournee9200682011-05-13 03:29:01 +0000599 bool VisitObjCStringLiteral(const ObjCStringLiteral *E)
John McCall45d55e42010-05-07 21:00:08 +0000600 { return Success(E); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000601 bool VisitAddrLabelExpr(const AddrLabelExpr *E)
John McCall45d55e42010-05-07 21:00:08 +0000602 { return Success(E); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000603 bool VisitCallExpr(const CallExpr *E);
604 bool VisitBlockExpr(const BlockExpr *E) {
John McCallc63de662011-02-02 13:00:07 +0000605 if (!E->getBlockDecl()->hasCaptures())
John McCall45d55e42010-05-07 21:00:08 +0000606 return Success(E);
607 return false;
Mike Stumpa6703322009-02-19 22:01:56 +0000608 }
Peter Collingbournee9200682011-05-13 03:29:01 +0000609 bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E)
John McCall45d55e42010-05-07 21:00:08 +0000610 { return Success((Expr*)0); }
Peter Collingbournee9200682011-05-13 03:29:01 +0000611 bool VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *E)
John McCall45d55e42010-05-07 21:00:08 +0000612 { return Success((Expr*)0); }
John McCallc07a0c72011-02-17 10:25:35 +0000613
Eli Friedman449fe542009-03-23 04:56:01 +0000614 // FIXME: Missing: @protocol, @selector
Anders Carlsson4a3585b2008-07-08 15:34:11 +0000615};
Chris Lattner05706e882008-07-11 18:11:29 +0000616} // end anonymous namespace
Anders Carlsson4a3585b2008-07-08 15:34:11 +0000617
John McCall45d55e42010-05-07 21:00:08 +0000618static bool EvaluatePointer(const Expr* E, LValue& Result, EvalInfo &Info) {
John McCallf0c4f352010-05-07 05:46:35 +0000619 assert(E->getType()->hasPointerRepresentation());
Peter Collingbournee9200682011-05-13 03:29:01 +0000620 return PointerExprEvaluator(Info, Result).Visit(E);
Chris Lattner05706e882008-07-11 18:11:29 +0000621}
622
John McCall45d55e42010-05-07 21:00:08 +0000623bool PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
John McCalle3027922010-08-25 11:45:40 +0000624 if (E->getOpcode() != BO_Add &&
625 E->getOpcode() != BO_Sub)
John McCall45d55e42010-05-07 21:00:08 +0000626 return false;
Mike Stump11289f42009-09-09 15:08:12 +0000627
Chris Lattner05706e882008-07-11 18:11:29 +0000628 const Expr *PExp = E->getLHS();
629 const Expr *IExp = E->getRHS();
630 if (IExp->getType()->isPointerType())
631 std::swap(PExp, IExp);
Mike Stump11289f42009-09-09 15:08:12 +0000632
John McCall45d55e42010-05-07 21:00:08 +0000633 if (!EvaluatePointer(PExp, Result, Info))
634 return false;
Mike Stump11289f42009-09-09 15:08:12 +0000635
John McCall45d55e42010-05-07 21:00:08 +0000636 llvm::APSInt Offset;
637 if (!EvaluateInteger(IExp, Offset, Info))
638 return false;
639 int64_t AdditionalOffset
640 = Offset.isSigned() ? Offset.getSExtValue()
641 : static_cast<int64_t>(Offset.getZExtValue());
Chris Lattner05706e882008-07-11 18:11:29 +0000642
Daniel Dunbar4c43e312010-03-20 05:53:45 +0000643 // Compute the new offset in the appropriate width.
644
645 QualType PointeeType =
646 PExp->getType()->getAs<PointerType>()->getPointeeType();
John McCall45d55e42010-05-07 21:00:08 +0000647 CharUnits SizeOfPointee;
Mike Stump11289f42009-09-09 15:08:12 +0000648
Anders Carlssonef56fba2009-02-19 04:55:58 +0000649 // Explicitly handle GNU void* and function pointer arithmetic extensions.
650 if (PointeeType->isVoidType() || PointeeType->isFunctionType())
John McCall45d55e42010-05-07 21:00:08 +0000651 SizeOfPointee = CharUnits::One();
Anders Carlssonef56fba2009-02-19 04:55:58 +0000652 else
John McCall45d55e42010-05-07 21:00:08 +0000653 SizeOfPointee = Info.Ctx.getTypeSizeInChars(PointeeType);
Eli Friedman9a156e52008-11-12 09:44:48 +0000654
John McCalle3027922010-08-25 11:45:40 +0000655 if (E->getOpcode() == BO_Add)
John McCall45d55e42010-05-07 21:00:08 +0000656 Result.Offset += AdditionalOffset * SizeOfPointee;
Chris Lattner05706e882008-07-11 18:11:29 +0000657 else
John McCall45d55e42010-05-07 21:00:08 +0000658 Result.Offset -= AdditionalOffset * SizeOfPointee;
Eli Friedman9a156e52008-11-12 09:44:48 +0000659
John McCall45d55e42010-05-07 21:00:08 +0000660 return true;
Chris Lattner05706e882008-07-11 18:11:29 +0000661}
Eli Friedman9a156e52008-11-12 09:44:48 +0000662
John McCall45d55e42010-05-07 21:00:08 +0000663bool PointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
664 return EvaluateLValue(E->getSubExpr(), Result, Info);
Eli Friedman9a156e52008-11-12 09:44:48 +0000665}
Mike Stump11289f42009-09-09 15:08:12 +0000666
Chris Lattner05706e882008-07-11 18:11:29 +0000667
Peter Collingbournee9200682011-05-13 03:29:01 +0000668bool PointerExprEvaluator::VisitCastExpr(const CastExpr* E) {
669 const Expr* SubExpr = E->getSubExpr();
Chris Lattner05706e882008-07-11 18:11:29 +0000670
Eli Friedman847a2bc2009-12-27 05:43:15 +0000671 switch (E->getCastKind()) {
672 default:
673 break;
674
John McCalle3027922010-08-25 11:45:40 +0000675 case CK_NoOp:
676 case CK_BitCast:
John McCall9320b872011-09-09 05:25:32 +0000677 case CK_CPointerToObjCPointerCast:
678 case CK_BlockPointerToObjCPointerCast:
John McCalle3027922010-08-25 11:45:40 +0000679 case CK_AnyPointerToBlockPointerCast:
Eli Friedman847a2bc2009-12-27 05:43:15 +0000680 return Visit(SubExpr);
681
Anders Carlsson18275092010-10-31 20:41:46 +0000682 case CK_DerivedToBase:
683 case CK_UncheckedDerivedToBase: {
684 LValue BaseLV;
685 if (!EvaluatePointer(E->getSubExpr(), BaseLV, Info))
686 return false;
687
688 // Now figure out the necessary offset to add to the baseLV to get from
689 // the derived class to the base class.
Ken Dyck02155cb2011-01-26 02:17:08 +0000690 CharUnits Offset = CharUnits::Zero();
Anders Carlsson18275092010-10-31 20:41:46 +0000691
692 QualType Ty = E->getSubExpr()->getType();
693 const CXXRecordDecl *DerivedDecl =
694 Ty->getAs<PointerType>()->getPointeeType()->getAsCXXRecordDecl();
695
696 for (CastExpr::path_const_iterator PathI = E->path_begin(),
697 PathE = E->path_end(); PathI != PathE; ++PathI) {
698 const CXXBaseSpecifier *Base = *PathI;
699
700 // FIXME: If the base is virtual, we'd need to determine the type of the
701 // most derived class and we don't support that right now.
702 if (Base->isVirtual())
703 return false;
704
705 const CXXRecordDecl *BaseDecl = Base->getType()->getAsCXXRecordDecl();
706 const ASTRecordLayout &Layout = Info.Ctx.getASTRecordLayout(DerivedDecl);
707
Ken Dyck02155cb2011-01-26 02:17:08 +0000708 Offset += Layout.getBaseClassOffset(BaseDecl);
Anders Carlsson18275092010-10-31 20:41:46 +0000709 DerivedDecl = BaseDecl;
710 }
711
712 Result.Base = BaseLV.getLValueBase();
Ken Dyck02155cb2011-01-26 02:17:08 +0000713 Result.Offset = BaseLV.getLValueOffset() + Offset;
Anders Carlsson18275092010-10-31 20:41:46 +0000714 return true;
715 }
716
John McCalle84af4e2010-11-13 01:35:44 +0000717 case CK_NullToPointer: {
718 Result.Base = 0;
719 Result.Offset = CharUnits::Zero();
720 return true;
721 }
722
John McCalle3027922010-08-25 11:45:40 +0000723 case CK_IntegralToPointer: {
John McCall45d55e42010-05-07 21:00:08 +0000724 APValue Value;
725 if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
Eli Friedman847a2bc2009-12-27 05:43:15 +0000726 break;
Daniel Dunbarce399542009-02-20 18:22:23 +0000727
John McCall45d55e42010-05-07 21:00:08 +0000728 if (Value.isInt()) {
Jay Foad6d4db0c2010-12-07 08:25:34 +0000729 Value.getInt() = Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
John McCall45d55e42010-05-07 21:00:08 +0000730 Result.Base = 0;
731 Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
732 return true;
733 } else {
734 // Cast is of an lvalue, no need to change value.
735 Result.Base = Value.getLValueBase();
736 Result.Offset = Value.getLValueOffset();
737 return true;
Chris Lattner05706e882008-07-11 18:11:29 +0000738 }
739 }
John McCalle3027922010-08-25 11:45:40 +0000740 case CK_ArrayToPointerDecay:
741 case CK_FunctionToPointerDecay:
John McCall45d55e42010-05-07 21:00:08 +0000742 return EvaluateLValue(SubExpr, Result, Info);
Eli Friedman9a156e52008-11-12 09:44:48 +0000743 }
744
John McCall45d55e42010-05-07 21:00:08 +0000745 return false;
Mike Stump11289f42009-09-09 15:08:12 +0000746}
Chris Lattner05706e882008-07-11 18:11:29 +0000747
Peter Collingbournee9200682011-05-13 03:29:01 +0000748bool PointerExprEvaluator::VisitCallExpr(const CallExpr *E) {
Mike Stump11289f42009-09-09 15:08:12 +0000749 if (E->isBuiltinCall(Info.Ctx) ==
David Chisnall481e3a82010-01-23 02:40:42 +0000750 Builtin::BI__builtin___CFStringMakeConstantString ||
751 E->isBuiltinCall(Info.Ctx) ==
752 Builtin::BI__builtin___NSStringMakeConstantString)
John McCall45d55e42010-05-07 21:00:08 +0000753 return Success(E);
Eli Friedmanc69d4542009-01-25 01:54:01 +0000754
Peter Collingbournee9200682011-05-13 03:29:01 +0000755 return ExprEvaluatorBaseTy::VisitCallExpr(E);
Eli Friedman9a156e52008-11-12 09:44:48 +0000756}
Chris Lattner05706e882008-07-11 18:11:29 +0000757
758//===----------------------------------------------------------------------===//
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000759// Vector Evaluation
760//===----------------------------------------------------------------------===//
761
762namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +0000763 class VectorExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +0000764 : public ExprEvaluatorBase<VectorExprEvaluator, APValue> {
Eli Friedman3ae59112009-02-23 04:23:56 +0000765 APValue GetZeroVector(QualType VecType);
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000766 public:
Mike Stump11289f42009-09-09 15:08:12 +0000767
Peter Collingbournee9200682011-05-13 03:29:01 +0000768 VectorExprEvaluator(EvalInfo &info) : ExprEvaluatorBaseTy(info) {}
Mike Stump11289f42009-09-09 15:08:12 +0000769
Peter Collingbournee9200682011-05-13 03:29:01 +0000770 APValue Success(const APValue &V, const Expr *E) { return V; }
771 APValue Error(const Expr *E) { return APValue(); }
Mike Stump11289f42009-09-09 15:08:12 +0000772
Eli Friedman3ae59112009-02-23 04:23:56 +0000773 APValue VisitUnaryReal(const UnaryOperator *E)
774 { return Visit(E->getSubExpr()); }
775 APValue VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E)
776 { return GetZeroVector(E->getType()); }
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000777 APValue VisitCastExpr(const CastExpr* E);
778 APValue VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
779 APValue VisitInitListExpr(const InitListExpr *E);
Eli Friedman3ae59112009-02-23 04:23:56 +0000780 APValue VisitUnaryImag(const UnaryOperator *E);
781 // FIXME: Missing: unary -, unary ~, binary add/sub/mul/div,
Eli Friedmanc2b50172009-02-22 11:46:18 +0000782 // binary comparisons, binary and/or/xor,
Eli Friedman3ae59112009-02-23 04:23:56 +0000783 // shufflevector, ExtVectorElementExpr
784 // (Note that these require implementing conversions
785 // between vector types.)
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000786 };
787} // end anonymous namespace
788
789static bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
790 if (!E->getType()->isVectorType())
791 return false;
Peter Collingbournee9200682011-05-13 03:29:01 +0000792 Result = VectorExprEvaluator(Info).Visit(E);
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000793 return !Result.isUninit();
794}
795
796APValue VectorExprEvaluator::VisitCastExpr(const CastExpr* E) {
John McCall9dd450b2009-09-21 23:43:11 +0000797 const VectorType *VTy = E->getType()->getAs<VectorType>();
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000798 QualType EltTy = VTy->getElementType();
799 unsigned NElts = VTy->getNumElements();
800 unsigned EltWidth = Info.Ctx.getTypeSize(EltTy);
Mike Stump11289f42009-09-09 15:08:12 +0000801
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000802 const Expr* SE = E->getSubExpr();
Nate Begeman2ffd3842009-06-26 18:22:18 +0000803 QualType SETy = SE->getType();
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000804
Eli Friedmanc757de22011-03-25 00:43:55 +0000805 switch (E->getCastKind()) {
806 case CK_VectorSplat: {
807 APValue Result = APValue();
808 if (SETy->isIntegerType()) {
809 APSInt IntResult;
810 if (!EvaluateInteger(SE, IntResult, Info))
811 return APValue();
812 Result = APValue(IntResult);
813 } else if (SETy->isRealFloatingType()) {
814 APFloat F(0.0);
815 if (!EvaluateFloat(SE, F, Info))
816 return APValue();
817 Result = APValue(F);
818 } else {
Anders Carlsson6fc22042011-03-25 11:22:47 +0000819 return APValue();
Eli Friedmanc757de22011-03-25 00:43:55 +0000820 }
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000821
822 // Splat and create vector APValue.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000823 SmallVector<APValue, 4> Elts(NElts, Result);
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000824 return APValue(&Elts[0], Elts.size());
Nate Begeman2ffd3842009-06-26 18:22:18 +0000825 }
Eli Friedmanc757de22011-03-25 00:43:55 +0000826 case CK_BitCast: {
827 if (SETy->isVectorType())
Peter Collingbournee9200682011-05-13 03:29:01 +0000828 return Visit(SE);
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000829
Eli Friedmanc757de22011-03-25 00:43:55 +0000830 if (!SETy->isIntegerType())
Anders Carlsson6fc22042011-03-25 11:22:47 +0000831 return APValue();
Mike Stump11289f42009-09-09 15:08:12 +0000832
Eli Friedmanc757de22011-03-25 00:43:55 +0000833 APSInt Init;
834 if (!EvaluateInteger(SE, Init, Info))
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000835 return APValue();
836
Eli Friedmanc757de22011-03-25 00:43:55 +0000837 assert((EltTy->isIntegerType() || EltTy->isRealFloatingType()) &&
838 "Vectors must be composed of ints or floats");
839
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000840 SmallVector<APValue, 4> Elts;
Eli Friedmanc757de22011-03-25 00:43:55 +0000841 for (unsigned i = 0; i != NElts; ++i) {
842 APSInt Tmp = Init.extOrTrunc(EltWidth);
843
844 if (EltTy->isIntegerType())
845 Elts.push_back(APValue(Tmp));
846 else
847 Elts.push_back(APValue(APFloat(Tmp)));
848
849 Init >>= EltWidth;
850 }
851 return APValue(&Elts[0], Elts.size());
Nate Begemanef1a7fa2009-07-01 07:50:47 +0000852 }
Eli Friedmanc757de22011-03-25 00:43:55 +0000853 case CK_LValueToRValue:
854 case CK_NoOp:
Peter Collingbournee9200682011-05-13 03:29:01 +0000855 return Visit(SE);
Eli Friedmanc757de22011-03-25 00:43:55 +0000856 default:
Anders Carlsson6fc22042011-03-25 11:22:47 +0000857 return APValue();
Eli Friedmanc757de22011-03-25 00:43:55 +0000858 }
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000859}
860
Mike Stump11289f42009-09-09 15:08:12 +0000861APValue
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000862VectorExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
Peter Collingbournee9200682011-05-13 03:29:01 +0000863 return this->Visit(E->getInitializer());
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000864}
865
Mike Stump11289f42009-09-09 15:08:12 +0000866APValue
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000867VectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
John McCall9dd450b2009-09-21 23:43:11 +0000868 const VectorType *VT = E->getType()->getAs<VectorType>();
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000869 unsigned NumInits = E->getNumInits();
Eli Friedman3ae59112009-02-23 04:23:56 +0000870 unsigned NumElements = VT->getNumElements();
Mike Stump11289f42009-09-09 15:08:12 +0000871
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000872 QualType EltTy = VT->getElementType();
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000873 SmallVector<APValue, 4> Elements;
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000874
John McCall875679e2010-06-11 17:54:15 +0000875 // If a vector is initialized with a single element, that value
876 // becomes every element of the vector, not just the first.
877 // This is the behavior described in the IBM AltiVec documentation.
878 if (NumInits == 1) {
Tanya Lattner5ac257d2011-04-15 22:42:59 +0000879
880 // Handle the case where the vector is initialized by a another
881 // vector (OpenCL 6.1.6).
882 if (E->getInit(0)->getType()->isVectorType())
883 return this->Visit(const_cast<Expr*>(E->getInit(0)));
884
John McCall875679e2010-06-11 17:54:15 +0000885 APValue InitValue;
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000886 if (EltTy->isIntegerType()) {
887 llvm::APSInt sInt(32);
John McCall875679e2010-06-11 17:54:15 +0000888 if (!EvaluateInteger(E->getInit(0), sInt, Info))
889 return APValue();
890 InitValue = APValue(sInt);
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000891 } else {
892 llvm::APFloat f(0.0);
John McCall875679e2010-06-11 17:54:15 +0000893 if (!EvaluateFloat(E->getInit(0), f, Info))
894 return APValue();
895 InitValue = APValue(f);
896 }
897 for (unsigned i = 0; i < NumElements; i++) {
898 Elements.push_back(InitValue);
899 }
900 } else {
901 for (unsigned i = 0; i < NumElements; i++) {
902 if (EltTy->isIntegerType()) {
903 llvm::APSInt sInt(32);
904 if (i < NumInits) {
905 if (!EvaluateInteger(E->getInit(i), sInt, Info))
906 return APValue();
907 } else {
908 sInt = Info.Ctx.MakeIntValue(0, EltTy);
909 }
910 Elements.push_back(APValue(sInt));
Eli Friedman3ae59112009-02-23 04:23:56 +0000911 } else {
John McCall875679e2010-06-11 17:54:15 +0000912 llvm::APFloat f(0.0);
913 if (i < NumInits) {
914 if (!EvaluateFloat(E->getInit(i), f, Info))
915 return APValue();
916 } else {
917 f = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy));
918 }
919 Elements.push_back(APValue(f));
Eli Friedman3ae59112009-02-23 04:23:56 +0000920 }
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000921 }
922 }
923 return APValue(&Elements[0], Elements.size());
924}
925
Mike Stump11289f42009-09-09 15:08:12 +0000926APValue
Eli Friedman3ae59112009-02-23 04:23:56 +0000927VectorExprEvaluator::GetZeroVector(QualType T) {
John McCall9dd450b2009-09-21 23:43:11 +0000928 const VectorType *VT = T->getAs<VectorType>();
Eli Friedman3ae59112009-02-23 04:23:56 +0000929 QualType EltTy = VT->getElementType();
930 APValue ZeroElement;
931 if (EltTy->isIntegerType())
932 ZeroElement = APValue(Info.Ctx.MakeIntValue(0, EltTy));
933 else
934 ZeroElement =
935 APValue(APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy)));
936
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000937 SmallVector<APValue, 4> Elements(VT->getNumElements(), ZeroElement);
Eli Friedman3ae59112009-02-23 04:23:56 +0000938 return APValue(&Elements[0], Elements.size());
939}
940
Eli Friedman3ae59112009-02-23 04:23:56 +0000941APValue VectorExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
942 if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
943 Info.EvalResult.HasSideEffects = true;
944 return GetZeroVector(E->getType());
945}
946
Nate Begeman2f2bdeb2009-01-18 03:20:47 +0000947//===----------------------------------------------------------------------===//
Chris Lattner05706e882008-07-11 18:11:29 +0000948// Integer Evaluation
949//===----------------------------------------------------------------------===//
Chris Lattner05706e882008-07-11 18:11:29 +0000950
951namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +0000952class IntExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +0000953 : public ExprEvaluatorBase<IntExprEvaluator, bool> {
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000954 APValue &Result;
Anders Carlsson0a1707c2008-07-08 05:13:58 +0000955public:
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000956 IntExprEvaluator(EvalInfo &info, APValue &result)
Peter Collingbournee9200682011-05-13 03:29:01 +0000957 : ExprEvaluatorBaseTy(info), Result(result) {}
Chris Lattner05706e882008-07-11 18:11:29 +0000958
Abramo Bagnara9ae292d2011-07-02 13:13:53 +0000959 bool Success(const llvm::APSInt &SI, const Expr *E) {
960 assert(E->getType()->isIntegralOrEnumerationType() &&
Douglas Gregorb90df602010-06-16 00:17:44 +0000961 "Invalid evaluation result.");
Abramo Bagnara9ae292d2011-07-02 13:13:53 +0000962 assert(SI.isSigned() == E->getType()->isSignedIntegerOrEnumerationType() &&
Daniel Dunbare3c92bc2009-02-19 18:37:50 +0000963 "Invalid evaluation result.");
Abramo Bagnara9ae292d2011-07-02 13:13:53 +0000964 assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
Daniel Dunbare3c92bc2009-02-19 18:37:50 +0000965 "Invalid evaluation result.");
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000966 Result = APValue(SI);
Daniel Dunbare3c92bc2009-02-19 18:37:50 +0000967 return true;
968 }
969
Daniel Dunbar8aafc892009-02-19 09:06:44 +0000970 bool Success(const llvm::APInt &I, const Expr *E) {
Douglas Gregorb90df602010-06-16 00:17:44 +0000971 assert(E->getType()->isIntegralOrEnumerationType() &&
972 "Invalid evaluation result.");
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000973 assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
Daniel Dunbare3c92bc2009-02-19 18:37:50 +0000974 "Invalid evaluation result.");
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000975 Result = APValue(APSInt(I));
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000976 Result.getInt().setIsUnsigned(
977 E->getType()->isUnsignedIntegerOrEnumerationType());
Daniel Dunbar8aafc892009-02-19 09:06:44 +0000978 return true;
979 }
980
981 bool Success(uint64_t Value, const Expr *E) {
Douglas Gregorb90df602010-06-16 00:17:44 +0000982 assert(E->getType()->isIntegralOrEnumerationType() &&
983 "Invalid evaluation result.");
Daniel Dunbarca097ad2009-02-19 20:17:33 +0000984 Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
Daniel Dunbar8aafc892009-02-19 09:06:44 +0000985 return true;
986 }
987
Ken Dyckdbc01912011-03-11 02:13:43 +0000988 bool Success(CharUnits Size, const Expr *E) {
989 return Success(Size.getQuantity(), E);
990 }
991
992
Anders Carlsson27b8c5c2008-11-30 18:14:57 +0000993 bool Error(SourceLocation L, diag::kind D, const Expr *E) {
Chris Lattnerfac05ae2008-11-12 07:43:42 +0000994 // Take the first error.
Anders Carlssonbd1df8e2008-11-30 16:38:33 +0000995 if (Info.EvalResult.Diag == 0) {
996 Info.EvalResult.DiagLoc = L;
997 Info.EvalResult.Diag = D;
Anders Carlsson27b8c5c2008-11-30 18:14:57 +0000998 Info.EvalResult.DiagExpr = E;
Chris Lattnerfac05ae2008-11-12 07:43:42 +0000999 }
Chris Lattner99415702008-07-12 00:14:42 +00001000 return false;
Chris Lattnerae8cc152008-07-11 19:24:49 +00001001 }
Mike Stump11289f42009-09-09 15:08:12 +00001002
Peter Collingbournee9200682011-05-13 03:29:01 +00001003 bool Success(const APValue &V, const Expr *E) {
1004 return Success(V.getInt(), E);
Chris Lattnerfac05ae2008-11-12 07:43:42 +00001005 }
Peter Collingbournee9200682011-05-13 03:29:01 +00001006 bool Error(const Expr *E) {
Anders Carlssonb33d6c82008-11-30 18:37:00 +00001007 return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
Anders Carlsson0a1707c2008-07-08 05:13:58 +00001008 }
Mike Stump11289f42009-09-09 15:08:12 +00001009
Peter Collingbournee9200682011-05-13 03:29:01 +00001010 //===--------------------------------------------------------------------===//
1011 // Visitor Methods
1012 //===--------------------------------------------------------------------===//
Anders Carlsson0a1707c2008-07-08 05:13:58 +00001013
Chris Lattner7174bf32008-07-12 00:38:25 +00001014 bool VisitIntegerLiteral(const IntegerLiteral *E) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001015 return Success(E->getValue(), E);
Chris Lattner7174bf32008-07-12 00:38:25 +00001016 }
1017 bool VisitCharacterLiteral(const CharacterLiteral *E) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001018 return Success(E->getValue(), E);
Chris Lattner7174bf32008-07-12 00:38:25 +00001019 }
Eli Friedmanfb8a93f2009-11-24 05:28:59 +00001020
1021 bool CheckReferencedDecl(const Expr *E, const Decl *D);
1022 bool VisitDeclRefExpr(const DeclRefExpr *E) {
Peter Collingbournee9200682011-05-13 03:29:01 +00001023 if (CheckReferencedDecl(E, E->getDecl()))
1024 return true;
1025
1026 return ExprEvaluatorBaseTy::VisitDeclRefExpr(E);
Eli Friedmanfb8a93f2009-11-24 05:28:59 +00001027 }
1028 bool VisitMemberExpr(const MemberExpr *E) {
1029 if (CheckReferencedDecl(E, E->getMemberDecl())) {
1030 // Conservatively assume a MemberExpr will have side-effects
1031 Info.EvalResult.HasSideEffects = true;
1032 return true;
1033 }
Peter Collingbournee9200682011-05-13 03:29:01 +00001034
1035 return ExprEvaluatorBaseTy::VisitMemberExpr(E);
Eli Friedmanfb8a93f2009-11-24 05:28:59 +00001036 }
1037
Peter Collingbournee9200682011-05-13 03:29:01 +00001038 bool VisitCallExpr(const CallExpr *E);
Chris Lattnere13042c2008-07-11 19:10:17 +00001039 bool VisitBinaryOperator(const BinaryOperator *E);
Douglas Gregor882211c2010-04-28 22:16:22 +00001040 bool VisitOffsetOfExpr(const OffsetOfExpr *E);
Chris Lattnere13042c2008-07-11 19:10:17 +00001041 bool VisitUnaryOperator(const UnaryOperator *E);
Anders Carlsson374b93d2008-07-08 05:49:43 +00001042
Peter Collingbournee9200682011-05-13 03:29:01 +00001043 bool VisitCastExpr(const CastExpr* E);
Peter Collingbournee190dee2011-03-11 19:24:49 +00001044 bool VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
Sebastian Redl6f282892008-11-11 17:56:53 +00001045
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001046 bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001047 return Success(E->getValue(), E);
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001048 }
Mike Stump11289f42009-09-09 15:08:12 +00001049
Anders Carlsson39def3a2008-12-21 22:39:40 +00001050 bool VisitGNUNullExpr(const GNUNullExpr *E) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001051 return Success(0, E);
Anders Carlsson39def3a2008-12-21 22:39:40 +00001052 }
Mike Stump11289f42009-09-09 15:08:12 +00001053
Douglas Gregor747eb782010-07-08 06:14:04 +00001054 bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001055 return Success(0, E);
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001056 }
1057
Eli Friedman4e7a2412009-02-27 04:45:43 +00001058 bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
1059 return Success(0, E);
1060 }
1061
Sebastian Redlbaad4e72009-01-05 20:52:13 +00001062 bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
Sebastian Redl8eb06f12010-09-13 20:56:31 +00001063 return Success(E->getValue(), E);
Sebastian Redlbaad4e72009-01-05 20:52:13 +00001064 }
1065
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00001066 bool VisitBinaryTypeTraitExpr(const BinaryTypeTraitExpr *E) {
1067 return Success(E->getValue(), E);
1068 }
1069
John Wiegley6242b6a2011-04-28 00:16:57 +00001070 bool VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
1071 return Success(E->getValue(), E);
1072 }
1073
John Wiegleyf9f65842011-04-25 06:54:41 +00001074 bool VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
1075 return Success(E->getValue(), E);
1076 }
1077
Eli Friedmana1c7b6c2009-02-28 03:59:05 +00001078 bool VisitUnaryReal(const UnaryOperator *E);
Eli Friedman4e7a2412009-02-27 04:45:43 +00001079 bool VisitUnaryImag(const UnaryOperator *E);
1080
Sebastian Redl5f0180d2010-09-10 20:55:47 +00001081 bool VisitCXXNoexceptExpr(const CXXNoexceptExpr *E);
Douglas Gregor820ba7b2011-01-04 17:33:58 +00001082 bool VisitSizeOfPackExpr(const SizeOfPackExpr *E);
1083
Chris Lattnerf8d7f722008-07-11 21:24:13 +00001084private:
Ken Dyck160146e2010-01-27 17:10:57 +00001085 CharUnits GetAlignOfExpr(const Expr *E);
1086 CharUnits GetAlignOfType(QualType T);
John McCall95007602010-05-10 23:27:23 +00001087 static QualType GetObjectType(const Expr *E);
Peter Collingbournee9200682011-05-13 03:29:01 +00001088 bool TryEvaluateBuiltinObjectSize(const CallExpr *E);
Eli Friedman4e7a2412009-02-27 04:45:43 +00001089 // FIXME: Missing: array subscript of vector, member of vector
Anders Carlsson9c181652008-07-08 14:35:21 +00001090};
Chris Lattner05706e882008-07-11 18:11:29 +00001091} // end anonymous namespace
Anders Carlsson4a3585b2008-07-08 15:34:11 +00001092
Daniel Dunbarce399542009-02-20 18:22:23 +00001093static bool EvaluateIntegerOrLValue(const Expr* E, APValue &Result, EvalInfo &Info) {
Douglas Gregorb90df602010-06-16 00:17:44 +00001094 assert(E->getType()->isIntegralOrEnumerationType());
Peter Collingbournee9200682011-05-13 03:29:01 +00001095 return IntExprEvaluator(Info, Result).Visit(E);
Daniel Dunbarce399542009-02-20 18:22:23 +00001096}
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001097
Daniel Dunbarce399542009-02-20 18:22:23 +00001098static bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
Douglas Gregorb90df602010-06-16 00:17:44 +00001099 assert(E->getType()->isIntegralOrEnumerationType());
John McCallf0c4f352010-05-07 05:46:35 +00001100
Daniel Dunbarce399542009-02-20 18:22:23 +00001101 APValue Val;
1102 if (!EvaluateIntegerOrLValue(E, Val, Info) || !Val.isInt())
1103 return false;
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001104 Result = Val.getInt();
1105 return true;
Anders Carlsson4a3585b2008-07-08 15:34:11 +00001106}
Anders Carlsson4a3585b2008-07-08 15:34:11 +00001107
Eli Friedmanfb8a93f2009-11-24 05:28:59 +00001108bool IntExprEvaluator::CheckReferencedDecl(const Expr* E, const Decl* D) {
Chris Lattner7174bf32008-07-12 00:38:25 +00001109 // Enums are integer constant exprs.
Abramo Bagnara2caedf42011-06-30 09:36:05 +00001110 if (const EnumConstantDecl *ECD = dyn_cast<EnumConstantDecl>(D)) {
Abramo Bagnara9ae292d2011-07-02 13:13:53 +00001111 // Check for signedness/width mismatches between E type and ECD value.
1112 bool SameSign = (ECD->getInitVal().isSigned()
1113 == E->getType()->isSignedIntegerOrEnumerationType());
1114 bool SameWidth = (ECD->getInitVal().getBitWidth()
1115 == Info.Ctx.getIntWidth(E->getType()));
1116 if (SameSign && SameWidth)
1117 return Success(ECD->getInitVal(), E);
1118 else {
1119 // Get rid of mismatch (otherwise Success assertions will fail)
1120 // by computing a new value matching the type of E.
1121 llvm::APSInt Val = ECD->getInitVal();
1122 if (!SameSign)
1123 Val.setIsSigned(!ECD->getInitVal().isSigned());
1124 if (!SameWidth)
1125 Val = Val.extOrTrunc(Info.Ctx.getIntWidth(E->getType()));
1126 return Success(Val, E);
1127 }
Abramo Bagnara2caedf42011-06-30 09:36:05 +00001128 }
Sebastian Redlc9ab3d42009-02-08 15:51:17 +00001129
1130 // In C++, const, non-volatile integers initialized with ICEs are ICEs.
Eli Friedman29f80c32009-03-30 23:39:01 +00001131 // In C, they can also be folded, although they are not ICEs.
Douglas Gregor0840cc02009-11-01 20:32:48 +00001132 if (Info.Ctx.getCanonicalType(E->getType()).getCVRQualifiers()
1133 == Qualifiers::Const) {
Anders Carlssonb0695ef2010-02-03 21:58:41 +00001134
1135 if (isa<ParmVarDecl>(D))
Peter Collingbournee9200682011-05-13 03:29:01 +00001136 return false;
Anders Carlssonb0695ef2010-02-03 21:58:41 +00001137
Eli Friedmanfb8a93f2009-11-24 05:28:59 +00001138 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Sebastian Redl5ca79842010-02-01 20:16:42 +00001139 if (const Expr *Init = VD->getAnyInitializer()) {
Eli Friedman1d6fb162009-12-03 20:31:57 +00001140 if (APValue *V = VD->getEvaluatedValue()) {
1141 if (V->isInt())
1142 return Success(V->getInt(), E);
Peter Collingbournee9200682011-05-13 03:29:01 +00001143 return false;
Eli Friedman1d6fb162009-12-03 20:31:57 +00001144 }
1145
1146 if (VD->isEvaluatingValue())
Peter Collingbournee9200682011-05-13 03:29:01 +00001147 return false;
Eli Friedman1d6fb162009-12-03 20:31:57 +00001148
1149 VD->setEvaluatingValue();
1150
Eli Friedman0b1fbd12010-09-06 00:10:32 +00001151 Expr::EvalResult EResult;
1152 if (Init->Evaluate(EResult, Info.Ctx) && !EResult.HasSideEffects &&
1153 EResult.Val.isInt()) {
Douglas Gregor31cf12c2009-05-26 18:54:04 +00001154 // Cache the evaluated value in the variable declaration.
Eli Friedman0b1fbd12010-09-06 00:10:32 +00001155 Result = EResult.Val;
Eli Friedman1d6fb162009-12-03 20:31:57 +00001156 VD->setEvaluatedValue(Result);
Douglas Gregor31cf12c2009-05-26 18:54:04 +00001157 return true;
1158 }
1159
Eli Friedman1d6fb162009-12-03 20:31:57 +00001160 VD->setEvaluatedValue(APValue());
Douglas Gregor31cf12c2009-05-26 18:54:04 +00001161 }
Sebastian Redlc9ab3d42009-02-08 15:51:17 +00001162 }
1163 }
1164
Chris Lattner7174bf32008-07-12 00:38:25 +00001165 // Otherwise, random variable references are not constants.
Peter Collingbournee9200682011-05-13 03:29:01 +00001166 return false;
Chris Lattner7174bf32008-07-12 00:38:25 +00001167}
1168
Chris Lattner86ee2862008-10-06 06:40:35 +00001169/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
1170/// as GCC.
1171static int EvaluateBuiltinClassifyType(const CallExpr *E) {
1172 // The following enum mimics the values returned by GCC.
Sebastian Redl0f8b23f2009-03-16 23:22:08 +00001173 // FIXME: Does GCC differ between lvalue and rvalue references here?
Chris Lattner86ee2862008-10-06 06:40:35 +00001174 enum gcc_type_class {
1175 no_type_class = -1,
1176 void_type_class, integer_type_class, char_type_class,
1177 enumeral_type_class, boolean_type_class,
1178 pointer_type_class, reference_type_class, offset_type_class,
1179 real_type_class, complex_type_class,
1180 function_type_class, method_type_class,
1181 record_type_class, union_type_class,
1182 array_type_class, string_type_class,
1183 lang_type_class
1184 };
Mike Stump11289f42009-09-09 15:08:12 +00001185
1186 // If no argument was supplied, default to "no_type_class". This isn't
Chris Lattner86ee2862008-10-06 06:40:35 +00001187 // ideal, however it is what gcc does.
1188 if (E->getNumArgs() == 0)
1189 return no_type_class;
Mike Stump11289f42009-09-09 15:08:12 +00001190
Chris Lattner86ee2862008-10-06 06:40:35 +00001191 QualType ArgTy = E->getArg(0)->getType();
1192 if (ArgTy->isVoidType())
1193 return void_type_class;
1194 else if (ArgTy->isEnumeralType())
1195 return enumeral_type_class;
1196 else if (ArgTy->isBooleanType())
1197 return boolean_type_class;
1198 else if (ArgTy->isCharType())
1199 return string_type_class; // gcc doesn't appear to use char_type_class
1200 else if (ArgTy->isIntegerType())
1201 return integer_type_class;
1202 else if (ArgTy->isPointerType())
1203 return pointer_type_class;
1204 else if (ArgTy->isReferenceType())
1205 return reference_type_class;
1206 else if (ArgTy->isRealType())
1207 return real_type_class;
1208 else if (ArgTy->isComplexType())
1209 return complex_type_class;
1210 else if (ArgTy->isFunctionType())
1211 return function_type_class;
Douglas Gregor8385a062010-04-26 21:31:17 +00001212 else if (ArgTy->isStructureOrClassType())
Chris Lattner86ee2862008-10-06 06:40:35 +00001213 return record_type_class;
1214 else if (ArgTy->isUnionType())
1215 return union_type_class;
1216 else if (ArgTy->isArrayType())
1217 return array_type_class;
1218 else if (ArgTy->isUnionType())
1219 return union_type_class;
1220 else // FIXME: offset_type_class, method_type_class, & lang_type_class?
1221 assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
1222 return -1;
1223}
1224
John McCall95007602010-05-10 23:27:23 +00001225/// Retrieves the "underlying object type" of the given expression,
1226/// as used by __builtin_object_size.
1227QualType IntExprEvaluator::GetObjectType(const Expr *E) {
1228 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
1229 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
1230 return VD->getType();
1231 } else if (isa<CompoundLiteralExpr>(E)) {
1232 return E->getType();
1233 }
1234
1235 return QualType();
1236}
1237
Peter Collingbournee9200682011-05-13 03:29:01 +00001238bool IntExprEvaluator::TryEvaluateBuiltinObjectSize(const CallExpr *E) {
John McCall95007602010-05-10 23:27:23 +00001239 // TODO: Perhaps we should let LLVM lower this?
1240 LValue Base;
1241 if (!EvaluatePointer(E->getArg(0), Base, Info))
1242 return false;
1243
1244 // If we can prove the base is null, lower to zero now.
1245 const Expr *LVBase = Base.getLValueBase();
1246 if (!LVBase) return Success(0, E);
1247
1248 QualType T = GetObjectType(LVBase);
1249 if (T.isNull() ||
1250 T->isIncompleteType() ||
Eli Friedmana170cd62010-08-05 02:49:48 +00001251 T->isFunctionType() ||
John McCall95007602010-05-10 23:27:23 +00001252 T->isVariablyModifiedType() ||
1253 T->isDependentType())
1254 return false;
1255
1256 CharUnits Size = Info.Ctx.getTypeSizeInChars(T);
1257 CharUnits Offset = Base.getLValueOffset();
1258
1259 if (!Offset.isNegative() && Offset <= Size)
1260 Size -= Offset;
1261 else
1262 Size = CharUnits::Zero();
Ken Dyckdbc01912011-03-11 02:13:43 +00001263 return Success(Size, E);
John McCall95007602010-05-10 23:27:23 +00001264}
1265
Peter Collingbournee9200682011-05-13 03:29:01 +00001266bool IntExprEvaluator::VisitCallExpr(const CallExpr *E) {
Douglas Gregore711f702009-02-14 18:57:46 +00001267 switch (E->isBuiltinCall(Info.Ctx)) {
Chris Lattner4deaa4e2008-10-06 05:28:25 +00001268 default:
Peter Collingbournee9200682011-05-13 03:29:01 +00001269 return ExprEvaluatorBaseTy::VisitCallExpr(E);
Mike Stump722cedf2009-10-26 18:35:08 +00001270
1271 case Builtin::BI__builtin_object_size: {
John McCall95007602010-05-10 23:27:23 +00001272 if (TryEvaluateBuiltinObjectSize(E))
1273 return true;
Mike Stump722cedf2009-10-26 18:35:08 +00001274
Eric Christopher99469702010-01-19 22:58:35 +00001275 // If evaluating the argument has side-effects we can't determine
1276 // the size of the object and lower it to unknown now.
Fariborz Jahanian4127b8e2009-11-05 18:03:03 +00001277 if (E->getArg(0)->HasSideEffects(Info.Ctx)) {
Benjamin Kramer0128f662010-01-03 18:18:37 +00001278 if (E->getArg(1)->EvaluateAsInt(Info.Ctx).getZExtValue() <= 1)
Chris Lattner4f105592009-11-03 19:48:51 +00001279 return Success(-1ULL, E);
Mike Stump722cedf2009-10-26 18:35:08 +00001280 return Success(0, E);
1281 }
Mike Stump876387b2009-10-27 22:09:17 +00001282
Mike Stump722cedf2009-10-26 18:35:08 +00001283 return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1284 }
1285
Chris Lattner4deaa4e2008-10-06 05:28:25 +00001286 case Builtin::BI__builtin_classify_type:
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001287 return Success(EvaluateBuiltinClassifyType(E), E);
Mike Stump11289f42009-09-09 15:08:12 +00001288
Anders Carlsson4c76e932008-11-24 04:21:33 +00001289 case Builtin::BI__builtin_constant_p:
Chris Lattner4deaa4e2008-10-06 05:28:25 +00001290 // __builtin_constant_p always has one operand: it returns true if that
1291 // operand can be folded, false otherwise.
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001292 return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
Chris Lattnerd545ad12009-09-23 06:06:36 +00001293
1294 case Builtin::BI__builtin_eh_return_data_regno: {
1295 int Operand = E->getArg(0)->EvaluateAsInt(Info.Ctx).getZExtValue();
Douglas Gregore8bbc122011-09-02 00:18:52 +00001296 Operand = Info.Ctx.getTargetInfo().getEHDataRegisterNumber(Operand);
Chris Lattnerd545ad12009-09-23 06:06:36 +00001297 return Success(Operand, E);
1298 }
Eli Friedmand5c93992010-02-13 00:10:10 +00001299
1300 case Builtin::BI__builtin_expect:
1301 return Visit(E->getArg(0));
Douglas Gregor6a6dac22010-09-10 06:27:15 +00001302
1303 case Builtin::BIstrlen:
1304 case Builtin::BI__builtin_strlen:
1305 // As an extension, we support strlen() and __builtin_strlen() as constant
1306 // expressions when the argument is a string literal.
Peter Collingbournee9200682011-05-13 03:29:01 +00001307 if (const StringLiteral *S
Douglas Gregor6a6dac22010-09-10 06:27:15 +00001308 = dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenImpCasts())) {
1309 // The string literal may have embedded null characters. Find the first
1310 // one and truncate there.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001311 StringRef Str = S->getString();
1312 StringRef::size_type Pos = Str.find(0);
1313 if (Pos != StringRef::npos)
Douglas Gregor6a6dac22010-09-10 06:27:15 +00001314 Str = Str.substr(0, Pos);
1315
1316 return Success(Str.size(), E);
1317 }
1318
1319 return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
Chris Lattner4deaa4e2008-10-06 05:28:25 +00001320 }
Chris Lattner7174bf32008-07-12 00:38:25 +00001321}
Anders Carlsson4a3585b2008-07-08 15:34:11 +00001322
Chris Lattnere13042c2008-07-11 19:10:17 +00001323bool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
John McCalle3027922010-08-25 11:45:40 +00001324 if (E->getOpcode() == BO_Comma) {
Anders Carlsson564730a2008-12-01 02:07:06 +00001325 if (!Visit(E->getRHS()))
1326 return false;
Anders Carlsson5b3638b2008-12-01 06:44:05 +00001327
Eli Friedman9cb9ff42009-02-26 10:19:36 +00001328 // If we can't evaluate the LHS, it might have side effects;
1329 // conservatively mark it.
1330 if (!E->getLHS()->isEvaluatable(Info.Ctx))
1331 Info.EvalResult.HasSideEffects = true;
Eli Friedman5a332ea2008-11-13 06:09:17 +00001332
Anders Carlsson564730a2008-12-01 02:07:06 +00001333 return true;
Eli Friedman5a332ea2008-11-13 06:09:17 +00001334 }
1335
1336 if (E->isLogicalOp()) {
1337 // These need to be handled specially because the operands aren't
1338 // necessarily integral
Anders Carlssonf50de0c2008-11-30 16:51:17 +00001339 bool lhsResult, rhsResult;
Mike Stump11289f42009-09-09 15:08:12 +00001340
Anders Carlssonf50de0c2008-11-30 16:51:17 +00001341 if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
Anders Carlsson59689ed2008-11-22 21:04:56 +00001342 // We were able to evaluate the LHS, see if we can get away with not
1343 // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
John McCalle3027922010-08-25 11:45:40 +00001344 if (lhsResult == (E->getOpcode() == BO_LOr))
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00001345 return Success(lhsResult, E);
Anders Carlsson4c76e932008-11-24 04:21:33 +00001346
Anders Carlssonf50de0c2008-11-30 16:51:17 +00001347 if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
John McCalle3027922010-08-25 11:45:40 +00001348 if (E->getOpcode() == BO_LOr)
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001349 return Success(lhsResult || rhsResult, E);
Anders Carlsson4c76e932008-11-24 04:21:33 +00001350 else
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001351 return Success(lhsResult && rhsResult, E);
Anders Carlsson4c76e932008-11-24 04:21:33 +00001352 }
1353 } else {
Anders Carlssonf50de0c2008-11-30 16:51:17 +00001354 if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
Anders Carlsson4c76e932008-11-24 04:21:33 +00001355 // We can't evaluate the LHS; however, sometimes the result
1356 // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
John McCalle3027922010-08-25 11:45:40 +00001357 if (rhsResult == (E->getOpcode() == BO_LOr) ||
1358 !rhsResult == (E->getOpcode() == BO_LAnd)) {
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001359 // Since we weren't able to evaluate the left hand side, it
Anders Carlssonf50de0c2008-11-30 16:51:17 +00001360 // must have had side effects.
1361 Info.EvalResult.HasSideEffects = true;
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001362
1363 return Success(rhsResult, E);
Anders Carlsson4c76e932008-11-24 04:21:33 +00001364 }
1365 }
Anders Carlsson59689ed2008-11-22 21:04:56 +00001366 }
Eli Friedman5a332ea2008-11-13 06:09:17 +00001367
Eli Friedman5a332ea2008-11-13 06:09:17 +00001368 return false;
1369 }
1370
Anders Carlssonacc79812008-11-16 07:17:21 +00001371 QualType LHSTy = E->getLHS()->getType();
1372 QualType RHSTy = E->getRHS()->getType();
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001373
1374 if (LHSTy->isAnyComplexType()) {
1375 assert(RHSTy->isAnyComplexType() && "Invalid comparison");
John McCall93d91dc2010-05-07 17:22:02 +00001376 ComplexValue LHS, RHS;
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001377
1378 if (!EvaluateComplex(E->getLHS(), LHS, Info))
1379 return false;
1380
1381 if (!EvaluateComplex(E->getRHS(), RHS, Info))
1382 return false;
1383
1384 if (LHS.isComplexFloat()) {
Mike Stump11289f42009-09-09 15:08:12 +00001385 APFloat::cmpResult CR_r =
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001386 LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
Mike Stump11289f42009-09-09 15:08:12 +00001387 APFloat::cmpResult CR_i =
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001388 LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
1389
John McCalle3027922010-08-25 11:45:40 +00001390 if (E->getOpcode() == BO_EQ)
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001391 return Success((CR_r == APFloat::cmpEqual &&
1392 CR_i == APFloat::cmpEqual), E);
1393 else {
John McCalle3027922010-08-25 11:45:40 +00001394 assert(E->getOpcode() == BO_NE &&
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001395 "Invalid complex comparison.");
Mike Stump11289f42009-09-09 15:08:12 +00001396 return Success(((CR_r == APFloat::cmpGreaterThan ||
Mon P Wang75c645c2010-04-29 05:53:29 +00001397 CR_r == APFloat::cmpLessThan ||
1398 CR_r == APFloat::cmpUnordered) ||
Mike Stump11289f42009-09-09 15:08:12 +00001399 (CR_i == APFloat::cmpGreaterThan ||
Mon P Wang75c645c2010-04-29 05:53:29 +00001400 CR_i == APFloat::cmpLessThan ||
1401 CR_i == APFloat::cmpUnordered)), E);
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001402 }
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001403 } else {
John McCalle3027922010-08-25 11:45:40 +00001404 if (E->getOpcode() == BO_EQ)
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001405 return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
1406 LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
1407 else {
John McCalle3027922010-08-25 11:45:40 +00001408 assert(E->getOpcode() == BO_NE &&
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001409 "Invalid compex comparison.");
1410 return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
1411 LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
1412 }
Daniel Dunbar74f2425b2009-01-29 06:43:41 +00001413 }
1414 }
Mike Stump11289f42009-09-09 15:08:12 +00001415
Anders Carlssonacc79812008-11-16 07:17:21 +00001416 if (LHSTy->isRealFloatingType() &&
1417 RHSTy->isRealFloatingType()) {
1418 APFloat RHS(0.0), LHS(0.0);
Mike Stump11289f42009-09-09 15:08:12 +00001419
Anders Carlssonacc79812008-11-16 07:17:21 +00001420 if (!EvaluateFloat(E->getRHS(), RHS, Info))
1421 return false;
Mike Stump11289f42009-09-09 15:08:12 +00001422
Anders Carlssonacc79812008-11-16 07:17:21 +00001423 if (!EvaluateFloat(E->getLHS(), LHS, Info))
1424 return false;
Mike Stump11289f42009-09-09 15:08:12 +00001425
Anders Carlssonacc79812008-11-16 07:17:21 +00001426 APFloat::cmpResult CR = LHS.compare(RHS);
Anders Carlsson899c7052008-11-16 22:46:56 +00001427
Anders Carlssonacc79812008-11-16 07:17:21 +00001428 switch (E->getOpcode()) {
1429 default:
1430 assert(0 && "Invalid binary operator!");
John McCalle3027922010-08-25 11:45:40 +00001431 case BO_LT:
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001432 return Success(CR == APFloat::cmpLessThan, E);
John McCalle3027922010-08-25 11:45:40 +00001433 case BO_GT:
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001434 return Success(CR == APFloat::cmpGreaterThan, E);
John McCalle3027922010-08-25 11:45:40 +00001435 case BO_LE:
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001436 return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
John McCalle3027922010-08-25 11:45:40 +00001437 case BO_GE:
Mike Stump11289f42009-09-09 15:08:12 +00001438 return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001439 E);
John McCalle3027922010-08-25 11:45:40 +00001440 case BO_EQ:
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001441 return Success(CR == APFloat::cmpEqual, E);
John McCalle3027922010-08-25 11:45:40 +00001442 case BO_NE:
Mike Stump11289f42009-09-09 15:08:12 +00001443 return Success(CR == APFloat::cmpGreaterThan
Mon P Wang75c645c2010-04-29 05:53:29 +00001444 || CR == APFloat::cmpLessThan
1445 || CR == APFloat::cmpUnordered, E);
Anders Carlssonacc79812008-11-16 07:17:21 +00001446 }
Anders Carlssonacc79812008-11-16 07:17:21 +00001447 }
Mike Stump11289f42009-09-09 15:08:12 +00001448
Eli Friedmana38da572009-04-28 19:17:36 +00001449 if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
John McCalle3027922010-08-25 11:45:40 +00001450 if (E->getOpcode() == BO_Sub || E->isEqualityOp()) {
John McCall45d55e42010-05-07 21:00:08 +00001451 LValue LHSValue;
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001452 if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
1453 return false;
Eli Friedman64004332009-03-23 04:38:34 +00001454
John McCall45d55e42010-05-07 21:00:08 +00001455 LValue RHSValue;
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001456 if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
1457 return false;
Eli Friedman64004332009-03-23 04:38:34 +00001458
Eli Friedman334046a2009-06-14 02:17:33 +00001459 // Reject any bases from the normal codepath; we special-case comparisons
1460 // to null.
1461 if (LHSValue.getLValueBase()) {
1462 if (!E->isEqualityOp())
1463 return false;
Ken Dyck02990832010-01-15 12:37:54 +00001464 if (RHSValue.getLValueBase() || !RHSValue.getLValueOffset().isZero())
Eli Friedman334046a2009-06-14 02:17:33 +00001465 return false;
1466 bool bres;
1467 if (!EvalPointerValueAsBool(LHSValue, bres))
1468 return false;
John McCalle3027922010-08-25 11:45:40 +00001469 return Success(bres ^ (E->getOpcode() == BO_EQ), E);
Eli Friedman334046a2009-06-14 02:17:33 +00001470 } else if (RHSValue.getLValueBase()) {
1471 if (!E->isEqualityOp())
1472 return false;
Ken Dyck02990832010-01-15 12:37:54 +00001473 if (LHSValue.getLValueBase() || !LHSValue.getLValueOffset().isZero())
Eli Friedman334046a2009-06-14 02:17:33 +00001474 return false;
1475 bool bres;
1476 if (!EvalPointerValueAsBool(RHSValue, bres))
1477 return false;
John McCalle3027922010-08-25 11:45:40 +00001478 return Success(bres ^ (E->getOpcode() == BO_EQ), E);
Eli Friedman334046a2009-06-14 02:17:33 +00001479 }
Eli Friedman64004332009-03-23 04:38:34 +00001480
John McCalle3027922010-08-25 11:45:40 +00001481 if (E->getOpcode() == BO_Sub) {
Chris Lattner882bdf22010-04-20 17:13:14 +00001482 QualType Type = E->getLHS()->getType();
1483 QualType ElementType = Type->getAs<PointerType>()->getPointeeType();
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001484
Ken Dyck02990832010-01-15 12:37:54 +00001485 CharUnits ElementSize = CharUnits::One();
Eli Friedmanfa90b152009-06-04 20:23:20 +00001486 if (!ElementType->isVoidType() && !ElementType->isFunctionType())
Ken Dyck02990832010-01-15 12:37:54 +00001487 ElementSize = Info.Ctx.getTypeSizeInChars(ElementType);
Eli Friedman64004332009-03-23 04:38:34 +00001488
Ken Dyck02990832010-01-15 12:37:54 +00001489 CharUnits Diff = LHSValue.getLValueOffset() -
1490 RHSValue.getLValueOffset();
1491 return Success(Diff / ElementSize, E);
Eli Friedmana38da572009-04-28 19:17:36 +00001492 }
1493 bool Result;
John McCalle3027922010-08-25 11:45:40 +00001494 if (E->getOpcode() == BO_EQ) {
Eli Friedmana38da572009-04-28 19:17:36 +00001495 Result = LHSValue.getLValueOffset() == RHSValue.getLValueOffset();
Eli Friedman8b171f62009-04-29 20:29:43 +00001496 } else {
Eli Friedmana38da572009-04-28 19:17:36 +00001497 Result = LHSValue.getLValueOffset() != RHSValue.getLValueOffset();
1498 }
1499 return Success(Result, E);
Anders Carlsson9f9e4242008-11-16 19:01:22 +00001500 }
1501 }
Douglas Gregorb90df602010-06-16 00:17:44 +00001502 if (!LHSTy->isIntegralOrEnumerationType() ||
1503 !RHSTy->isIntegralOrEnumerationType()) {
Eli Friedman5a332ea2008-11-13 06:09:17 +00001504 // We can't continue from here for non-integral types, and they
1505 // could potentially confuse the following operations.
Eli Friedman5a332ea2008-11-13 06:09:17 +00001506 return false;
1507 }
1508
Anders Carlsson9c181652008-07-08 14:35:21 +00001509 // The LHS of a constant expr is always evaluated and needed.
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001510 if (!Visit(E->getLHS()))
Chris Lattner99415702008-07-12 00:14:42 +00001511 return false; // error in subexpression.
Eli Friedmanbd840592008-07-27 05:46:18 +00001512
Eli Friedman94c25c62009-03-24 01:14:50 +00001513 APValue RHSVal;
1514 if (!EvaluateIntegerOrLValue(E->getRHS(), RHSVal, Info))
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001515 return false;
Eli Friedman94c25c62009-03-24 01:14:50 +00001516
1517 // Handle cases like (unsigned long)&a + 4.
1518 if (E->isAdditiveOp() && Result.isLValue() && RHSVal.isInt()) {
Ken Dyck02990832010-01-15 12:37:54 +00001519 CharUnits Offset = Result.getLValueOffset();
1520 CharUnits AdditionalOffset = CharUnits::fromQuantity(
1521 RHSVal.getInt().getZExtValue());
John McCalle3027922010-08-25 11:45:40 +00001522 if (E->getOpcode() == BO_Add)
Ken Dyck02990832010-01-15 12:37:54 +00001523 Offset += AdditionalOffset;
Eli Friedman94c25c62009-03-24 01:14:50 +00001524 else
Ken Dyck02990832010-01-15 12:37:54 +00001525 Offset -= AdditionalOffset;
1526 Result = APValue(Result.getLValueBase(), Offset);
Eli Friedman94c25c62009-03-24 01:14:50 +00001527 return true;
1528 }
1529
1530 // Handle cases like 4 + (unsigned long)&a
John McCalle3027922010-08-25 11:45:40 +00001531 if (E->getOpcode() == BO_Add &&
Eli Friedman94c25c62009-03-24 01:14:50 +00001532 RHSVal.isLValue() && Result.isInt()) {
Ken Dyck02990832010-01-15 12:37:54 +00001533 CharUnits Offset = RHSVal.getLValueOffset();
1534 Offset += CharUnits::fromQuantity(Result.getInt().getZExtValue());
1535 Result = APValue(RHSVal.getLValueBase(), Offset);
Eli Friedman94c25c62009-03-24 01:14:50 +00001536 return true;
1537 }
1538
1539 // All the following cases expect both operands to be an integer
1540 if (!Result.isInt() || !RHSVal.isInt())
Chris Lattnere13042c2008-07-11 19:10:17 +00001541 return false;
Eli Friedman5a332ea2008-11-13 06:09:17 +00001542
Eli Friedman94c25c62009-03-24 01:14:50 +00001543 APSInt& RHS = RHSVal.getInt();
1544
Anders Carlsson9c181652008-07-08 14:35:21 +00001545 switch (E->getOpcode()) {
Chris Lattnerfac05ae2008-11-12 07:43:42 +00001546 default:
Anders Carlssonb33d6c82008-11-30 18:37:00 +00001547 return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
John McCalle3027922010-08-25 11:45:40 +00001548 case BO_Mul: return Success(Result.getInt() * RHS, E);
1549 case BO_Add: return Success(Result.getInt() + RHS, E);
1550 case BO_Sub: return Success(Result.getInt() - RHS, E);
1551 case BO_And: return Success(Result.getInt() & RHS, E);
1552 case BO_Xor: return Success(Result.getInt() ^ RHS, E);
1553 case BO_Or: return Success(Result.getInt() | RHS, E);
1554 case BO_Div:
Chris Lattner99415702008-07-12 00:14:42 +00001555 if (RHS == 0)
Anders Carlssonb33d6c82008-11-30 18:37:00 +00001556 return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001557 return Success(Result.getInt() / RHS, E);
John McCalle3027922010-08-25 11:45:40 +00001558 case BO_Rem:
Chris Lattner99415702008-07-12 00:14:42 +00001559 if (RHS == 0)
Anders Carlssonb33d6c82008-11-30 18:37:00 +00001560 return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001561 return Success(Result.getInt() % RHS, E);
John McCalle3027922010-08-25 11:45:40 +00001562 case BO_Shl: {
John McCall18a2c2c2010-11-09 22:22:12 +00001563 // During constant-folding, a negative shift is an opposite shift.
1564 if (RHS.isSigned() && RHS.isNegative()) {
1565 RHS = -RHS;
1566 goto shift_right;
1567 }
1568
1569 shift_left:
1570 unsigned SA
1571 = (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001572 return Success(Result.getInt() << SA, E);
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00001573 }
John McCalle3027922010-08-25 11:45:40 +00001574 case BO_Shr: {
John McCall18a2c2c2010-11-09 22:22:12 +00001575 // During constant-folding, a negative shift is an opposite shift.
1576 if (RHS.isSigned() && RHS.isNegative()) {
1577 RHS = -RHS;
1578 goto shift_left;
1579 }
1580
1581 shift_right:
Mike Stump11289f42009-09-09 15:08:12 +00001582 unsigned SA =
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001583 (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
1584 return Success(Result.getInt() >> SA, E);
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00001585 }
Mike Stump11289f42009-09-09 15:08:12 +00001586
John McCalle3027922010-08-25 11:45:40 +00001587 case BO_LT: return Success(Result.getInt() < RHS, E);
1588 case BO_GT: return Success(Result.getInt() > RHS, E);
1589 case BO_LE: return Success(Result.getInt() <= RHS, E);
1590 case BO_GE: return Success(Result.getInt() >= RHS, E);
1591 case BO_EQ: return Success(Result.getInt() == RHS, E);
1592 case BO_NE: return Success(Result.getInt() != RHS, E);
Eli Friedman8553a982008-11-13 02:13:11 +00001593 }
Anders Carlsson9c181652008-07-08 14:35:21 +00001594}
1595
Ken Dyck160146e2010-01-27 17:10:57 +00001596CharUnits IntExprEvaluator::GetAlignOfType(QualType T) {
Sebastian Redl22e2e5c2009-11-23 17:18:46 +00001597 // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
1598 // the result is the size of the referenced type."
1599 // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
1600 // result shall be the alignment of the referenced type."
1601 if (const ReferenceType *Ref = T->getAs<ReferenceType>())
1602 T = Ref->getPointeeType();
Chad Rosier99ee7822011-07-26 07:03:04 +00001603
1604 // __alignof is defined to return the preferred alignment.
1605 return Info.Ctx.toCharUnitsFromBits(
1606 Info.Ctx.getPreferredTypeAlign(T.getTypePtr()));
Chris Lattner24aeeab2009-01-24 21:09:06 +00001607}
1608
Ken Dyck160146e2010-01-27 17:10:57 +00001609CharUnits IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
Chris Lattner68061312009-01-24 21:53:27 +00001610 E = E->IgnoreParens();
1611
1612 // alignof decl is always accepted, even if it doesn't make sense: we default
Mike Stump11289f42009-09-09 15:08:12 +00001613 // to 1 in those cases.
Chris Lattner68061312009-01-24 21:53:27 +00001614 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
Ken Dyck160146e2010-01-27 17:10:57 +00001615 return Info.Ctx.getDeclAlign(DRE->getDecl(),
1616 /*RefAsPointee*/true);
Eli Friedman64004332009-03-23 04:38:34 +00001617
Chris Lattner68061312009-01-24 21:53:27 +00001618 if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
Ken Dyck160146e2010-01-27 17:10:57 +00001619 return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
1620 /*RefAsPointee*/true);
Chris Lattner68061312009-01-24 21:53:27 +00001621
Chris Lattner24aeeab2009-01-24 21:09:06 +00001622 return GetAlignOfType(E->getType());
1623}
1624
1625
Peter Collingbournee190dee2011-03-11 19:24:49 +00001626/// VisitUnaryExprOrTypeTraitExpr - Evaluate a sizeof, alignof or vec_step with
1627/// a result as the expression's type.
1628bool IntExprEvaluator::VisitUnaryExprOrTypeTraitExpr(
1629 const UnaryExprOrTypeTraitExpr *E) {
1630 switch(E->getKind()) {
1631 case UETT_AlignOf: {
Chris Lattner24aeeab2009-01-24 21:09:06 +00001632 if (E->isArgumentType())
Ken Dyckdbc01912011-03-11 02:13:43 +00001633 return Success(GetAlignOfType(E->getArgumentType()), E);
Chris Lattner24aeeab2009-01-24 21:09:06 +00001634 else
Ken Dyckdbc01912011-03-11 02:13:43 +00001635 return Success(GetAlignOfExpr(E->getArgumentExpr()), E);
Chris Lattner24aeeab2009-01-24 21:09:06 +00001636 }
Eli Friedman64004332009-03-23 04:38:34 +00001637
Peter Collingbournee190dee2011-03-11 19:24:49 +00001638 case UETT_VecStep: {
1639 QualType Ty = E->getTypeOfArgument();
Sebastian Redl6f282892008-11-11 17:56:53 +00001640
Peter Collingbournee190dee2011-03-11 19:24:49 +00001641 if (Ty->isVectorType()) {
1642 unsigned n = Ty->getAs<VectorType>()->getNumElements();
Eli Friedman64004332009-03-23 04:38:34 +00001643
Peter Collingbournee190dee2011-03-11 19:24:49 +00001644 // The vec_step built-in functions that take a 3-component
1645 // vector return 4. (OpenCL 1.1 spec 6.11.12)
1646 if (n == 3)
1647 n = 4;
Eli Friedman2aa38fe2009-01-24 22:19:05 +00001648
Peter Collingbournee190dee2011-03-11 19:24:49 +00001649 return Success(n, E);
1650 } else
1651 return Success(1, E);
1652 }
1653
1654 case UETT_SizeOf: {
1655 QualType SrcTy = E->getTypeOfArgument();
1656 // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
1657 // the result is the size of the referenced type."
1658 // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
1659 // result shall be the alignment of the referenced type."
1660 if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
1661 SrcTy = Ref->getPointeeType();
1662
1663 // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
1664 // extension.
1665 if (SrcTy->isVoidType() || SrcTy->isFunctionType())
1666 return Success(1, E);
1667
1668 // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
1669 if (!SrcTy->isConstantSizeType())
1670 return false;
1671
1672 // Get information about the size.
1673 return Success(Info.Ctx.getTypeSizeInChars(SrcTy), E);
1674 }
1675 }
1676
1677 llvm_unreachable("unknown expr/type trait");
1678 return false;
Chris Lattnerf8d7f722008-07-11 21:24:13 +00001679}
1680
Peter Collingbournee9200682011-05-13 03:29:01 +00001681bool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *OOE) {
Douglas Gregor882211c2010-04-28 22:16:22 +00001682 CharUnits Result;
Peter Collingbournee9200682011-05-13 03:29:01 +00001683 unsigned n = OOE->getNumComponents();
Douglas Gregor882211c2010-04-28 22:16:22 +00001684 if (n == 0)
1685 return false;
Peter Collingbournee9200682011-05-13 03:29:01 +00001686 QualType CurrentType = OOE->getTypeSourceInfo()->getType();
Douglas Gregor882211c2010-04-28 22:16:22 +00001687 for (unsigned i = 0; i != n; ++i) {
1688 OffsetOfExpr::OffsetOfNode ON = OOE->getComponent(i);
1689 switch (ON.getKind()) {
1690 case OffsetOfExpr::OffsetOfNode::Array: {
Peter Collingbournee9200682011-05-13 03:29:01 +00001691 const Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
Douglas Gregor882211c2010-04-28 22:16:22 +00001692 APSInt IdxResult;
1693 if (!EvaluateInteger(Idx, IdxResult, Info))
1694 return false;
1695 const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
1696 if (!AT)
1697 return false;
1698 CurrentType = AT->getElementType();
1699 CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
1700 Result += IdxResult.getSExtValue() * ElementSize;
1701 break;
1702 }
1703
1704 case OffsetOfExpr::OffsetOfNode::Field: {
1705 FieldDecl *MemberDecl = ON.getField();
1706 const RecordType *RT = CurrentType->getAs<RecordType>();
1707 if (!RT)
1708 return false;
1709 RecordDecl *RD = RT->getDecl();
1710 const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
John McCall4e819612011-01-20 07:57:12 +00001711 unsigned i = MemberDecl->getFieldIndex();
Douglas Gregord1702062010-04-29 00:18:15 +00001712 assert(i < RL.getFieldCount() && "offsetof field in wrong type");
Ken Dyck86a7fcc2011-01-18 01:56:16 +00001713 Result += Info.Ctx.toCharUnitsFromBits(RL.getFieldOffset(i));
Douglas Gregor882211c2010-04-28 22:16:22 +00001714 CurrentType = MemberDecl->getType().getNonReferenceType();
1715 break;
1716 }
1717
1718 case OffsetOfExpr::OffsetOfNode::Identifier:
1719 llvm_unreachable("dependent __builtin_offsetof");
Douglas Gregord1702062010-04-29 00:18:15 +00001720 return false;
1721
1722 case OffsetOfExpr::OffsetOfNode::Base: {
1723 CXXBaseSpecifier *BaseSpec = ON.getBase();
1724 if (BaseSpec->isVirtual())
1725 return false;
1726
1727 // Find the layout of the class whose base we are looking into.
1728 const RecordType *RT = CurrentType->getAs<RecordType>();
1729 if (!RT)
1730 return false;
1731 RecordDecl *RD = RT->getDecl();
1732 const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
1733
1734 // Find the base class itself.
1735 CurrentType = BaseSpec->getType();
1736 const RecordType *BaseRT = CurrentType->getAs<RecordType>();
1737 if (!BaseRT)
1738 return false;
1739
1740 // Add the offset to the base.
Ken Dyck02155cb2011-01-26 02:17:08 +00001741 Result += RL.getBaseClassOffset(cast<CXXRecordDecl>(BaseRT->getDecl()));
Douglas Gregord1702062010-04-29 00:18:15 +00001742 break;
1743 }
Douglas Gregor882211c2010-04-28 22:16:22 +00001744 }
1745 }
Peter Collingbournee9200682011-05-13 03:29:01 +00001746 return Success(Result, OOE);
Douglas Gregor882211c2010-04-28 22:16:22 +00001747}
1748
Chris Lattnere13042c2008-07-11 19:10:17 +00001749bool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
John McCalle3027922010-08-25 11:45:40 +00001750 if (E->getOpcode() == UO_LNot) {
Eli Friedman5a332ea2008-11-13 06:09:17 +00001751 // LNot's operand isn't necessarily an integer, so we handle it specially.
1752 bool bres;
1753 if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1754 return false;
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001755 return Success(!bres, E);
Eli Friedman5a332ea2008-11-13 06:09:17 +00001756 }
1757
Daniel Dunbar79e042a2009-02-21 18:14:20 +00001758 // Only handle integral operations...
Douglas Gregorb90df602010-06-16 00:17:44 +00001759 if (!E->getSubExpr()->getType()->isIntegralOrEnumerationType())
Daniel Dunbar79e042a2009-02-21 18:14:20 +00001760 return false;
1761
Chris Lattnercdf34e72008-07-11 22:52:41 +00001762 // Get the operand value into 'Result'.
1763 if (!Visit(E->getSubExpr()))
Chris Lattnerf09ad162008-07-11 22:15:16 +00001764 return false;
Anders Carlsson9c181652008-07-08 14:35:21 +00001765
Chris Lattnerf09ad162008-07-11 22:15:16 +00001766 switch (E->getOpcode()) {
Chris Lattner7174bf32008-07-12 00:38:25 +00001767 default:
Chris Lattnerf09ad162008-07-11 22:15:16 +00001768 // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
1769 // See C99 6.6p3.
Anders Carlssonb33d6c82008-11-30 18:37:00 +00001770 return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
John McCalle3027922010-08-25 11:45:40 +00001771 case UO_Extension:
Chris Lattner7174bf32008-07-12 00:38:25 +00001772 // FIXME: Should extension allow i-c-e extension expressions in its scope?
1773 // If so, we could clear the diagnostic ID.
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00001774 return true;
John McCalle3027922010-08-25 11:45:40 +00001775 case UO_Plus:
Mike Stump11289f42009-09-09 15:08:12 +00001776 // The result is always just the subexpr.
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00001777 return true;
John McCalle3027922010-08-25 11:45:40 +00001778 case UO_Minus:
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001779 if (!Result.isInt()) return false;
1780 return Success(-Result.getInt(), E);
John McCalle3027922010-08-25 11:45:40 +00001781 case UO_Not:
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001782 if (!Result.isInt()) return false;
1783 return Success(~Result.getInt(), E);
Anders Carlsson9c181652008-07-08 14:35:21 +00001784 }
Anders Carlsson9c181652008-07-08 14:35:21 +00001785}
Mike Stump11289f42009-09-09 15:08:12 +00001786
Chris Lattner477c4be2008-07-12 01:15:53 +00001787/// HandleCast - This is used to evaluate implicit or explicit casts where the
1788/// result type is integer.
Peter Collingbournee9200682011-05-13 03:29:01 +00001789bool IntExprEvaluator::VisitCastExpr(const CastExpr *E) {
1790 const Expr *SubExpr = E->getSubExpr();
Anders Carlsson27b8c5c2008-11-30 18:14:57 +00001791 QualType DestType = E->getType();
Daniel Dunbarcf04aa12009-02-19 22:16:29 +00001792 QualType SrcType = SubExpr->getType();
Anders Carlsson27b8c5c2008-11-30 18:14:57 +00001793
Eli Friedmanc757de22011-03-25 00:43:55 +00001794 switch (E->getCastKind()) {
Eli Friedmanc757de22011-03-25 00:43:55 +00001795 case CK_BaseToDerived:
1796 case CK_DerivedToBase:
1797 case CK_UncheckedDerivedToBase:
1798 case CK_Dynamic:
1799 case CK_ToUnion:
1800 case CK_ArrayToPointerDecay:
1801 case CK_FunctionToPointerDecay:
1802 case CK_NullToPointer:
1803 case CK_NullToMemberPointer:
1804 case CK_BaseToDerivedMemberPointer:
1805 case CK_DerivedToBaseMemberPointer:
1806 case CK_ConstructorConversion:
1807 case CK_IntegralToPointer:
1808 case CK_ToVoid:
1809 case CK_VectorSplat:
1810 case CK_IntegralToFloating:
1811 case CK_FloatingCast:
John McCall9320b872011-09-09 05:25:32 +00001812 case CK_CPointerToObjCPointerCast:
1813 case CK_BlockPointerToObjCPointerCast:
Eli Friedmanc757de22011-03-25 00:43:55 +00001814 case CK_AnyPointerToBlockPointerCast:
1815 case CK_ObjCObjectLValueCast:
1816 case CK_FloatingRealToComplex:
1817 case CK_FloatingComplexToReal:
1818 case CK_FloatingComplexCast:
1819 case CK_FloatingComplexToIntegralComplex:
1820 case CK_IntegralRealToComplex:
1821 case CK_IntegralComplexCast:
1822 case CK_IntegralComplexToFloatingComplex:
1823 llvm_unreachable("invalid cast kind for integral value");
1824
Eli Friedman9faf2f92011-03-25 19:07:11 +00001825 case CK_BitCast:
Eli Friedmanc757de22011-03-25 00:43:55 +00001826 case CK_Dependent:
1827 case CK_GetObjCProperty:
1828 case CK_LValueBitCast:
1829 case CK_UserDefinedConversion:
John McCall2d637d22011-09-10 06:18:15 +00001830 case CK_ARCProduceObject:
1831 case CK_ARCConsumeObject:
1832 case CK_ARCReclaimReturnedObject:
1833 case CK_ARCExtendBlockObject:
Eli Friedmanc757de22011-03-25 00:43:55 +00001834 return false;
1835
1836 case CK_LValueToRValue:
1837 case CK_NoOp:
1838 return Visit(E->getSubExpr());
1839
1840 case CK_MemberPointerToBoolean:
1841 case CK_PointerToBoolean:
1842 case CK_IntegralToBoolean:
1843 case CK_FloatingToBoolean:
1844 case CK_FloatingComplexToBoolean:
1845 case CK_IntegralComplexToBoolean: {
Eli Friedman9a156e52008-11-12 09:44:48 +00001846 bool BoolResult;
1847 if (!HandleConversionToBool(SubExpr, BoolResult, Info))
1848 return false;
Daniel Dunbar8aafc892009-02-19 09:06:44 +00001849 return Success(BoolResult, E);
Eli Friedman9a156e52008-11-12 09:44:48 +00001850 }
1851
Eli Friedmanc757de22011-03-25 00:43:55 +00001852 case CK_IntegralCast: {
Chris Lattner477c4be2008-07-12 01:15:53 +00001853 if (!Visit(SubExpr))
Chris Lattnere13042c2008-07-11 19:10:17 +00001854 return false;
Daniel Dunbarb6f953e2009-01-29 06:16:07 +00001855
Eli Friedman742421e2009-02-20 01:15:07 +00001856 if (!Result.isInt()) {
1857 // Only allow casts of lvalues if they are lossless.
1858 return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
1859 }
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001860
Daniel Dunbar1c8560d2009-02-19 22:24:01 +00001861 return Success(HandleIntToIntCast(DestType, SrcType,
Daniel Dunbarca097ad2009-02-19 20:17:33 +00001862 Result.getInt(), Info.Ctx), E);
Chris Lattner477c4be2008-07-12 01:15:53 +00001863 }
Mike Stump11289f42009-09-09 15:08:12 +00001864
Eli Friedmanc757de22011-03-25 00:43:55 +00001865 case CK_PointerToIntegral: {
John McCall45d55e42010-05-07 21:00:08 +00001866 LValue LV;
Chris Lattnercdf34e72008-07-11 22:52:41 +00001867 if (!EvaluatePointer(SubExpr, LV, Info))
Chris Lattnere13042c2008-07-11 19:10:17 +00001868 return false;
Eli Friedman9a156e52008-11-12 09:44:48 +00001869
Daniel Dunbar1c8560d2009-02-19 22:24:01 +00001870 if (LV.getLValueBase()) {
1871 // Only allow based lvalue casts if they are lossless.
1872 if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
1873 return false;
Eli Friedman9a156e52008-11-12 09:44:48 +00001874
John McCall45d55e42010-05-07 21:00:08 +00001875 LV.moveInto(Result);
Daniel Dunbar1c8560d2009-02-19 22:24:01 +00001876 return true;
1877 }
1878
Ken Dyck02990832010-01-15 12:37:54 +00001879 APSInt AsInt = Info.Ctx.MakeIntValue(LV.getLValueOffset().getQuantity(),
1880 SrcType);
Daniel Dunbar1c8560d2009-02-19 22:24:01 +00001881 return Success(HandleIntToIntCast(DestType, SrcType, AsInt, Info.Ctx), E);
Anders Carlssonb5ad0212008-07-08 14:30:00 +00001882 }
Eli Friedman9a156e52008-11-12 09:44:48 +00001883
Eli Friedmanc757de22011-03-25 00:43:55 +00001884 case CK_IntegralComplexToReal: {
John McCall93d91dc2010-05-07 17:22:02 +00001885 ComplexValue C;
Eli Friedmand3a5a9d2009-04-22 19:23:09 +00001886 if (!EvaluateComplex(SubExpr, C, Info))
1887 return false;
Eli Friedmanc757de22011-03-25 00:43:55 +00001888 return Success(C.getComplexIntReal(), E);
Eli Friedmand3a5a9d2009-04-22 19:23:09 +00001889 }
Eli Friedmanc2b50172009-02-22 11:46:18 +00001890
Eli Friedmanc757de22011-03-25 00:43:55 +00001891 case CK_FloatingToIntegral: {
1892 APFloat F(0.0);
1893 if (!EvaluateFloat(SubExpr, F, Info))
1894 return false;
Chris Lattner477c4be2008-07-12 01:15:53 +00001895
Eli Friedmanc757de22011-03-25 00:43:55 +00001896 return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
1897 }
1898 }
Mike Stump11289f42009-09-09 15:08:12 +00001899
Eli Friedmanc757de22011-03-25 00:43:55 +00001900 llvm_unreachable("unknown cast resulting in integral value");
1901 return false;
Anders Carlsson9c181652008-07-08 14:35:21 +00001902}
Anders Carlssonb5ad0212008-07-08 14:30:00 +00001903
Eli Friedmana1c7b6c2009-02-28 03:59:05 +00001904bool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1905 if (E->getSubExpr()->getType()->isAnyComplexType()) {
John McCall93d91dc2010-05-07 17:22:02 +00001906 ComplexValue LV;
Eli Friedmana1c7b6c2009-02-28 03:59:05 +00001907 if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1908 return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1909 return Success(LV.getComplexIntReal(), E);
1910 }
1911
1912 return Visit(E->getSubExpr());
1913}
1914
Eli Friedman4e7a2412009-02-27 04:45:43 +00001915bool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
Eli Friedmana1c7b6c2009-02-28 03:59:05 +00001916 if (E->getSubExpr()->getType()->isComplexIntegerType()) {
John McCall93d91dc2010-05-07 17:22:02 +00001917 ComplexValue LV;
Eli Friedmana1c7b6c2009-02-28 03:59:05 +00001918 if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1919 return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1920 return Success(LV.getComplexIntImag(), E);
1921 }
1922
Eli Friedman4e7a2412009-02-27 04:45:43 +00001923 if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
1924 Info.EvalResult.HasSideEffects = true;
1925 return Success(0, E);
1926}
1927
Douglas Gregor820ba7b2011-01-04 17:33:58 +00001928bool IntExprEvaluator::VisitSizeOfPackExpr(const SizeOfPackExpr *E) {
1929 return Success(E->getPackLength(), E);
1930}
1931
Sebastian Redl5f0180d2010-09-10 20:55:47 +00001932bool IntExprEvaluator::VisitCXXNoexceptExpr(const CXXNoexceptExpr *E) {
1933 return Success(E->getValue(), E);
1934}
1935
Chris Lattner05706e882008-07-11 18:11:29 +00001936//===----------------------------------------------------------------------===//
Eli Friedman24c01542008-08-22 00:06:13 +00001937// Float Evaluation
1938//===----------------------------------------------------------------------===//
1939
1940namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +00001941class FloatExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +00001942 : public ExprEvaluatorBase<FloatExprEvaluator, bool> {
Eli Friedman24c01542008-08-22 00:06:13 +00001943 APFloat &Result;
1944public:
1945 FloatExprEvaluator(EvalInfo &info, APFloat &result)
Peter Collingbournee9200682011-05-13 03:29:01 +00001946 : ExprEvaluatorBaseTy(info), Result(result) {}
Eli Friedman24c01542008-08-22 00:06:13 +00001947
Peter Collingbournee9200682011-05-13 03:29:01 +00001948 bool Success(const APValue &V, const Expr *e) {
1949 Result = V.getFloat();
1950 return true;
1951 }
1952 bool Error(const Stmt *S) {
Eli Friedman24c01542008-08-22 00:06:13 +00001953 return false;
1954 }
1955
Chris Lattner4deaa4e2008-10-06 05:28:25 +00001956 bool VisitCallExpr(const CallExpr *E);
Eli Friedman24c01542008-08-22 00:06:13 +00001957
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00001958 bool VisitUnaryOperator(const UnaryOperator *E);
Eli Friedman24c01542008-08-22 00:06:13 +00001959 bool VisitBinaryOperator(const BinaryOperator *E);
1960 bool VisitFloatingLiteral(const FloatingLiteral *E);
Peter Collingbournee9200682011-05-13 03:29:01 +00001961 bool VisitCastExpr(const CastExpr *E);
1962 bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E);
Eli Friedmanc2b50172009-02-22 11:46:18 +00001963
John McCallb1fb0d32010-05-07 22:08:54 +00001964 bool VisitUnaryReal(const UnaryOperator *E);
1965 bool VisitUnaryImag(const UnaryOperator *E);
Eli Friedman449fe542009-03-23 04:56:01 +00001966
John McCalla2fabff2010-10-09 01:34:31 +00001967 bool VisitDeclRefExpr(const DeclRefExpr *E);
1968
John McCallb1fb0d32010-05-07 22:08:54 +00001969 // FIXME: Missing: array subscript of vector, member of vector,
1970 // ImplicitValueInitExpr
Eli Friedman24c01542008-08-22 00:06:13 +00001971};
1972} // end anonymous namespace
1973
1974static bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
John McCallf0c4f352010-05-07 05:46:35 +00001975 assert(E->getType()->isRealFloatingType());
Peter Collingbournee9200682011-05-13 03:29:01 +00001976 return FloatExprEvaluator(Info, Result).Visit(E);
Eli Friedman24c01542008-08-22 00:06:13 +00001977}
1978
Jay Foad39c79802011-01-12 09:06:06 +00001979static bool TryEvaluateBuiltinNaN(const ASTContext &Context,
John McCall16291492010-02-28 13:00:19 +00001980 QualType ResultTy,
1981 const Expr *Arg,
1982 bool SNaN,
1983 llvm::APFloat &Result) {
1984 const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
1985 if (!S) return false;
1986
1987 const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
1988
1989 llvm::APInt fill;
1990
1991 // Treat empty strings as if they were zero.
1992 if (S->getString().empty())
1993 fill = llvm::APInt(32, 0);
1994 else if (S->getString().getAsInteger(0, fill))
1995 return false;
1996
1997 if (SNaN)
1998 Result = llvm::APFloat::getSNaN(Sem, false, &fill);
1999 else
2000 Result = llvm::APFloat::getQNaN(Sem, false, &fill);
2001 return true;
2002}
2003
Chris Lattner4deaa4e2008-10-06 05:28:25 +00002004bool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
Douglas Gregore711f702009-02-14 18:57:46 +00002005 switch (E->isBuiltinCall(Info.Ctx)) {
Peter Collingbournee9200682011-05-13 03:29:01 +00002006 default:
2007 return ExprEvaluatorBaseTy::VisitCallExpr(E);
2008
Chris Lattner4deaa4e2008-10-06 05:28:25 +00002009 case Builtin::BI__builtin_huge_val:
2010 case Builtin::BI__builtin_huge_valf:
2011 case Builtin::BI__builtin_huge_vall:
2012 case Builtin::BI__builtin_inf:
2013 case Builtin::BI__builtin_inff:
Daniel Dunbar1be9f882008-10-14 05:41:12 +00002014 case Builtin::BI__builtin_infl: {
2015 const llvm::fltSemantics &Sem =
2016 Info.Ctx.getFloatTypeSemantics(E->getType());
Chris Lattner37346e02008-10-06 05:53:16 +00002017 Result = llvm::APFloat::getInf(Sem);
2018 return true;
Daniel Dunbar1be9f882008-10-14 05:41:12 +00002019 }
Mike Stump11289f42009-09-09 15:08:12 +00002020
John McCall16291492010-02-28 13:00:19 +00002021 case Builtin::BI__builtin_nans:
2022 case Builtin::BI__builtin_nansf:
2023 case Builtin::BI__builtin_nansl:
2024 return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
2025 true, Result);
2026
Chris Lattner0b7282e2008-10-06 06:31:58 +00002027 case Builtin::BI__builtin_nan:
2028 case Builtin::BI__builtin_nanf:
2029 case Builtin::BI__builtin_nanl:
Mike Stump2346cd22009-05-30 03:56:50 +00002030 // If this is __builtin_nan() turn this into a nan, otherwise we
Chris Lattner0b7282e2008-10-06 06:31:58 +00002031 // can't constant fold it.
John McCall16291492010-02-28 13:00:19 +00002032 return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
2033 false, Result);
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002034
2035 case Builtin::BI__builtin_fabs:
2036 case Builtin::BI__builtin_fabsf:
2037 case Builtin::BI__builtin_fabsl:
2038 if (!EvaluateFloat(E->getArg(0), Result, Info))
2039 return false;
Mike Stump11289f42009-09-09 15:08:12 +00002040
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002041 if (Result.isNegative())
2042 Result.changeSign();
2043 return true;
2044
Mike Stump11289f42009-09-09 15:08:12 +00002045 case Builtin::BI__builtin_copysign:
2046 case Builtin::BI__builtin_copysignf:
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002047 case Builtin::BI__builtin_copysignl: {
2048 APFloat RHS(0.);
2049 if (!EvaluateFloat(E->getArg(0), Result, Info) ||
2050 !EvaluateFloat(E->getArg(1), RHS, Info))
2051 return false;
2052 Result.copySign(RHS);
2053 return true;
2054 }
Chris Lattner4deaa4e2008-10-06 05:28:25 +00002055 }
2056}
2057
John McCalla2fabff2010-10-09 01:34:31 +00002058bool FloatExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
Peter Collingbournee9200682011-05-13 03:29:01 +00002059 if (ExprEvaluatorBaseTy::VisitDeclRefExpr(E))
2060 return true;
2061
John McCalla2fabff2010-10-09 01:34:31 +00002062 const Decl *D = E->getDecl();
2063 if (!isa<VarDecl>(D) || isa<ParmVarDecl>(D)) return false;
2064 const VarDecl *VD = cast<VarDecl>(D);
2065
2066 // Require the qualifiers to be const and not volatile.
2067 CanQualType T = Info.Ctx.getCanonicalType(E->getType());
2068 if (!T.isConstQualified() || T.isVolatileQualified())
2069 return false;
2070
2071 const Expr *Init = VD->getAnyInitializer();
2072 if (!Init) return false;
2073
2074 if (APValue *V = VD->getEvaluatedValue()) {
2075 if (V->isFloat()) {
2076 Result = V->getFloat();
2077 return true;
2078 }
2079 return false;
2080 }
2081
2082 if (VD->isEvaluatingValue())
2083 return false;
2084
2085 VD->setEvaluatingValue();
2086
2087 Expr::EvalResult InitResult;
2088 if (Init->Evaluate(InitResult, Info.Ctx) && !InitResult.HasSideEffects &&
2089 InitResult.Val.isFloat()) {
2090 // Cache the evaluated value in the variable declaration.
2091 Result = InitResult.Val.getFloat();
2092 VD->setEvaluatedValue(InitResult.Val);
2093 return true;
2094 }
2095
2096 VD->setEvaluatedValue(APValue());
2097 return false;
2098}
2099
John McCallb1fb0d32010-05-07 22:08:54 +00002100bool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
Eli Friedman95719532010-08-14 20:52:13 +00002101 if (E->getSubExpr()->getType()->isAnyComplexType()) {
2102 ComplexValue CV;
2103 if (!EvaluateComplex(E->getSubExpr(), CV, Info))
2104 return false;
2105 Result = CV.FloatReal;
2106 return true;
2107 }
2108
2109 return Visit(E->getSubExpr());
John McCallb1fb0d32010-05-07 22:08:54 +00002110}
2111
2112bool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
Eli Friedman95719532010-08-14 20:52:13 +00002113 if (E->getSubExpr()->getType()->isAnyComplexType()) {
2114 ComplexValue CV;
2115 if (!EvaluateComplex(E->getSubExpr(), CV, Info))
2116 return false;
2117 Result = CV.FloatImag;
2118 return true;
2119 }
2120
2121 if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
2122 Info.EvalResult.HasSideEffects = true;
2123 const llvm::fltSemantics &Sem = Info.Ctx.getFloatTypeSemantics(E->getType());
2124 Result = llvm::APFloat::getZero(Sem);
John McCallb1fb0d32010-05-07 22:08:54 +00002125 return true;
2126}
2127
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002128bool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
John McCalle3027922010-08-25 11:45:40 +00002129 if (E->getOpcode() == UO_Deref)
Nuno Lopes0e33c682008-11-19 17:44:31 +00002130 return false;
2131
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002132 if (!EvaluateFloat(E->getSubExpr(), Result, Info))
2133 return false;
2134
2135 switch (E->getOpcode()) {
2136 default: return false;
John McCalle3027922010-08-25 11:45:40 +00002137 case UO_Plus:
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002138 return true;
John McCalle3027922010-08-25 11:45:40 +00002139 case UO_Minus:
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002140 Result.changeSign();
2141 return true;
2142 }
2143}
Chris Lattner4deaa4e2008-10-06 05:28:25 +00002144
Eli Friedman24c01542008-08-22 00:06:13 +00002145bool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
John McCalle3027922010-08-25 11:45:40 +00002146 if (E->getOpcode() == BO_Comma) {
Eli Friedman141fbf32009-11-16 04:25:37 +00002147 if (!EvaluateFloat(E->getRHS(), Result, Info))
2148 return false;
2149
2150 // If we can't evaluate the LHS, it might have side effects;
2151 // conservatively mark it.
2152 if (!E->getLHS()->isEvaluatable(Info.Ctx))
2153 Info.EvalResult.HasSideEffects = true;
2154
2155 return true;
2156 }
2157
Anders Carlssona5df61a2010-10-31 01:21:47 +00002158 // We can't evaluate pointer-to-member operations.
2159 if (E->isPtrMemOp())
2160 return false;
2161
Eli Friedman24c01542008-08-22 00:06:13 +00002162 // FIXME: Diagnostics? I really don't understand how the warnings
2163 // and errors are supposed to work.
Daniel Dunbarc3d79cf2008-10-16 03:51:50 +00002164 APFloat RHS(0.0);
Eli Friedman24c01542008-08-22 00:06:13 +00002165 if (!EvaluateFloat(E->getLHS(), Result, Info))
2166 return false;
2167 if (!EvaluateFloat(E->getRHS(), RHS, Info))
2168 return false;
2169
2170 switch (E->getOpcode()) {
2171 default: return false;
John McCalle3027922010-08-25 11:45:40 +00002172 case BO_Mul:
Eli Friedman24c01542008-08-22 00:06:13 +00002173 Result.multiply(RHS, APFloat::rmNearestTiesToEven);
2174 return true;
John McCalle3027922010-08-25 11:45:40 +00002175 case BO_Add:
Eli Friedman24c01542008-08-22 00:06:13 +00002176 Result.add(RHS, APFloat::rmNearestTiesToEven);
2177 return true;
John McCalle3027922010-08-25 11:45:40 +00002178 case BO_Sub:
Eli Friedman24c01542008-08-22 00:06:13 +00002179 Result.subtract(RHS, APFloat::rmNearestTiesToEven);
2180 return true;
John McCalle3027922010-08-25 11:45:40 +00002181 case BO_Div:
Eli Friedman24c01542008-08-22 00:06:13 +00002182 Result.divide(RHS, APFloat::rmNearestTiesToEven);
2183 return true;
Eli Friedman24c01542008-08-22 00:06:13 +00002184 }
2185}
2186
2187bool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
2188 Result = E->getValue();
2189 return true;
2190}
2191
Peter Collingbournee9200682011-05-13 03:29:01 +00002192bool FloatExprEvaluator::VisitCastExpr(const CastExpr *E) {
2193 const Expr* SubExpr = E->getSubExpr();
Mike Stump11289f42009-09-09 15:08:12 +00002194
Eli Friedman8bfbe3a2011-03-25 00:54:52 +00002195 switch (E->getCastKind()) {
2196 default:
2197 return false;
2198
2199 case CK_LValueToRValue:
2200 case CK_NoOp:
2201 return Visit(SubExpr);
2202
2203 case CK_IntegralToFloating: {
Eli Friedman9a156e52008-11-12 09:44:48 +00002204 APSInt IntResult;
Daniel Dunbare3c92bc2009-02-19 18:37:50 +00002205 if (!EvaluateInteger(SubExpr, IntResult, Info))
Eli Friedman9a156e52008-11-12 09:44:48 +00002206 return false;
Mike Stump11289f42009-09-09 15:08:12 +00002207 Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
Daniel Dunbarb6f953e2009-01-29 06:16:07 +00002208 IntResult, Info.Ctx);
Eli Friedman9a156e52008-11-12 09:44:48 +00002209 return true;
2210 }
Eli Friedman8bfbe3a2011-03-25 00:54:52 +00002211
2212 case CK_FloatingCast: {
Eli Friedman9a156e52008-11-12 09:44:48 +00002213 if (!Visit(SubExpr))
2214 return false;
Daniel Dunbarb6f953e2009-01-29 06:16:07 +00002215 Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
2216 Result, Info.Ctx);
Eli Friedman9a156e52008-11-12 09:44:48 +00002217 return true;
2218 }
John McCalld7646252010-11-14 08:17:51 +00002219
Eli Friedman8bfbe3a2011-03-25 00:54:52 +00002220 case CK_FloatingComplexToReal: {
John McCalld7646252010-11-14 08:17:51 +00002221 ComplexValue V;
2222 if (!EvaluateComplex(SubExpr, V, Info))
2223 return false;
2224 Result = V.getComplexFloatReal();
2225 return true;
2226 }
Eli Friedman8bfbe3a2011-03-25 00:54:52 +00002227 }
Eli Friedman9a156e52008-11-12 09:44:48 +00002228
2229 return false;
2230}
2231
Peter Collingbournee9200682011-05-13 03:29:01 +00002232bool FloatExprEvaluator::VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
Eli Friedman9a156e52008-11-12 09:44:48 +00002233 Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
2234 return true;
2235}
2236
Eli Friedman24c01542008-08-22 00:06:13 +00002237//===----------------------------------------------------------------------===//
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002238// Complex Evaluation (for float and integer)
Anders Carlsson537969c2008-11-16 20:27:53 +00002239//===----------------------------------------------------------------------===//
2240
2241namespace {
Benjamin Kramer26222b62009-11-28 19:03:38 +00002242class ComplexExprEvaluator
Peter Collingbournee9200682011-05-13 03:29:01 +00002243 : public ExprEvaluatorBase<ComplexExprEvaluator, bool> {
John McCall93d91dc2010-05-07 17:22:02 +00002244 ComplexValue &Result;
Mike Stump11289f42009-09-09 15:08:12 +00002245
Anders Carlsson537969c2008-11-16 20:27:53 +00002246public:
John McCall93d91dc2010-05-07 17:22:02 +00002247 ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
Peter Collingbournee9200682011-05-13 03:29:01 +00002248 : ExprEvaluatorBaseTy(info), Result(Result) {}
2249
2250 bool Success(const APValue &V, const Expr *e) {
2251 Result.setFrom(V);
2252 return true;
2253 }
2254 bool Error(const Expr *E) {
2255 return false;
2256 }
Mike Stump11289f42009-09-09 15:08:12 +00002257
Anders Carlsson537969c2008-11-16 20:27:53 +00002258 //===--------------------------------------------------------------------===//
2259 // Visitor Methods
2260 //===--------------------------------------------------------------------===//
2261
Peter Collingbournee9200682011-05-13 03:29:01 +00002262 bool VisitImaginaryLiteral(const ImaginaryLiteral *E);
Mike Stump11289f42009-09-09 15:08:12 +00002263
Peter Collingbournee9200682011-05-13 03:29:01 +00002264 bool VisitCastExpr(const CastExpr *E);
Mike Stump11289f42009-09-09 15:08:12 +00002265
John McCall93d91dc2010-05-07 17:22:02 +00002266 bool VisitBinaryOperator(const BinaryOperator *E);
Abramo Bagnara9e0e7092010-12-11 16:05:48 +00002267 bool VisitUnaryOperator(const UnaryOperator *E);
Abramo Bagnara9e0e7092010-12-11 16:05:48 +00002268 // FIXME Missing: ImplicitValueInitExpr
Anders Carlsson537969c2008-11-16 20:27:53 +00002269};
2270} // end anonymous namespace
2271
John McCall93d91dc2010-05-07 17:22:02 +00002272static bool EvaluateComplex(const Expr *E, ComplexValue &Result,
2273 EvalInfo &Info) {
John McCallf0c4f352010-05-07 05:46:35 +00002274 assert(E->getType()->isAnyComplexType());
Peter Collingbournee9200682011-05-13 03:29:01 +00002275 return ComplexExprEvaluator(Info, Result).Visit(E);
Anders Carlsson537969c2008-11-16 20:27:53 +00002276}
2277
Peter Collingbournee9200682011-05-13 03:29:01 +00002278bool ComplexExprEvaluator::VisitImaginaryLiteral(const ImaginaryLiteral *E) {
2279 const Expr* SubExpr = E->getSubExpr();
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002280
2281 if (SubExpr->getType()->isRealFloatingType()) {
2282 Result.makeComplexFloat();
2283 APFloat &Imag = Result.FloatImag;
2284 if (!EvaluateFloat(SubExpr, Imag, Info))
2285 return false;
2286
2287 Result.FloatReal = APFloat(Imag.getSemantics());
2288 return true;
2289 } else {
2290 assert(SubExpr->getType()->isIntegerType() &&
2291 "Unexpected imaginary literal.");
2292
2293 Result.makeComplexInt();
2294 APSInt &Imag = Result.IntImag;
2295 if (!EvaluateInteger(SubExpr, Imag, Info))
2296 return false;
2297
2298 Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
2299 return true;
2300 }
2301}
2302
Peter Collingbournee9200682011-05-13 03:29:01 +00002303bool ComplexExprEvaluator::VisitCastExpr(const CastExpr *E) {
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002304
John McCallfcef3cf2010-12-14 17:51:41 +00002305 switch (E->getCastKind()) {
2306 case CK_BitCast:
John McCallfcef3cf2010-12-14 17:51:41 +00002307 case CK_BaseToDerived:
2308 case CK_DerivedToBase:
2309 case CK_UncheckedDerivedToBase:
2310 case CK_Dynamic:
2311 case CK_ToUnion:
2312 case CK_ArrayToPointerDecay:
2313 case CK_FunctionToPointerDecay:
2314 case CK_NullToPointer:
2315 case CK_NullToMemberPointer:
2316 case CK_BaseToDerivedMemberPointer:
2317 case CK_DerivedToBaseMemberPointer:
2318 case CK_MemberPointerToBoolean:
2319 case CK_ConstructorConversion:
2320 case CK_IntegralToPointer:
2321 case CK_PointerToIntegral:
2322 case CK_PointerToBoolean:
2323 case CK_ToVoid:
2324 case CK_VectorSplat:
2325 case CK_IntegralCast:
2326 case CK_IntegralToBoolean:
2327 case CK_IntegralToFloating:
2328 case CK_FloatingToIntegral:
2329 case CK_FloatingToBoolean:
2330 case CK_FloatingCast:
John McCall9320b872011-09-09 05:25:32 +00002331 case CK_CPointerToObjCPointerCast:
2332 case CK_BlockPointerToObjCPointerCast:
John McCallfcef3cf2010-12-14 17:51:41 +00002333 case CK_AnyPointerToBlockPointerCast:
2334 case CK_ObjCObjectLValueCast:
2335 case CK_FloatingComplexToReal:
2336 case CK_FloatingComplexToBoolean:
2337 case CK_IntegralComplexToReal:
2338 case CK_IntegralComplexToBoolean:
John McCall2d637d22011-09-10 06:18:15 +00002339 case CK_ARCProduceObject:
2340 case CK_ARCConsumeObject:
2341 case CK_ARCReclaimReturnedObject:
2342 case CK_ARCExtendBlockObject:
John McCallfcef3cf2010-12-14 17:51:41 +00002343 llvm_unreachable("invalid cast kind for complex value");
John McCallc5e62b42010-11-13 09:02:35 +00002344
John McCallfcef3cf2010-12-14 17:51:41 +00002345 case CK_LValueToRValue:
2346 case CK_NoOp:
2347 return Visit(E->getSubExpr());
2348
2349 case CK_Dependent:
2350 case CK_GetObjCProperty:
Eli Friedmanc757de22011-03-25 00:43:55 +00002351 case CK_LValueBitCast:
John McCallfcef3cf2010-12-14 17:51:41 +00002352 case CK_UserDefinedConversion:
2353 return false;
2354
2355 case CK_FloatingRealToComplex: {
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002356 APFloat &Real = Result.FloatReal;
John McCallfcef3cf2010-12-14 17:51:41 +00002357 if (!EvaluateFloat(E->getSubExpr(), Real, Info))
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002358 return false;
2359
John McCallfcef3cf2010-12-14 17:51:41 +00002360 Result.makeComplexFloat();
2361 Result.FloatImag = APFloat(Real.getSemantics());
2362 return true;
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002363 }
2364
John McCallfcef3cf2010-12-14 17:51:41 +00002365 case CK_FloatingComplexCast: {
2366 if (!Visit(E->getSubExpr()))
2367 return false;
2368
2369 QualType To = E->getType()->getAs<ComplexType>()->getElementType();
2370 QualType From
2371 = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
2372
2373 Result.FloatReal
2374 = HandleFloatToFloatCast(To, From, Result.FloatReal, Info.Ctx);
2375 Result.FloatImag
2376 = HandleFloatToFloatCast(To, From, Result.FloatImag, Info.Ctx);
2377 return true;
2378 }
2379
2380 case CK_FloatingComplexToIntegralComplex: {
2381 if (!Visit(E->getSubExpr()))
2382 return false;
2383
2384 QualType To = E->getType()->getAs<ComplexType>()->getElementType();
2385 QualType From
2386 = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
2387 Result.makeComplexInt();
2388 Result.IntReal = HandleFloatToIntCast(To, From, Result.FloatReal, Info.Ctx);
2389 Result.IntImag = HandleFloatToIntCast(To, From, Result.FloatImag, Info.Ctx);
2390 return true;
2391 }
2392
2393 case CK_IntegralRealToComplex: {
2394 APSInt &Real = Result.IntReal;
2395 if (!EvaluateInteger(E->getSubExpr(), Real, Info))
2396 return false;
2397
2398 Result.makeComplexInt();
2399 Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
2400 return true;
2401 }
2402
2403 case CK_IntegralComplexCast: {
2404 if (!Visit(E->getSubExpr()))
2405 return false;
2406
2407 QualType To = E->getType()->getAs<ComplexType>()->getElementType();
2408 QualType From
2409 = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
2410
2411 Result.IntReal = HandleIntToIntCast(To, From, Result.IntReal, Info.Ctx);
2412 Result.IntImag = HandleIntToIntCast(To, From, Result.IntImag, Info.Ctx);
2413 return true;
2414 }
2415
2416 case CK_IntegralComplexToFloatingComplex: {
2417 if (!Visit(E->getSubExpr()))
2418 return false;
2419
2420 QualType To = E->getType()->getAs<ComplexType>()->getElementType();
2421 QualType From
2422 = E->getSubExpr()->getType()->getAs<ComplexType>()->getElementType();
2423 Result.makeComplexFloat();
2424 Result.FloatReal = HandleIntToFloatCast(To, From, Result.IntReal, Info.Ctx);
2425 Result.FloatImag = HandleIntToFloatCast(To, From, Result.IntImag, Info.Ctx);
2426 return true;
2427 }
2428 }
2429
2430 llvm_unreachable("unknown cast resulting in complex value");
Eli Friedmanc3e9df32010-08-16 23:27:44 +00002431 return false;
2432}
2433
John McCall93d91dc2010-05-07 17:22:02 +00002434bool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
Abramo Bagnara9e0e7092010-12-11 16:05:48 +00002435 if (E->getOpcode() == BO_Comma) {
2436 if (!Visit(E->getRHS()))
2437 return false;
2438
2439 // If we can't evaluate the LHS, it might have side effects;
2440 // conservatively mark it.
2441 if (!E->getLHS()->isEvaluatable(Info.Ctx))
2442 Info.EvalResult.HasSideEffects = true;
2443
2444 return true;
2445 }
John McCall93d91dc2010-05-07 17:22:02 +00002446 if (!Visit(E->getLHS()))
2447 return false;
Mike Stump11289f42009-09-09 15:08:12 +00002448
John McCall93d91dc2010-05-07 17:22:02 +00002449 ComplexValue RHS;
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002450 if (!EvaluateComplex(E->getRHS(), RHS, Info))
John McCall93d91dc2010-05-07 17:22:02 +00002451 return false;
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002452
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002453 assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
2454 "Invalid operands to binary operator.");
Anders Carlsson9ddf7be2008-11-16 21:51:21 +00002455 switch (E->getOpcode()) {
John McCall93d91dc2010-05-07 17:22:02 +00002456 default: return false;
John McCalle3027922010-08-25 11:45:40 +00002457 case BO_Add:
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002458 if (Result.isComplexFloat()) {
2459 Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
2460 APFloat::rmNearestTiesToEven);
2461 Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
2462 APFloat::rmNearestTiesToEven);
2463 } else {
2464 Result.getComplexIntReal() += RHS.getComplexIntReal();
2465 Result.getComplexIntImag() += RHS.getComplexIntImag();
2466 }
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002467 break;
John McCalle3027922010-08-25 11:45:40 +00002468 case BO_Sub:
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002469 if (Result.isComplexFloat()) {
2470 Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
2471 APFloat::rmNearestTiesToEven);
2472 Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
2473 APFloat::rmNearestTiesToEven);
2474 } else {
2475 Result.getComplexIntReal() -= RHS.getComplexIntReal();
2476 Result.getComplexIntImag() -= RHS.getComplexIntImag();
2477 }
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002478 break;
John McCalle3027922010-08-25 11:45:40 +00002479 case BO_Mul:
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002480 if (Result.isComplexFloat()) {
John McCall93d91dc2010-05-07 17:22:02 +00002481 ComplexValue LHS = Result;
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002482 APFloat &LHS_r = LHS.getComplexFloatReal();
2483 APFloat &LHS_i = LHS.getComplexFloatImag();
2484 APFloat &RHS_r = RHS.getComplexFloatReal();
2485 APFloat &RHS_i = RHS.getComplexFloatImag();
Mike Stump11289f42009-09-09 15:08:12 +00002486
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002487 APFloat Tmp = LHS_r;
2488 Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
2489 Result.getComplexFloatReal() = Tmp;
2490 Tmp = LHS_i;
2491 Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
2492 Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
2493
2494 Tmp = LHS_r;
2495 Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
2496 Result.getComplexFloatImag() = Tmp;
2497 Tmp = LHS_i;
2498 Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
2499 Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
2500 } else {
John McCall93d91dc2010-05-07 17:22:02 +00002501 ComplexValue LHS = Result;
Mike Stump11289f42009-09-09 15:08:12 +00002502 Result.getComplexIntReal() =
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002503 (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
2504 LHS.getComplexIntImag() * RHS.getComplexIntImag());
Mike Stump11289f42009-09-09 15:08:12 +00002505 Result.getComplexIntImag() =
Daniel Dunbar0aa26062009-01-29 01:32:56 +00002506 (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
2507 LHS.getComplexIntImag() * RHS.getComplexIntReal());
2508 }
2509 break;
Abramo Bagnara9e0e7092010-12-11 16:05:48 +00002510 case BO_Div:
2511 if (Result.isComplexFloat()) {
2512 ComplexValue LHS = Result;
2513 APFloat &LHS_r = LHS.getComplexFloatReal();
2514 APFloat &LHS_i = LHS.getComplexFloatImag();
2515 APFloat &RHS_r = RHS.getComplexFloatReal();
2516 APFloat &RHS_i = RHS.getComplexFloatImag();
2517 APFloat &Res_r = Result.getComplexFloatReal();
2518 APFloat &Res_i = Result.getComplexFloatImag();
2519
2520 APFloat Den = RHS_r;
2521 Den.multiply(RHS_r, APFloat::rmNearestTiesToEven);
2522 APFloat Tmp = RHS_i;
2523 Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
2524 Den.add(Tmp, APFloat::rmNearestTiesToEven);
2525
2526 Res_r = LHS_r;
2527 Res_r.multiply(RHS_r, APFloat::rmNearestTiesToEven);
2528 Tmp = LHS_i;
2529 Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
2530 Res_r.add(Tmp, APFloat::rmNearestTiesToEven);
2531 Res_r.divide(Den, APFloat::rmNearestTiesToEven);
2532
2533 Res_i = LHS_i;
2534 Res_i.multiply(RHS_r, APFloat::rmNearestTiesToEven);
2535 Tmp = LHS_r;
2536 Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
2537 Res_i.subtract(Tmp, APFloat::rmNearestTiesToEven);
2538 Res_i.divide(Den, APFloat::rmNearestTiesToEven);
2539 } else {
2540 if (RHS.getComplexIntReal() == 0 && RHS.getComplexIntImag() == 0) {
2541 // FIXME: what about diagnostics?
2542 return false;
2543 }
2544 ComplexValue LHS = Result;
2545 APSInt Den = RHS.getComplexIntReal() * RHS.getComplexIntReal() +
2546 RHS.getComplexIntImag() * RHS.getComplexIntImag();
2547 Result.getComplexIntReal() =
2548 (LHS.getComplexIntReal() * RHS.getComplexIntReal() +
2549 LHS.getComplexIntImag() * RHS.getComplexIntImag()) / Den;
2550 Result.getComplexIntImag() =
2551 (LHS.getComplexIntImag() * RHS.getComplexIntReal() -
2552 LHS.getComplexIntReal() * RHS.getComplexIntImag()) / Den;
2553 }
2554 break;
Anders Carlsson9ddf7be2008-11-16 21:51:21 +00002555 }
2556
John McCall93d91dc2010-05-07 17:22:02 +00002557 return true;
Anders Carlsson9ddf7be2008-11-16 21:51:21 +00002558}
2559
Abramo Bagnara9e0e7092010-12-11 16:05:48 +00002560bool ComplexExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
2561 // Get the operand value into 'Result'.
2562 if (!Visit(E->getSubExpr()))
2563 return false;
2564
2565 switch (E->getOpcode()) {
2566 default:
2567 // FIXME: what about diagnostics?
2568 return false;
2569 case UO_Extension:
2570 return true;
2571 case UO_Plus:
2572 // The result is always just the subexpr.
2573 return true;
2574 case UO_Minus:
2575 if (Result.isComplexFloat()) {
2576 Result.getComplexFloatReal().changeSign();
2577 Result.getComplexFloatImag().changeSign();
2578 }
2579 else {
2580 Result.getComplexIntReal() = -Result.getComplexIntReal();
2581 Result.getComplexIntImag() = -Result.getComplexIntImag();
2582 }
2583 return true;
2584 case UO_Not:
2585 if (Result.isComplexFloat())
2586 Result.getComplexFloatImag().changeSign();
2587 else
2588 Result.getComplexIntImag() = -Result.getComplexIntImag();
2589 return true;
2590 }
2591}
2592
Anders Carlsson537969c2008-11-16 20:27:53 +00002593//===----------------------------------------------------------------------===//
Chris Lattner67d7b922008-11-16 21:24:15 +00002594// Top level Expr::Evaluate method.
Chris Lattner05706e882008-07-11 18:11:29 +00002595//===----------------------------------------------------------------------===//
2596
John McCallc07a0c72011-02-17 10:25:35 +00002597static bool Evaluate(EvalInfo &Info, const Expr *E) {
John McCall45d55e42010-05-07 21:00:08 +00002598 if (E->getType()->isVectorType()) {
2599 if (!EvaluateVector(E, Info.EvalResult.Val, Info))
Nate Begeman2f2bdeb2009-01-18 03:20:47 +00002600 return false;
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00002601 } else if (E->getType()->isIntegralOrEnumerationType()) {
Peter Collingbournee9200682011-05-13 03:29:01 +00002602 if (!IntExprEvaluator(Info, Info.EvalResult.Val).Visit(E))
Anders Carlsson475f4bc2008-11-22 21:50:49 +00002603 return false;
John McCallc07a0c72011-02-17 10:25:35 +00002604 if (Info.EvalResult.Val.isLValue() &&
2605 !IsGlobalLValue(Info.EvalResult.Val.getLValueBase()))
John McCall11086fc2010-07-07 05:08:32 +00002606 return false;
John McCall45d55e42010-05-07 21:00:08 +00002607 } else if (E->getType()->hasPointerRepresentation()) {
2608 LValue LV;
2609 if (!EvaluatePointer(E, LV, Info))
Anders Carlsson475f4bc2008-11-22 21:50:49 +00002610 return false;
Abramo Bagnaraf8199452010-05-14 17:07:14 +00002611 if (!IsGlobalLValue(LV.Base))
John McCall95007602010-05-10 23:27:23 +00002612 return false;
John McCall45d55e42010-05-07 21:00:08 +00002613 LV.moveInto(Info.EvalResult.Val);
2614 } else if (E->getType()->isRealFloatingType()) {
2615 llvm::APFloat F(0.0);
2616 if (!EvaluateFloat(E, F, Info))
Anders Carlsson475f4bc2008-11-22 21:50:49 +00002617 return false;
Mike Stump11289f42009-09-09 15:08:12 +00002618
John McCall45d55e42010-05-07 21:00:08 +00002619 Info.EvalResult.Val = APValue(F);
2620 } else if (E->getType()->isAnyComplexType()) {
2621 ComplexValue C;
2622 if (!EvaluateComplex(E, C, Info))
Anders Carlsson475f4bc2008-11-22 21:50:49 +00002623 return false;
John McCall45d55e42010-05-07 21:00:08 +00002624 C.moveInto(Info.EvalResult.Val);
Daniel Dunbarf50e60b2009-01-28 22:24:07 +00002625 } else
Anders Carlsson7c282e42008-11-22 22:56:32 +00002626 return false;
Anders Carlsson475f4bc2008-11-22 21:50:49 +00002627
Anders Carlsson7b6f0af2008-11-30 16:58:53 +00002628 return true;
2629}
2630
John McCallc07a0c72011-02-17 10:25:35 +00002631/// Evaluate - Return true if this is a constant which we can fold using
2632/// any crazy technique (that has nothing to do with language standards) that
2633/// we want to. If this function returns true, it returns the folded constant
2634/// in Result.
2635bool Expr::Evaluate(EvalResult &Result, const ASTContext &Ctx) const {
2636 EvalInfo Info(Ctx, Result);
2637 return ::Evaluate(Info, this);
2638}
2639
Jay Foad39c79802011-01-12 09:06:06 +00002640bool Expr::EvaluateAsBooleanCondition(bool &Result,
2641 const ASTContext &Ctx) const {
John McCall1be1c632010-01-05 23:42:56 +00002642 EvalResult Scratch;
2643 EvalInfo Info(Ctx, Scratch);
2644
2645 return HandleConversionToBool(this, Result, Info);
2646}
2647
Jay Foad39c79802011-01-12 09:06:06 +00002648bool Expr::EvaluateAsLValue(EvalResult &Result, const ASTContext &Ctx) const {
Anders Carlsson43168122009-04-10 04:54:13 +00002649 EvalInfo Info(Ctx, Result);
2650
John McCall45d55e42010-05-07 21:00:08 +00002651 LValue LV;
John McCall95007602010-05-10 23:27:23 +00002652 if (EvaluateLValue(this, LV, Info) &&
2653 !Result.HasSideEffects &&
Abramo Bagnaraf8199452010-05-14 17:07:14 +00002654 IsGlobalLValue(LV.Base)) {
2655 LV.moveInto(Result.Val);
2656 return true;
2657 }
2658 return false;
2659}
2660
Jay Foad39c79802011-01-12 09:06:06 +00002661bool Expr::EvaluateAsAnyLValue(EvalResult &Result,
2662 const ASTContext &Ctx) const {
Abramo Bagnaraf8199452010-05-14 17:07:14 +00002663 EvalInfo Info(Ctx, Result);
2664
2665 LValue LV;
2666 if (EvaluateLValue(this, LV, Info)) {
John McCall45d55e42010-05-07 21:00:08 +00002667 LV.moveInto(Result.Val);
2668 return true;
2669 }
2670 return false;
Eli Friedman7d45c482009-09-13 10:17:44 +00002671}
2672
Chris Lattner67d7b922008-11-16 21:24:15 +00002673/// isEvaluatable - Call Evaluate to see if this expression can be constant
Chris Lattnercb136912008-10-06 06:49:02 +00002674/// folded, but discard the result.
Jay Foad39c79802011-01-12 09:06:06 +00002675bool Expr::isEvaluatable(const ASTContext &Ctx) const {
Anders Carlsson5b3638b2008-12-01 06:44:05 +00002676 EvalResult Result;
2677 return Evaluate(Result, Ctx) && !Result.HasSideEffects;
Chris Lattnercb136912008-10-06 06:49:02 +00002678}
Anders Carlsson59689ed2008-11-22 21:04:56 +00002679
Jay Foad39c79802011-01-12 09:06:06 +00002680bool Expr::HasSideEffects(const ASTContext &Ctx) const {
Fariborz Jahanian4127b8e2009-11-05 18:03:03 +00002681 Expr::EvalResult Result;
2682 EvalInfo Info(Ctx, Result);
Peter Collingbournee9200682011-05-13 03:29:01 +00002683 return HasSideEffect(Info).Visit(this);
Fariborz Jahanian4127b8e2009-11-05 18:03:03 +00002684}
2685
Jay Foad39c79802011-01-12 09:06:06 +00002686APSInt Expr::EvaluateAsInt(const ASTContext &Ctx) const {
Anders Carlsson6736d1a22008-12-19 20:58:05 +00002687 EvalResult EvalResult;
2688 bool Result = Evaluate(EvalResult, Ctx);
Jeffrey Yasskinb3321532010-12-23 01:01:28 +00002689 (void)Result;
Anders Carlsson59689ed2008-11-22 21:04:56 +00002690 assert(Result && "Could not evaluate expression");
Anders Carlsson6736d1a22008-12-19 20:58:05 +00002691 assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
Anders Carlsson59689ed2008-11-22 21:04:56 +00002692
Anders Carlsson6736d1a22008-12-19 20:58:05 +00002693 return EvalResult.Val.getInt();
Anders Carlsson59689ed2008-11-22 21:04:56 +00002694}
John McCall864e3962010-05-07 05:32:02 +00002695
Abramo Bagnaraf8199452010-05-14 17:07:14 +00002696 bool Expr::EvalResult::isGlobalLValue() const {
2697 assert(Val.isLValue());
2698 return IsGlobalLValue(Val.getLValueBase());
2699 }
2700
2701
John McCall864e3962010-05-07 05:32:02 +00002702/// isIntegerConstantExpr - this recursive routine will test if an expression is
2703/// an integer constant expression.
2704
2705/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
2706/// comma, etc
2707///
2708/// FIXME: Handle offsetof. Two things to do: Handle GCC's __builtin_offsetof
2709/// to support gcc 4.0+ and handle the idiom GCC recognizes with a null pointer
2710/// cast+dereference.
2711
2712// CheckICE - This function does the fundamental ICE checking: the returned
2713// ICEDiag contains a Val of 0, 1, or 2, and a possibly null SourceLocation.
2714// Note that to reduce code duplication, this helper does no evaluation
2715// itself; the caller checks whether the expression is evaluatable, and
2716// in the rare cases where CheckICE actually cares about the evaluated
2717// value, it calls into Evalute.
2718//
2719// Meanings of Val:
2720// 0: This expression is an ICE if it can be evaluated by Evaluate.
2721// 1: This expression is not an ICE, but if it isn't evaluated, it's
2722// a legal subexpression for an ICE. This return value is used to handle
2723// the comma operator in C99 mode.
2724// 2: This expression is not an ICE, and is not a legal subexpression for one.
2725
Dan Gohman28ade552010-07-26 21:25:24 +00002726namespace {
2727
John McCall864e3962010-05-07 05:32:02 +00002728struct ICEDiag {
2729 unsigned Val;
2730 SourceLocation Loc;
2731
2732 public:
2733 ICEDiag(unsigned v, SourceLocation l) : Val(v), Loc(l) {}
2734 ICEDiag() : Val(0) {}
2735};
2736
Dan Gohman28ade552010-07-26 21:25:24 +00002737}
2738
2739static ICEDiag NoDiag() { return ICEDiag(); }
John McCall864e3962010-05-07 05:32:02 +00002740
2741static ICEDiag CheckEvalInICE(const Expr* E, ASTContext &Ctx) {
2742 Expr::EvalResult EVResult;
2743 if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2744 !EVResult.Val.isInt()) {
2745 return ICEDiag(2, E->getLocStart());
2746 }
2747 return NoDiag();
2748}
2749
2750static ICEDiag CheckICE(const Expr* E, ASTContext &Ctx) {
2751 assert(!E->isValueDependent() && "Should not see value dependent exprs!");
Douglas Gregorb90df602010-06-16 00:17:44 +00002752 if (!E->getType()->isIntegralOrEnumerationType()) {
John McCall864e3962010-05-07 05:32:02 +00002753 return ICEDiag(2, E->getLocStart());
2754 }
2755
2756 switch (E->getStmtClass()) {
John McCallbd066782011-02-09 08:16:59 +00002757#define ABSTRACT_STMT(Node)
John McCall864e3962010-05-07 05:32:02 +00002758#define STMT(Node, Base) case Expr::Node##Class:
2759#define EXPR(Node, Base)
2760#include "clang/AST/StmtNodes.inc"
2761 case Expr::PredefinedExprClass:
2762 case Expr::FloatingLiteralClass:
2763 case Expr::ImaginaryLiteralClass:
2764 case Expr::StringLiteralClass:
2765 case Expr::ArraySubscriptExprClass:
2766 case Expr::MemberExprClass:
2767 case Expr::CompoundAssignOperatorClass:
2768 case Expr::CompoundLiteralExprClass:
2769 case Expr::ExtVectorElementExprClass:
2770 case Expr::InitListExprClass:
2771 case Expr::DesignatedInitExprClass:
2772 case Expr::ImplicitValueInitExprClass:
2773 case Expr::ParenListExprClass:
2774 case Expr::VAArgExprClass:
2775 case Expr::AddrLabelExprClass:
2776 case Expr::StmtExprClass:
2777 case Expr::CXXMemberCallExprClass:
Peter Collingbourne41f85462011-02-09 21:07:24 +00002778 case Expr::CUDAKernelCallExprClass:
John McCall864e3962010-05-07 05:32:02 +00002779 case Expr::CXXDynamicCastExprClass:
2780 case Expr::CXXTypeidExprClass:
Francois Pichet5cc0a672010-09-08 23:47:05 +00002781 case Expr::CXXUuidofExprClass:
John McCall864e3962010-05-07 05:32:02 +00002782 case Expr::CXXNullPtrLiteralExprClass:
2783 case Expr::CXXThisExprClass:
2784 case Expr::CXXThrowExprClass:
2785 case Expr::CXXNewExprClass:
2786 case Expr::CXXDeleteExprClass:
2787 case Expr::CXXPseudoDestructorExprClass:
2788 case Expr::UnresolvedLookupExprClass:
2789 case Expr::DependentScopeDeclRefExprClass:
2790 case Expr::CXXConstructExprClass:
2791 case Expr::CXXBindTemporaryExprClass:
John McCall5d413782010-12-06 08:20:24 +00002792 case Expr::ExprWithCleanupsClass:
John McCall864e3962010-05-07 05:32:02 +00002793 case Expr::CXXTemporaryObjectExprClass:
2794 case Expr::CXXUnresolvedConstructExprClass:
2795 case Expr::CXXDependentScopeMemberExprClass:
2796 case Expr::UnresolvedMemberExprClass:
2797 case Expr::ObjCStringLiteralClass:
2798 case Expr::ObjCEncodeExprClass:
2799 case Expr::ObjCMessageExprClass:
2800 case Expr::ObjCSelectorExprClass:
2801 case Expr::ObjCProtocolExprClass:
2802 case Expr::ObjCIvarRefExprClass:
2803 case Expr::ObjCPropertyRefExprClass:
John McCall864e3962010-05-07 05:32:02 +00002804 case Expr::ObjCIsaExprClass:
2805 case Expr::ShuffleVectorExprClass:
2806 case Expr::BlockExprClass:
2807 case Expr::BlockDeclRefExprClass:
2808 case Expr::NoStmtClass:
John McCall8d69a212010-11-15 23:31:06 +00002809 case Expr::OpaqueValueExprClass:
Douglas Gregore8e9dd62011-01-03 17:17:50 +00002810 case Expr::PackExpansionExprClass:
Douglas Gregorcdbc5392011-01-15 01:15:58 +00002811 case Expr::SubstNonTypeTemplateParmPackExprClass:
Tanya Lattner55808c12011-06-04 00:47:47 +00002812 case Expr::AsTypeExprClass:
John McCall31168b02011-06-15 23:02:42 +00002813 case Expr::ObjCIndirectCopyRestoreExprClass:
Douglas Gregorfe314812011-06-21 17:03:29 +00002814 case Expr::MaterializeTemporaryExprClass:
John McCall864e3962010-05-07 05:32:02 +00002815 return ICEDiag(2, E->getLocStart());
2816
Douglas Gregor820ba7b2011-01-04 17:33:58 +00002817 case Expr::SizeOfPackExprClass:
John McCall864e3962010-05-07 05:32:02 +00002818 case Expr::GNUNullExprClass:
2819 // GCC considers the GNU __null value to be an integral constant expression.
2820 return NoDiag();
2821
John McCall7c454bb2011-07-15 05:09:51 +00002822 case Expr::SubstNonTypeTemplateParmExprClass:
2823 return
2824 CheckICE(cast<SubstNonTypeTemplateParmExpr>(E)->getReplacement(), Ctx);
2825
John McCall864e3962010-05-07 05:32:02 +00002826 case Expr::ParenExprClass:
2827 return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
Peter Collingbourne91147592011-04-15 00:35:48 +00002828 case Expr::GenericSelectionExprClass:
2829 return CheckICE(cast<GenericSelectionExpr>(E)->getResultExpr(), Ctx);
John McCall864e3962010-05-07 05:32:02 +00002830 case Expr::IntegerLiteralClass:
2831 case Expr::CharacterLiteralClass:
2832 case Expr::CXXBoolLiteralExprClass:
Douglas Gregor747eb782010-07-08 06:14:04 +00002833 case Expr::CXXScalarValueInitExprClass:
John McCall864e3962010-05-07 05:32:02 +00002834 case Expr::UnaryTypeTraitExprClass:
Francois Pichet9dfa3ce2010-12-07 00:08:36 +00002835 case Expr::BinaryTypeTraitExprClass:
John Wiegley6242b6a2011-04-28 00:16:57 +00002836 case Expr::ArrayTypeTraitExprClass:
John Wiegleyf9f65842011-04-25 06:54:41 +00002837 case Expr::ExpressionTraitExprClass:
Sebastian Redl4202c0f2010-09-10 20:55:43 +00002838 case Expr::CXXNoexceptExprClass:
John McCall864e3962010-05-07 05:32:02 +00002839 return NoDiag();
2840 case Expr::CallExprClass:
Alexis Hunt3b791862010-08-30 17:47:05 +00002841 case Expr::CXXOperatorCallExprClass: {
John McCall864e3962010-05-07 05:32:02 +00002842 const CallExpr *CE = cast<CallExpr>(E);
2843 if (CE->isBuiltinCall(Ctx))
2844 return CheckEvalInICE(E, Ctx);
2845 return ICEDiag(2, E->getLocStart());
2846 }
2847 case Expr::DeclRefExprClass:
2848 if (isa<EnumConstantDecl>(cast<DeclRefExpr>(E)->getDecl()))
2849 return NoDiag();
2850 if (Ctx.getLangOptions().CPlusPlus &&
2851 E->getType().getCVRQualifiers() == Qualifiers::Const) {
2852 const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
2853
2854 // Parameter variables are never constants. Without this check,
2855 // getAnyInitializer() can find a default argument, which leads
2856 // to chaos.
2857 if (isa<ParmVarDecl>(D))
2858 return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2859
2860 // C++ 7.1.5.1p2
2861 // A variable of non-volatile const-qualified integral or enumeration
2862 // type initialized by an ICE can be used in ICEs.
2863 if (const VarDecl *Dcl = dyn_cast<VarDecl>(D)) {
2864 Qualifiers Quals = Ctx.getCanonicalType(Dcl->getType()).getQualifiers();
2865 if (Quals.hasVolatile() || !Quals.hasConst())
2866 return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2867
2868 // Look for a declaration of this variable that has an initializer.
2869 const VarDecl *ID = 0;
2870 const Expr *Init = Dcl->getAnyInitializer(ID);
2871 if (Init) {
2872 if (ID->isInitKnownICE()) {
2873 // We have already checked whether this subexpression is an
2874 // integral constant expression.
2875 if (ID->isInitICE())
2876 return NoDiag();
2877 else
2878 return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2879 }
2880
2881 // It's an ICE whether or not the definition we found is
2882 // out-of-line. See DR 721 and the discussion in Clang PR
2883 // 6206 for details.
2884
2885 if (Dcl->isCheckingICE()) {
2886 return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2887 }
2888
2889 Dcl->setCheckingICE();
2890 ICEDiag Result = CheckICE(Init, Ctx);
2891 // Cache the result of the ICE test.
2892 Dcl->setInitKnownICE(Result.Val == 0);
2893 return Result;
2894 }
2895 }
2896 }
2897 return ICEDiag(2, E->getLocStart());
2898 case Expr::UnaryOperatorClass: {
2899 const UnaryOperator *Exp = cast<UnaryOperator>(E);
2900 switch (Exp->getOpcode()) {
John McCalle3027922010-08-25 11:45:40 +00002901 case UO_PostInc:
2902 case UO_PostDec:
2903 case UO_PreInc:
2904 case UO_PreDec:
2905 case UO_AddrOf:
2906 case UO_Deref:
John McCall864e3962010-05-07 05:32:02 +00002907 return ICEDiag(2, E->getLocStart());
John McCalle3027922010-08-25 11:45:40 +00002908 case UO_Extension:
2909 case UO_LNot:
2910 case UO_Plus:
2911 case UO_Minus:
2912 case UO_Not:
2913 case UO_Real:
2914 case UO_Imag:
John McCall864e3962010-05-07 05:32:02 +00002915 return CheckICE(Exp->getSubExpr(), Ctx);
John McCall864e3962010-05-07 05:32:02 +00002916 }
2917
2918 // OffsetOf falls through here.
2919 }
2920 case Expr::OffsetOfExprClass: {
2921 // Note that per C99, offsetof must be an ICE. And AFAIK, using
2922 // Evaluate matches the proposed gcc behavior for cases like
2923 // "offsetof(struct s{int x[4];}, x[!.0])". This doesn't affect
2924 // compliance: we should warn earlier for offsetof expressions with
2925 // array subscripts that aren't ICEs, and if the array subscripts
2926 // are ICEs, the value of the offsetof must be an integer constant.
2927 return CheckEvalInICE(E, Ctx);
2928 }
Peter Collingbournee190dee2011-03-11 19:24:49 +00002929 case Expr::UnaryExprOrTypeTraitExprClass: {
2930 const UnaryExprOrTypeTraitExpr *Exp = cast<UnaryExprOrTypeTraitExpr>(E);
2931 if ((Exp->getKind() == UETT_SizeOf) &&
2932 Exp->getTypeOfArgument()->isVariableArrayType())
John McCall864e3962010-05-07 05:32:02 +00002933 return ICEDiag(2, E->getLocStart());
2934 return NoDiag();
2935 }
2936 case Expr::BinaryOperatorClass: {
2937 const BinaryOperator *Exp = cast<BinaryOperator>(E);
2938 switch (Exp->getOpcode()) {
John McCalle3027922010-08-25 11:45:40 +00002939 case BO_PtrMemD:
2940 case BO_PtrMemI:
2941 case BO_Assign:
2942 case BO_MulAssign:
2943 case BO_DivAssign:
2944 case BO_RemAssign:
2945 case BO_AddAssign:
2946 case BO_SubAssign:
2947 case BO_ShlAssign:
2948 case BO_ShrAssign:
2949 case BO_AndAssign:
2950 case BO_XorAssign:
2951 case BO_OrAssign:
John McCall864e3962010-05-07 05:32:02 +00002952 return ICEDiag(2, E->getLocStart());
2953
John McCalle3027922010-08-25 11:45:40 +00002954 case BO_Mul:
2955 case BO_Div:
2956 case BO_Rem:
2957 case BO_Add:
2958 case BO_Sub:
2959 case BO_Shl:
2960 case BO_Shr:
2961 case BO_LT:
2962 case BO_GT:
2963 case BO_LE:
2964 case BO_GE:
2965 case BO_EQ:
2966 case BO_NE:
2967 case BO_And:
2968 case BO_Xor:
2969 case BO_Or:
2970 case BO_Comma: {
John McCall864e3962010-05-07 05:32:02 +00002971 ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2972 ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
John McCalle3027922010-08-25 11:45:40 +00002973 if (Exp->getOpcode() == BO_Div ||
2974 Exp->getOpcode() == BO_Rem) {
John McCall864e3962010-05-07 05:32:02 +00002975 // Evaluate gives an error for undefined Div/Rem, so make sure
2976 // we don't evaluate one.
John McCall4b136332011-02-26 08:27:17 +00002977 if (LHSResult.Val == 0 && RHSResult.Val == 0) {
John McCall864e3962010-05-07 05:32:02 +00002978 llvm::APSInt REval = Exp->getRHS()->EvaluateAsInt(Ctx);
2979 if (REval == 0)
2980 return ICEDiag(1, E->getLocStart());
2981 if (REval.isSigned() && REval.isAllOnesValue()) {
2982 llvm::APSInt LEval = Exp->getLHS()->EvaluateAsInt(Ctx);
2983 if (LEval.isMinSignedValue())
2984 return ICEDiag(1, E->getLocStart());
2985 }
2986 }
2987 }
John McCalle3027922010-08-25 11:45:40 +00002988 if (Exp->getOpcode() == BO_Comma) {
John McCall864e3962010-05-07 05:32:02 +00002989 if (Ctx.getLangOptions().C99) {
2990 // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
2991 // if it isn't evaluated.
2992 if (LHSResult.Val == 0 && RHSResult.Val == 0)
2993 return ICEDiag(1, E->getLocStart());
2994 } else {
2995 // In both C89 and C++, commas in ICEs are illegal.
2996 return ICEDiag(2, E->getLocStart());
2997 }
2998 }
2999 if (LHSResult.Val >= RHSResult.Val)
3000 return LHSResult;
3001 return RHSResult;
3002 }
John McCalle3027922010-08-25 11:45:40 +00003003 case BO_LAnd:
3004 case BO_LOr: {
John McCall864e3962010-05-07 05:32:02 +00003005 ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
Douglas Gregorfcafc6e2011-05-24 16:02:01 +00003006
3007 // C++0x [expr.const]p2:
3008 // [...] subexpressions of logical AND (5.14), logical OR
3009 // (5.15), and condi- tional (5.16) operations that are not
3010 // evaluated are not considered.
3011 if (Ctx.getLangOptions().CPlusPlus0x && LHSResult.Val == 0) {
3012 if (Exp->getOpcode() == BO_LAnd &&
3013 Exp->getLHS()->EvaluateAsInt(Ctx) == 0)
3014 return LHSResult;
3015
3016 if (Exp->getOpcode() == BO_LOr &&
3017 Exp->getLHS()->EvaluateAsInt(Ctx) != 0)
3018 return LHSResult;
3019 }
3020
John McCall864e3962010-05-07 05:32:02 +00003021 ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
3022 if (LHSResult.Val == 0 && RHSResult.Val == 1) {
3023 // Rare case where the RHS has a comma "side-effect"; we need
3024 // to actually check the condition to see whether the side
3025 // with the comma is evaluated.
John McCalle3027922010-08-25 11:45:40 +00003026 if ((Exp->getOpcode() == BO_LAnd) !=
John McCall864e3962010-05-07 05:32:02 +00003027 (Exp->getLHS()->EvaluateAsInt(Ctx) == 0))
3028 return RHSResult;
3029 return NoDiag();
3030 }
3031
3032 if (LHSResult.Val >= RHSResult.Val)
3033 return LHSResult;
3034 return RHSResult;
3035 }
3036 }
3037 }
3038 case Expr::ImplicitCastExprClass:
3039 case Expr::CStyleCastExprClass:
3040 case Expr::CXXFunctionalCastExprClass:
3041 case Expr::CXXStaticCastExprClass:
3042 case Expr::CXXReinterpretCastExprClass:
John McCall31168b02011-06-15 23:02:42 +00003043 case Expr::CXXConstCastExprClass:
3044 case Expr::ObjCBridgedCastExprClass: {
John McCall864e3962010-05-07 05:32:02 +00003045 const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
Douglas Gregorb90df602010-06-16 00:17:44 +00003046 if (SubExpr->getType()->isIntegralOrEnumerationType())
John McCall864e3962010-05-07 05:32:02 +00003047 return CheckICE(SubExpr, Ctx);
3048 if (isa<FloatingLiteral>(SubExpr->IgnoreParens()))
3049 return NoDiag();
3050 return ICEDiag(2, E->getLocStart());
3051 }
John McCallc07a0c72011-02-17 10:25:35 +00003052 case Expr::BinaryConditionalOperatorClass: {
3053 const BinaryConditionalOperator *Exp = cast<BinaryConditionalOperator>(E);
3054 ICEDiag CommonResult = CheckICE(Exp->getCommon(), Ctx);
3055 if (CommonResult.Val == 2) return CommonResult;
3056 ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
3057 if (FalseResult.Val == 2) return FalseResult;
3058 if (CommonResult.Val == 1) return CommonResult;
3059 if (FalseResult.Val == 1 &&
3060 Exp->getCommon()->EvaluateAsInt(Ctx) == 0) return NoDiag();
3061 return FalseResult;
3062 }
John McCall864e3962010-05-07 05:32:02 +00003063 case Expr::ConditionalOperatorClass: {
3064 const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
3065 // If the condition (ignoring parens) is a __builtin_constant_p call,
3066 // then only the true side is actually considered in an integer constant
3067 // expression, and it is fully evaluated. This is an important GNU
3068 // extension. See GCC PR38377 for discussion.
3069 if (const CallExpr *CallCE
3070 = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
3071 if (CallCE->isBuiltinCall(Ctx) == Builtin::BI__builtin_constant_p) {
3072 Expr::EvalResult EVResult;
3073 if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
3074 !EVResult.Val.isInt()) {
3075 return ICEDiag(2, E->getLocStart());
3076 }
3077 return NoDiag();
3078 }
3079 ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
John McCall864e3962010-05-07 05:32:02 +00003080 if (CondResult.Val == 2)
3081 return CondResult;
Douglas Gregorfcafc6e2011-05-24 16:02:01 +00003082
3083 // C++0x [expr.const]p2:
3084 // subexpressions of [...] conditional (5.16) operations that
3085 // are not evaluated are not considered
3086 bool TrueBranch = Ctx.getLangOptions().CPlusPlus0x
3087 ? Exp->getCond()->EvaluateAsInt(Ctx) != 0
3088 : false;
3089 ICEDiag TrueResult = NoDiag();
3090 if (!Ctx.getLangOptions().CPlusPlus0x || TrueBranch)
3091 TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
3092 ICEDiag FalseResult = NoDiag();
3093 if (!Ctx.getLangOptions().CPlusPlus0x || !TrueBranch)
3094 FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
3095
John McCall864e3962010-05-07 05:32:02 +00003096 if (TrueResult.Val == 2)
3097 return TrueResult;
3098 if (FalseResult.Val == 2)
3099 return FalseResult;
3100 if (CondResult.Val == 1)
3101 return CondResult;
3102 if (TrueResult.Val == 0 && FalseResult.Val == 0)
3103 return NoDiag();
3104 // Rare case where the diagnostics depend on which side is evaluated
3105 // Note that if we get here, CondResult is 0, and at least one of
3106 // TrueResult and FalseResult is non-zero.
3107 if (Exp->getCond()->EvaluateAsInt(Ctx) == 0) {
3108 return FalseResult;
3109 }
3110 return TrueResult;
3111 }
3112 case Expr::CXXDefaultArgExprClass:
3113 return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
3114 case Expr::ChooseExprClass: {
3115 return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(Ctx), Ctx);
3116 }
3117 }
3118
3119 // Silence a GCC warning
3120 return ICEDiag(2, E->getLocStart());
3121}
3122
3123bool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
3124 SourceLocation *Loc, bool isEvaluated) const {
3125 ICEDiag d = CheckICE(this, Ctx);
3126 if (d.Val != 0) {
3127 if (Loc) *Loc = d.Loc;
3128 return false;
3129 }
3130 EvalResult EvalResult;
3131 if (!Evaluate(EvalResult, Ctx))
3132 llvm_unreachable("ICE cannot be evaluated!");
3133 assert(!EvalResult.HasSideEffects && "ICE with side effects!");
3134 assert(EvalResult.Val.isInt() && "ICE that isn't integer!");
3135 Result = EvalResult.Val.getInt();
3136 return true;
3137}