blob: 1506f448aa091b0a54f5a34941232f4c948c4b11 [file] [log] [blame]
Chris Lattnerb542afe2008-07-11 19:10:17 +00001//===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
Anders Carlssonc44eec62008-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"
Seo Sanghyeon0fe52e12008-07-08 07:23:12 +000016#include "clang/AST/StmtVisitor.h"
Chris Lattner54176fd2008-07-12 00:14:42 +000017#include "clang/Basic/Diagnostic.h"
Anders Carlsson06a36752008-07-08 05:49:43 +000018#include "clang/Basic/TargetInfo.h"
Anders Carlssonc754aa62008-07-08 05:13:58 +000019#include "llvm/Support/Compiler.h"
Anders Carlssonc44eec62008-07-03 04:20:39 +000020using namespace clang;
Chris Lattnerf5eeb052008-07-11 18:11:29 +000021using llvm::APSInt;
Eli Friedmand8bfe7f2008-08-22 00:06:13 +000022using llvm::APFloat;
Anders Carlssonc44eec62008-07-03 04:20:39 +000023
Chris Lattner87eae5e2008-07-11 22:52:41 +000024/// EvalInfo - This is a private struct used by the evaluator to capture
25/// information about a subexpression as it is folded. It retains information
26/// about the AST context, but also maintains information about the folded
27/// expression.
28///
29/// If an expression could be evaluated, it is still possible it is not a C
30/// "integer constant expression" or constant expression. If not, this struct
31/// captures information about how and why not.
32///
33/// One bit of information passed *into* the request for constant folding
34/// indicates whether the subexpression is "evaluated" or not according to C
35/// rules. For example, the RHS of (0 && foo()) is not evaluated. We can
36/// evaluate the expression regardless of what the RHS is, but C only allows
37/// certain things in certain situations.
38struct EvalInfo {
39 ASTContext &Ctx;
40
41 /// isEvaluated - True if the subexpression is required to be evaluated, false
42 /// if it is short-circuited (according to C rules).
43 bool isEvaluated;
44
Chris Lattner54176fd2008-07-12 00:14:42 +000045 /// ICEDiag - If the expression is unfoldable, then ICEDiag contains the
46 /// error diagnostic indicating why it is not foldable and DiagLoc indicates a
47 /// caret position for the error. If it is foldable, but the expression is
48 /// not an integer constant expression, ICEDiag contains the extension
49 /// diagnostic to emit which describes why it isn't an integer constant
50 /// expression. If this expression *is* an integer-constant-expr, then
51 /// ICEDiag is zero.
Chris Lattner87eae5e2008-07-11 22:52:41 +000052 ///
Chris Lattner54176fd2008-07-12 00:14:42 +000053 /// The caller can choose to emit this diagnostic or not, depending on whether
54 /// they require an i-c-e or a constant or not. DiagLoc indicates the caret
55 /// position for the report.
56 ///
57 /// If ICEDiag is zero, then this expression is an i-c-e.
Chris Lattner87eae5e2008-07-11 22:52:41 +000058 unsigned ICEDiag;
59 SourceLocation DiagLoc;
60
61 EvalInfo(ASTContext &ctx) : Ctx(ctx), isEvaluated(true), ICEDiag(0) {}
62};
63
64
65static bool EvaluatePointer(const Expr *E, APValue &Result, EvalInfo &Info);
66static bool EvaluateInteger(const Expr *E, APSInt &Result, EvalInfo &Info);
Eli Friedmand8bfe7f2008-08-22 00:06:13 +000067static bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
Chris Lattnerf5eeb052008-07-11 18:11:29 +000068
69//===----------------------------------------------------------------------===//
70// Pointer Evaluation
71//===----------------------------------------------------------------------===//
72
Anders Carlssonc754aa62008-07-08 05:13:58 +000073namespace {
Anders Carlsson2bad1682008-07-08 14:30:00 +000074class VISIBILITY_HIDDEN PointerExprEvaluator
75 : public StmtVisitor<PointerExprEvaluator, APValue> {
Chris Lattner87eae5e2008-07-11 22:52:41 +000076 EvalInfo &Info;
Anders Carlsson2bad1682008-07-08 14:30:00 +000077public:
Anders Carlsson2bad1682008-07-08 14:30:00 +000078
Chris Lattner87eae5e2008-07-11 22:52:41 +000079 PointerExprEvaluator(EvalInfo &info) : Info(info) {}
Chris Lattnerf5eeb052008-07-11 18:11:29 +000080
Anders Carlsson2bad1682008-07-08 14:30:00 +000081 APValue VisitStmt(Stmt *S) {
82 // FIXME: Remove this when we support more expressions.
Anders Carlsson650c92f2008-07-08 15:34:11 +000083 printf("Unhandled pointer statement\n");
Anders Carlsson2bad1682008-07-08 14:30:00 +000084 S->dump();
85 return APValue();
86 }
87
88 APValue VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
89
Anders Carlsson650c92f2008-07-08 15:34:11 +000090 APValue VisitBinaryOperator(const BinaryOperator *E);
91 APValue VisitCastExpr(const CastExpr* E);
Anders Carlsson650c92f2008-07-08 15:34:11 +000092};
Chris Lattnerf5eeb052008-07-11 18:11:29 +000093} // end anonymous namespace
Anders Carlsson650c92f2008-07-08 15:34:11 +000094
Chris Lattner87eae5e2008-07-11 22:52:41 +000095static bool EvaluatePointer(const Expr* E, APValue& Result, EvalInfo &Info) {
Chris Lattnerf5eeb052008-07-11 18:11:29 +000096 if (!E->getType()->isPointerType())
97 return false;
Chris Lattner87eae5e2008-07-11 22:52:41 +000098 Result = PointerExprEvaluator(Info).Visit(const_cast<Expr*>(E));
Chris Lattnerf5eeb052008-07-11 18:11:29 +000099 return Result.isLValue();
100}
101
102APValue PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
103 if (E->getOpcode() != BinaryOperator::Add &&
104 E->getOpcode() != BinaryOperator::Sub)
105 return APValue();
106
107 const Expr *PExp = E->getLHS();
108 const Expr *IExp = E->getRHS();
109 if (IExp->getType()->isPointerType())
110 std::swap(PExp, IExp);
111
112 APValue ResultLValue;
Chris Lattner87eae5e2008-07-11 22:52:41 +0000113 if (!EvaluatePointer(PExp, ResultLValue, Info))
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000114 return APValue();
115
116 llvm::APSInt AdditionalOffset(32);
Chris Lattner87eae5e2008-07-11 22:52:41 +0000117 if (!EvaluateInteger(IExp, AdditionalOffset, Info))
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000118 return APValue();
119
120 uint64_t Offset = ResultLValue.getLValueOffset();
121 if (E->getOpcode() == BinaryOperator::Add)
122 Offset += AdditionalOffset.getZExtValue();
123 else
124 Offset -= AdditionalOffset.getZExtValue();
125
126 return APValue(ResultLValue.getLValueBase(), Offset);
127}
128
129
Chris Lattnerb542afe2008-07-11 19:10:17 +0000130APValue PointerExprEvaluator::VisitCastExpr(const CastExpr* E) {
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000131 const Expr* SubExpr = E->getSubExpr();
132
133 // Check for pointer->pointer cast
134 if (SubExpr->getType()->isPointerType()) {
135 APValue Result;
Chris Lattner87eae5e2008-07-11 22:52:41 +0000136 if (EvaluatePointer(SubExpr, Result, Info))
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000137 return Result;
138 return APValue();
139 }
140
Eli Friedmand9f4bcd2008-07-27 05:46:18 +0000141 if (SubExpr->getType()->isIntegralType()) {
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000142 llvm::APSInt Result(32);
Chris Lattner87eae5e2008-07-11 22:52:41 +0000143 if (EvaluateInteger(SubExpr, Result, Info)) {
144 Result.extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000145 return APValue(0, Result.getZExtValue());
146 }
147 }
148
149 assert(0 && "Unhandled cast");
150 return APValue();
151}
152
153
154//===----------------------------------------------------------------------===//
155// Integer Evaluation
156//===----------------------------------------------------------------------===//
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000157
158namespace {
Anders Carlssonc754aa62008-07-08 05:13:58 +0000159class VISIBILITY_HIDDEN IntExprEvaluator
Chris Lattnerb542afe2008-07-11 19:10:17 +0000160 : public StmtVisitor<IntExprEvaluator, bool> {
Chris Lattner87eae5e2008-07-11 22:52:41 +0000161 EvalInfo &Info;
Chris Lattnerb542afe2008-07-11 19:10:17 +0000162 APSInt &Result;
Anders Carlssonc754aa62008-07-08 05:13:58 +0000163public:
Chris Lattner87eae5e2008-07-11 22:52:41 +0000164 IntExprEvaluator(EvalInfo &info, APSInt &result)
165 : Info(info), Result(result) {}
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000166
Chris Lattner7a767782008-07-11 19:24:49 +0000167 unsigned getIntTypeSizeInBits(QualType T) const {
Chris Lattner54176fd2008-07-12 00:14:42 +0000168 return (unsigned)Info.Ctx.getIntWidth(T);
169 }
170
171 bool Extension(SourceLocation L, diag::kind D) {
172 Info.DiagLoc = L;
173 Info.ICEDiag = D;
174 return true; // still a constant.
175 }
176
177 bool Error(SourceLocation L, diag::kind D) {
178 // If this is in an unevaluated portion of the subexpression, ignore the
179 // error.
180 if (!Info.isEvaluated)
181 return true;
182
183 Info.DiagLoc = L;
184 Info.ICEDiag = D;
185 return false;
Chris Lattner7a767782008-07-11 19:24:49 +0000186 }
187
Anders Carlssonc754aa62008-07-08 05:13:58 +0000188 //===--------------------------------------------------------------------===//
189 // Visitor Methods
190 //===--------------------------------------------------------------------===//
Chris Lattner7a767782008-07-11 19:24:49 +0000191
Chris Lattnerb542afe2008-07-11 19:10:17 +0000192 bool VisitStmt(Stmt *S) {
Chris Lattner54176fd2008-07-12 00:14:42 +0000193 return Error(S->getLocStart(), diag::err_expr_not_constant);
Anders Carlssonc754aa62008-07-08 05:13:58 +0000194 }
195
Chris Lattnerb542afe2008-07-11 19:10:17 +0000196 bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
Anders Carlssonc754aa62008-07-08 05:13:58 +0000197
Chris Lattner4c4867e2008-07-12 00:38:25 +0000198 bool VisitIntegerLiteral(const IntegerLiteral *E) {
199 Result = E->getValue();
200 Result.setIsUnsigned(E->getType()->isUnsignedIntegerType());
201 return true;
202 }
203 bool VisitCharacterLiteral(const CharacterLiteral *E) {
204 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
205 Result = E->getValue();
206 Result.setIsUnsigned(E->getType()->isUnsignedIntegerType());
207 return true;
208 }
209 bool VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
210 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
211 Result = Info.Ctx.typesAreCompatible(E->getArgType1(), E->getArgType2());
212 return true;
213 }
214 bool VisitDeclRefExpr(const DeclRefExpr *E);
215 bool VisitCallExpr(const CallExpr *E);
Chris Lattnerb542afe2008-07-11 19:10:17 +0000216 bool VisitBinaryOperator(const BinaryOperator *E);
217 bool VisitUnaryOperator(const UnaryOperator *E);
Anders Carlsson06a36752008-07-08 05:49:43 +0000218
Chris Lattner732b2232008-07-12 01:15:53 +0000219 bool VisitCastExpr(CastExpr* E) {
Chris Lattner732b2232008-07-12 01:15:53 +0000220 return HandleCast(E->getLocStart(), E->getSubExpr(), E->getType());
Anders Carlsson650c92f2008-07-08 15:34:11 +0000221 }
Chris Lattnerfcee0012008-07-11 21:24:13 +0000222 bool VisitSizeOfAlignOfTypeExpr(const SizeOfAlignOfTypeExpr *E) {
Chris Lattner54176fd2008-07-12 00:14:42 +0000223 return EvaluateSizeAlignOf(E->isSizeOf(), E->getArgumentType(),
224 E->getType());
Chris Lattnerfcee0012008-07-11 21:24:13 +0000225 }
Chris Lattner4c4867e2008-07-12 00:38:25 +0000226
Chris Lattnerfcee0012008-07-11 21:24:13 +0000227private:
Chris Lattner732b2232008-07-12 01:15:53 +0000228 bool HandleCast(SourceLocation CastLoc, Expr *SubExpr, QualType DestType);
Chris Lattnerfcee0012008-07-11 21:24:13 +0000229 bool EvaluateSizeAlignOf(bool isSizeOf, QualType SrcTy, QualType DstTy);
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000230};
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000231} // end anonymous namespace
Anders Carlsson650c92f2008-07-08 15:34:11 +0000232
Chris Lattner87eae5e2008-07-11 22:52:41 +0000233static bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
234 return IntExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
Anders Carlsson650c92f2008-07-08 15:34:11 +0000235}
Anders Carlsson650c92f2008-07-08 15:34:11 +0000236
Chris Lattner4c4867e2008-07-12 00:38:25 +0000237bool IntExprEvaluator::VisitDeclRefExpr(const DeclRefExpr *E) {
238 // Enums are integer constant exprs.
239 if (const EnumConstantDecl *D = dyn_cast<EnumConstantDecl>(E->getDecl())) {
240 Result = D->getInitVal();
241 return true;
242 }
243
244 // Otherwise, random variable references are not constants.
245 return Error(E->getLocStart(), diag::err_expr_not_constant);
246}
247
Chris Lattnera4d55d82008-10-06 06:40:35 +0000248/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
249/// as GCC.
250static int EvaluateBuiltinClassifyType(const CallExpr *E) {
251 // The following enum mimics the values returned by GCC.
252 enum gcc_type_class {
253 no_type_class = -1,
254 void_type_class, integer_type_class, char_type_class,
255 enumeral_type_class, boolean_type_class,
256 pointer_type_class, reference_type_class, offset_type_class,
257 real_type_class, complex_type_class,
258 function_type_class, method_type_class,
259 record_type_class, union_type_class,
260 array_type_class, string_type_class,
261 lang_type_class
262 };
263
264 // If no argument was supplied, default to "no_type_class". This isn't
265 // ideal, however it is what gcc does.
266 if (E->getNumArgs() == 0)
267 return no_type_class;
268
269 QualType ArgTy = E->getArg(0)->getType();
270 if (ArgTy->isVoidType())
271 return void_type_class;
272 else if (ArgTy->isEnumeralType())
273 return enumeral_type_class;
274 else if (ArgTy->isBooleanType())
275 return boolean_type_class;
276 else if (ArgTy->isCharType())
277 return string_type_class; // gcc doesn't appear to use char_type_class
278 else if (ArgTy->isIntegerType())
279 return integer_type_class;
280 else if (ArgTy->isPointerType())
281 return pointer_type_class;
282 else if (ArgTy->isReferenceType())
283 return reference_type_class;
284 else if (ArgTy->isRealType())
285 return real_type_class;
286 else if (ArgTy->isComplexType())
287 return complex_type_class;
288 else if (ArgTy->isFunctionType())
289 return function_type_class;
290 else if (ArgTy->isStructureType())
291 return record_type_class;
292 else if (ArgTy->isUnionType())
293 return union_type_class;
294 else if (ArgTy->isArrayType())
295 return array_type_class;
296 else if (ArgTy->isUnionType())
297 return union_type_class;
298 else // FIXME: offset_type_class, method_type_class, & lang_type_class?
299 assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
300 return -1;
301}
302
Chris Lattner4c4867e2008-07-12 00:38:25 +0000303bool IntExprEvaluator::VisitCallExpr(const CallExpr *E) {
304 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
Chris Lattner4c4867e2008-07-12 00:38:25 +0000305
Chris Lattner019f4e82008-10-06 05:28:25 +0000306 switch (E->isBuiltinCall()) {
307 default:
308 return Error(E->getLocStart(), diag::err_expr_not_constant);
309 case Builtin::BI__builtin_classify_type:
Chris Lattnera4d55d82008-10-06 06:40:35 +0000310 Result.setIsSigned(true);
311 Result = EvaluateBuiltinClassifyType(E);
Chris Lattner019f4e82008-10-06 05:28:25 +0000312 return true;
313
314 case Builtin::BI__builtin_constant_p: {
315 // __builtin_constant_p always has one operand: it returns true if that
316 // operand can be folded, false otherwise.
317 APValue Res;
318 Result = E->getArg(0)->tryEvaluate(Res, Info.Ctx);
319 return true;
320 }
321 }
Chris Lattner4c4867e2008-07-12 00:38:25 +0000322}
Anders Carlsson650c92f2008-07-08 15:34:11 +0000323
Chris Lattnerb542afe2008-07-11 19:10:17 +0000324bool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000325 // The LHS of a constant expr is always evaluated and needed.
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000326 llvm::APSInt RHS(32);
Chris Lattner54176fd2008-07-12 00:14:42 +0000327 if (!Visit(E->getLHS()))
328 return false; // error in subexpression.
329
330 bool OldEval = Info.isEvaluated;
331
332 // The short-circuiting &&/|| operators don't necessarily evaluate their
333 // RHS. Make sure to pass isEvaluated down correctly.
334 if ((E->getOpcode() == BinaryOperator::LAnd && Result == 0) ||
335 (E->getOpcode() == BinaryOperator::LOr && Result != 0))
336 Info.isEvaluated = false;
Eli Friedmand9f4bcd2008-07-27 05:46:18 +0000337
338 // FIXME: Handle pointer subtraction
339
340 // FIXME Maybe we want to succeed even where we can't evaluate the
341 // right side of LAnd/LOr?
342 // For example, see http://llvm.org/bugs/show_bug.cgi?id=2525
Chris Lattner54176fd2008-07-12 00:14:42 +0000343 if (!EvaluateInteger(E->getRHS(), RHS, Info))
Chris Lattnerb542afe2008-07-11 19:10:17 +0000344 return false;
Chris Lattner54176fd2008-07-12 00:14:42 +0000345 Info.isEvaluated = OldEval;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000346
347 switch (E->getOpcode()) {
Chris Lattner54176fd2008-07-12 00:14:42 +0000348 default: return Error(E->getOperatorLoc(), diag::err_expr_not_constant);
349 case BinaryOperator::Mul: Result *= RHS; return true;
350 case BinaryOperator::Add: Result += RHS; return true;
351 case BinaryOperator::Sub: Result -= RHS; return true;
352 case BinaryOperator::And: Result &= RHS; return true;
353 case BinaryOperator::Xor: Result ^= RHS; return true;
354 case BinaryOperator::Or: Result |= RHS; return true;
Chris Lattner75a48812008-07-11 22:15:16 +0000355 case BinaryOperator::Div:
Chris Lattner54176fd2008-07-12 00:14:42 +0000356 if (RHS == 0)
357 return Error(E->getOperatorLoc(), diag::err_expr_divide_by_zero);
Chris Lattner75a48812008-07-11 22:15:16 +0000358 Result /= RHS;
Chris Lattner54176fd2008-07-12 00:14:42 +0000359 return true;
Chris Lattner75a48812008-07-11 22:15:16 +0000360 case BinaryOperator::Rem:
Chris Lattner54176fd2008-07-12 00:14:42 +0000361 if (RHS == 0)
362 return Error(E->getOperatorLoc(), diag::err_expr_divide_by_zero);
Chris Lattner75a48812008-07-11 22:15:16 +0000363 Result %= RHS;
Chris Lattner54176fd2008-07-12 00:14:42 +0000364 return true;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000365 case BinaryOperator::Shl:
Chris Lattner54176fd2008-07-12 00:14:42 +0000366 // FIXME: Warn about out of range shift amounts!
Chris Lattnerb542afe2008-07-11 19:10:17 +0000367 Result <<= (unsigned)RHS.getLimitedValue(Result.getBitWidth()-1);
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000368 break;
369 case BinaryOperator::Shr:
Chris Lattnerb542afe2008-07-11 19:10:17 +0000370 Result >>= (unsigned)RHS.getLimitedValue(Result.getBitWidth()-1);
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000371 break;
Chris Lattnerb542afe2008-07-11 19:10:17 +0000372
Chris Lattnerac7cb602008-07-11 19:29:32 +0000373 case BinaryOperator::LT:
374 Result = Result < RHS;
375 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
376 break;
377 case BinaryOperator::GT:
378 Result = Result > RHS;
379 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
380 break;
381 case BinaryOperator::LE:
382 Result = Result <= RHS;
383 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
384 break;
385 case BinaryOperator::GE:
386 Result = Result >= RHS;
387 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
388 break;
389 case BinaryOperator::EQ:
390 Result = Result == RHS;
391 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
392 break;
393 case BinaryOperator::NE:
394 Result = Result != RHS;
395 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
396 break;
Chris Lattner54176fd2008-07-12 00:14:42 +0000397 case BinaryOperator::LAnd:
398 Result = Result != 0 && RHS != 0;
399 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
400 break;
401 case BinaryOperator::LOr:
402 Result = Result != 0 || RHS != 0;
403 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
404 break;
405
Anders Carlsson06a36752008-07-08 05:49:43 +0000406
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000407 case BinaryOperator::Comma:
Chris Lattner54176fd2008-07-12 00:14:42 +0000408 // Result of the comma is just the result of the RHS.
409 Result = RHS;
410
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000411 // C99 6.6p3: "shall not contain assignment, ..., or comma operators,
412 // *except* when they are contained within a subexpression that is not
413 // evaluated". Note that Assignment can never happen due to constraints
414 // on the LHS subexpr, so we don't need to check it here.
Chris Lattner54176fd2008-07-12 00:14:42 +0000415 if (!Info.isEvaluated)
416 return true;
417
418 // If the value is evaluated, we can accept it as an extension.
419 return Extension(E->getOperatorLoc(), diag::ext_comma_in_constant_expr);
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000420 }
421
422 Result.setIsUnsigned(E->getType()->isUnsignedIntegerType());
Chris Lattnerb542afe2008-07-11 19:10:17 +0000423 return true;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000424}
425
Chris Lattnerfcee0012008-07-11 21:24:13 +0000426/// EvaluateSizeAlignOf - Evaluate sizeof(SrcTy) or alignof(SrcTy) with a result
427/// as a DstTy type.
428bool IntExprEvaluator::EvaluateSizeAlignOf(bool isSizeOf, QualType SrcTy,
429 QualType DstTy) {
430 // Return the result in the right width.
431 Result.zextOrTrunc(getIntTypeSizeInBits(DstTy));
432 Result.setIsUnsigned(DstTy->isUnsignedIntegerType());
433
434 // sizeof(void) and __alignof__(void) = 1 as a gcc extension.
435 if (SrcTy->isVoidType())
436 Result = 1;
437
438 // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
439 if (!SrcTy->isConstantSizeType()) {
440 // FIXME: Should we attempt to evaluate this?
441 return false;
442 }
443
444 // GCC extension: sizeof(function) = 1.
445 if (SrcTy->isFunctionType()) {
446 // FIXME: AlignOf shouldn't be unconditionally 4!
447 Result = isSizeOf ? 1 : 4;
448 return true;
449 }
450
451 // Get information about the size or align.
Chris Lattner87eae5e2008-07-11 22:52:41 +0000452 unsigned CharSize = Info.Ctx.Target.getCharWidth();
Chris Lattnerfcee0012008-07-11 21:24:13 +0000453 if (isSizeOf)
454 Result = getIntTypeSizeInBits(SrcTy) / CharSize;
455 else
Chris Lattner87eae5e2008-07-11 22:52:41 +0000456 Result = Info.Ctx.getTypeAlign(SrcTy) / CharSize;
Chris Lattnerfcee0012008-07-11 21:24:13 +0000457 return true;
458}
459
Chris Lattnerb542afe2008-07-11 19:10:17 +0000460bool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
Chris Lattner4c4867e2008-07-12 00:38:25 +0000461 // Special case unary operators that do not need their subexpression
462 // evaluated. offsetof/sizeof/alignof are all special.
Chris Lattner75a48812008-07-11 22:15:16 +0000463 if (E->isOffsetOfOp()) {
Chris Lattner4c4867e2008-07-12 00:38:25 +0000464 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
Chris Lattner87eae5e2008-07-11 22:52:41 +0000465 Result = E->evaluateOffsetOf(Info.Ctx);
Chris Lattner75a48812008-07-11 22:15:16 +0000466 Result.setIsUnsigned(E->getType()->isUnsignedIntegerType());
467 return true;
468 }
469
470 if (E->isSizeOfAlignOfOp())
Chris Lattnerfcee0012008-07-11 21:24:13 +0000471 return EvaluateSizeAlignOf(E->getOpcode() == UnaryOperator::SizeOf,
472 E->getSubExpr()->getType(), E->getType());
Chris Lattner75a48812008-07-11 22:15:16 +0000473
Chris Lattner87eae5e2008-07-11 22:52:41 +0000474 // Get the operand value into 'Result'.
475 if (!Visit(E->getSubExpr()))
Chris Lattner75a48812008-07-11 22:15:16 +0000476 return false;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000477
Chris Lattner75a48812008-07-11 22:15:16 +0000478 switch (E->getOpcode()) {
Chris Lattner4c4867e2008-07-12 00:38:25 +0000479 default:
Chris Lattner75a48812008-07-11 22:15:16 +0000480 // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
481 // See C99 6.6p3.
Chris Lattner4c4867e2008-07-12 00:38:25 +0000482 return Error(E->getOperatorLoc(), diag::err_expr_not_constant);
Chris Lattner75a48812008-07-11 22:15:16 +0000483 case UnaryOperator::LNot: {
484 bool Val = Result == 0;
485 Result.zextOrTrunc(getIntTypeSizeInBits(E->getType()));
486 Result = Val;
487 break;
488 }
489 case UnaryOperator::Extension:
Chris Lattner4c4867e2008-07-12 00:38:25 +0000490 // FIXME: Should extension allow i-c-e extension expressions in its scope?
491 // If so, we could clear the diagnostic ID.
Chris Lattner75a48812008-07-11 22:15:16 +0000492 case UnaryOperator::Plus:
Chris Lattner4c4867e2008-07-12 00:38:25 +0000493 // The result is always just the subexpr.
Chris Lattner75a48812008-07-11 22:15:16 +0000494 break;
495 case UnaryOperator::Minus:
496 Result = -Result;
497 break;
498 case UnaryOperator::Not:
499 Result = ~Result;
500 break;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000501 }
502
503 Result.setIsUnsigned(E->getType()->isUnsignedIntegerType());
Chris Lattnerb542afe2008-07-11 19:10:17 +0000504 return true;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000505}
506
Chris Lattner732b2232008-07-12 01:15:53 +0000507/// HandleCast - This is used to evaluate implicit or explicit casts where the
508/// result type is integer.
509bool IntExprEvaluator::HandleCast(SourceLocation CastLoc,
510 Expr *SubExpr, QualType DestType) {
Chris Lattner7a767782008-07-11 19:24:49 +0000511 unsigned DestWidth = getIntTypeSizeInBits(DestType);
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000512
513 // Handle simple integer->integer casts.
514 if (SubExpr->getType()->isIntegerType()) {
Chris Lattner732b2232008-07-12 01:15:53 +0000515 if (!Visit(SubExpr))
Chris Lattnerb542afe2008-07-11 19:10:17 +0000516 return false;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000517
518 // Figure out if this is a truncate, extend or noop cast.
519 // If the input is signed, do a sign extend, noop, or truncate.
520 if (DestType->isBooleanType()) {
521 // Conversion to bool compares against zero.
522 Result = Result != 0;
523 Result.zextOrTrunc(DestWidth);
Chris Lattner7a767782008-07-11 19:24:49 +0000524 } else
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000525 Result.extOrTrunc(DestWidth);
Chris Lattner732b2232008-07-12 01:15:53 +0000526 Result.setIsUnsigned(DestType->isUnsignedIntegerType());
527 return true;
528 }
529
530 // FIXME: Clean this up!
531 if (SubExpr->getType()->isPointerType()) {
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000532 APValue LV;
Chris Lattner87eae5e2008-07-11 22:52:41 +0000533 if (!EvaluatePointer(SubExpr, LV, Info))
Chris Lattnerb542afe2008-07-11 19:10:17 +0000534 return false;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000535 if (LV.getLValueBase())
Chris Lattnerb542afe2008-07-11 19:10:17 +0000536 return false;
Anders Carlsson06a36752008-07-08 05:49:43 +0000537
Anders Carlsson559e56b2008-07-08 16:49:00 +0000538 Result.extOrTrunc(DestWidth);
539 Result = LV.getLValueOffset();
Chris Lattner732b2232008-07-12 01:15:53 +0000540 Result.setIsUnsigned(DestType->isUnsignedIntegerType());
541 return true;
Anders Carlsson2bad1682008-07-08 14:30:00 +0000542 }
543
Chris Lattner732b2232008-07-12 01:15:53 +0000544 if (!SubExpr->getType()->isRealFloatingType())
545 return Error(CastLoc, diag::err_expr_not_constant);
546
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000547 APFloat F(0.0);
548 if (!EvaluateFloat(SubExpr, F, Info))
Chris Lattner732b2232008-07-12 01:15:53 +0000549 return Error(CastLoc, diag::err_expr_not_constant);
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000550
Chris Lattner732b2232008-07-12 01:15:53 +0000551 // If the destination is boolean, compare against zero.
552 if (DestType->isBooleanType()) {
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000553 Result = !F.isZero();
Chris Lattner732b2232008-07-12 01:15:53 +0000554 Result.zextOrTrunc(DestWidth);
555 Result.setIsUnsigned(DestType->isUnsignedIntegerType());
556 return true;
557 }
558
559 // Determine whether we are converting to unsigned or signed.
560 bool DestSigned = DestType->isSignedIntegerType();
561
562 // FIXME: Warning for overflow.
563 uint64_t Space[4];
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000564 (void)F.convertToInteger(Space, DestWidth, DestSigned,
565 llvm::APFloat::rmTowardZero);
Chris Lattner732b2232008-07-12 01:15:53 +0000566 Result = llvm::APInt(DestWidth, 4, Space);
567 Result.setIsUnsigned(!DestSigned);
Chris Lattnerb542afe2008-07-11 19:10:17 +0000568 return true;
Anders Carlssona25ae3d2008-07-08 14:35:21 +0000569}
Anders Carlsson2bad1682008-07-08 14:30:00 +0000570
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000571//===----------------------------------------------------------------------===//
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000572// Float Evaluation
573//===----------------------------------------------------------------------===//
574
575namespace {
576class VISIBILITY_HIDDEN FloatExprEvaluator
577 : public StmtVisitor<FloatExprEvaluator, bool> {
578 EvalInfo &Info;
579 APFloat &Result;
580public:
581 FloatExprEvaluator(EvalInfo &info, APFloat &result)
582 : Info(info), Result(result) {}
583
584 bool VisitStmt(Stmt *S) {
585 return false;
586 }
587
588 bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
Chris Lattner019f4e82008-10-06 05:28:25 +0000589 bool VisitCallExpr(const CallExpr *E);
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000590
591 bool VisitBinaryOperator(const BinaryOperator *E);
592 bool VisitFloatingLiteral(const FloatingLiteral *E);
593};
594} // end anonymous namespace
595
596static bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
597 return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
598}
599
Chris Lattner019f4e82008-10-06 05:28:25 +0000600bool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
Chris Lattner34a74ab2008-10-06 05:53:16 +0000601 const llvm::fltSemantics &Sem =
602 Info.Ctx.getFloatTypeSemantics(E->getType());
Chris Lattner019f4e82008-10-06 05:28:25 +0000603
604 switch (E->isBuiltinCall()) {
Chris Lattner34a74ab2008-10-06 05:53:16 +0000605 default: return false;
Chris Lattner019f4e82008-10-06 05:28:25 +0000606 case Builtin::BI__builtin_huge_val:
607 case Builtin::BI__builtin_huge_valf:
608 case Builtin::BI__builtin_huge_vall:
609 case Builtin::BI__builtin_inf:
610 case Builtin::BI__builtin_inff:
611 case Builtin::BI__builtin_infl:
Chris Lattner34a74ab2008-10-06 05:53:16 +0000612 Result = llvm::APFloat::getInf(Sem);
613 return true;
Chris Lattner9e621712008-10-06 06:31:58 +0000614
615 case Builtin::BI__builtin_nan:
616 case Builtin::BI__builtin_nanf:
617 case Builtin::BI__builtin_nanl:
618 // If this is __builtin_nan("") turn this into a simple nan, otherwise we
619 // can't constant fold it.
620 if (const StringLiteral *S =
621 dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenCasts())) {
622 if (!S->isWide() && S->getByteLength() == 0) { // empty string.
623 Result = llvm::APFloat::getNaN(Sem);
624 return true;
625 }
626 }
627 return false;
Chris Lattner019f4e82008-10-06 05:28:25 +0000628 }
629}
630
631
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000632bool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
633 // FIXME: Diagnostics? I really don't understand how the warnings
634 // and errors are supposed to work.
635 APFloat LHS(0.0), RHS(0.0);
636 if (!EvaluateFloat(E->getLHS(), Result, Info))
637 return false;
638 if (!EvaluateFloat(E->getRHS(), RHS, Info))
639 return false;
640
641 switch (E->getOpcode()) {
642 default: return false;
643 case BinaryOperator::Mul:
644 Result.multiply(RHS, APFloat::rmNearestTiesToEven);
645 return true;
646 case BinaryOperator::Add:
647 Result.add(RHS, APFloat::rmNearestTiesToEven);
648 return true;
649 case BinaryOperator::Sub:
650 Result.subtract(RHS, APFloat::rmNearestTiesToEven);
651 return true;
652 case BinaryOperator::Div:
653 Result.divide(RHS, APFloat::rmNearestTiesToEven);
654 return true;
655 case BinaryOperator::Rem:
656 Result.mod(RHS, APFloat::rmNearestTiesToEven);
657 return true;
658 }
659}
660
661bool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
662 Result = E->getValue();
663 return true;
664}
665
666//===----------------------------------------------------------------------===//
Chris Lattnerf5eeb052008-07-11 18:11:29 +0000667// Top level TryEvaluate.
668//===----------------------------------------------------------------------===//
669
Chris Lattner019f4e82008-10-06 05:28:25 +0000670/// tryEvaluate - Return true if this is a constant which we can fold using
671/// any crazy technique (that has nothing to do with language standards) that
672/// we want to. If this function returns true, it returns the folded constant
673/// in Result.
Chris Lattnerb542afe2008-07-11 19:10:17 +0000674bool Expr::tryEvaluate(APValue &Result, ASTContext &Ctx) const {
Chris Lattner87eae5e2008-07-11 22:52:41 +0000675 EvalInfo Info(Ctx);
Anders Carlsson06a36752008-07-08 05:49:43 +0000676 if (getType()->isIntegerType()) {
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000677 llvm::APSInt sInt(32);
Chris Lattner87eae5e2008-07-11 22:52:41 +0000678 if (EvaluateInteger(this, sInt, Info)) {
Anders Carlsson06a36752008-07-08 05:49:43 +0000679 Result = APValue(sInt);
680 return true;
681 }
Eli Friedmand8bfe7f2008-08-22 00:06:13 +0000682 } else if (getType()->isPointerType()) {
683 if (EvaluatePointer(this, Result, Info)) {
684 return true;
685 }
686 } else if (getType()->isRealFloatingType()) {
687 llvm::APFloat f(0.0);
688 if (EvaluateFloat(this, f, Info)) {
689 Result = APValue(f);
690 return true;
691 }
692 }
Anders Carlsson165a70f2008-08-10 17:03:01 +0000693
Anders Carlssonc44eec62008-07-03 04:20:39 +0000694 return false;
695}