ExprConstant.cpp revision ed8abf18329df67b0abcbb3a10458bd8c1d2a595
1b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner//===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
2c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
3c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//                     The LLVM Compiler Infrastructure
4c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
5c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file is distributed under the University of Illinois Open Source
6c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// License. See LICENSE.TXT for details.
7c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
8c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
9c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
10c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file implements the Expr constant evaluator.
11c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
12c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
13c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
14c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/APValue.h"
15c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/ASTContext.h"
16199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck#include "clang/AST/CharUnits.h"
1719cc4abea06a9b49e0e16a50d335c064cd723572Anders Carlsson#include "clang/AST/RecordLayout.h"
180fe52e1bcaa69ba127f1bda036f057fec1f478deSeo Sanghyeon#include "clang/AST/StmtVisitor.h"
198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/TypeLoc.h"
20500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/AST/ASTDiagnostic.h"
218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/Expr.h"
221b63e4f732dbc73d90abf886b4d21f8e3a165f6dChris Lattner#include "clang/Basic/Builtins.h"
2306a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson#include "clang/Basic/TargetInfo.h"
247462b39a9bccaf4392687831036713f09f9c0681Mike Stump#include "llvm/ADT/SmallString.h"
254572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump#include <cstring>
264572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump
27c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlssonusing namespace clang;
28f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerusing llvm::APSInt;
29d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanusing llvm::APFloat;
30c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
3187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// EvalInfo - This is a private struct used by the evaluator to capture
3287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// information about a subexpression as it is folded.  It retains information
3387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// about the AST context, but also maintains information about the folded
3487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// expression.
3587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
3687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// If an expression could be evaluated, it is still possible it is not a C
3787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// "integer constant expression" or constant expression.  If not, this struct
3887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// captures information about how and why not.
3987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
4087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// One bit of information passed *into* the request for constant folding
4187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// indicates whether the subexpression is "evaluated" or not according to C
4287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// rules.  For example, the RHS of (0 && foo()) is not evaluated.  We can
4387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// evaluate the expression regardless of what the RHS is, but C only allows
4487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// certain things in certain situations.
4587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstruct EvalInfo {
4687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  ASTContext &Ctx;
471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4854da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  /// EvalResult - Contains information about the evaluation.
4954da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  Expr::EvalResult &EvalResult;
50f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson
5142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo(ASTContext &ctx, Expr::EvalResult& evalresult)
5242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    : Ctx(ctx), EvalResult(evalresult) {}
5387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner};
5487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
55f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallnamespace {
56f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  struct ComplexValue {
57f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  private:
58f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool IsInt;
59f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
60f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  public:
61f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt IntReal, IntImag;
62f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat FloatReal, FloatImag;
63f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
64f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue() : FloatReal(APFloat::Bogus), FloatImag(APFloat::Bogus) {}
65f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
66f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexFloat() { IsInt = false; }
67f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexFloat() const { return !IsInt; }
68f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatReal() { return FloatReal; }
69f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatImag() { return FloatImag; }
70f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
71f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexInt() { IsInt = true; }
72f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexInt() const { return IsInt; }
73f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntReal() { return IntReal; }
74f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntImag() { return IntImag; }
75f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
76f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void moveInto(APValue &v) {
77f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (isComplexFloat())
78f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(FloatReal, FloatImag);
79f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      else
80f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(IntReal, IntImag);
81f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    }
82f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  };
83efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
84efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  struct LValue {
85efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *Base;
86efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits Offset;
87efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
88efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *getLValueBase() { return Base; }
89efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits getLValueOffset() { return Offset; }
90efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
91efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    void moveInto(APValue &v) {
92efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      v = APValue(Base, Offset);
93efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    }
94efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  };
95f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall}
9687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
97efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info);
98efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr *E, LValue &Result, EvalInfo &Info);
9987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluateInteger(const Expr *E, APSInt  &Result, EvalInfo &Info);
100d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattnerstatic bool EvaluateIntegerOrLValue(const Expr *E, APValue  &Result,
101d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattner                                    EvalInfo &Info);
102d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
103f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallstatic bool EvaluateComplex(const Expr *E, ComplexValue &Res, EvalInfo &Info);
104f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
105f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// Misc utilities
1074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
1084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
109e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarastatic bool IsGlobalLValue(const Expr* E) {
11042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!E) return true;
11142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
11242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
11342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (isa<FunctionDecl>(DRE->getDecl()))
11442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
11542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
11642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->hasGlobalStorage();
11742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
11842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
11942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const CompoundLiteralExpr *CLE = dyn_cast<CompoundLiteralExpr>(E))
12142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return CLE->isFileScope();
12242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return true;
12442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
12542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
126efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvalPointerValueAsBool(LValue& Value, bool& Result) {
127efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  const Expr* Base = Value.Base;
128a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1293554283157190e67918fad4221a5e6faf9317362John McCall  // A null base expression indicates a null pointer.  These are always
1303554283157190e67918fad4221a5e6faf9317362John McCall  // evaluatable, and they are false unless the offset is zero.
1313554283157190e67918fad4221a5e6faf9317362John McCall  if (!Base) {
1323554283157190e67918fad4221a5e6faf9317362John McCall    Result = !Value.Offset.isZero();
1333554283157190e67918fad4221a5e6faf9317362John McCall    return true;
1343554283157190e67918fad4221a5e6faf9317362John McCall  }
1353554283157190e67918fad4221a5e6faf9317362John McCall
13642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // Require the base expression to be a global l-value.
137e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (!IsGlobalLValue(Base)) return false;
13842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1393554283157190e67918fad4221a5e6faf9317362John McCall  // We have a non-null base expression.  These are generally known to
1403554283157190e67918fad4221a5e6faf9317362John McCall  // be true, but if it'a decl-ref to a weak symbol it can be null at
1413554283157190e67918fad4221a5e6faf9317362John McCall  // runtime.
1423554283157190e67918fad4221a5e6faf9317362John McCall  Result = true;
143a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1443554283157190e67918fad4221a5e6faf9317362John McCall  const DeclRefExpr* DeclRef = dyn_cast<DeclRefExpr>(Base);
145a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (!DeclRef)
146a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return true;
147a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1483554283157190e67918fad4221a5e6faf9317362John McCall  // If it's a weak symbol, it isn't constant-evaluable.
149a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  const ValueDecl* Decl = DeclRef->getDecl();
150a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (Decl->hasAttr<WeakAttr>() ||
151a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakRefAttr>() ||
152a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakImportAttr>())
153a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return false;
154a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1555bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman  return true;
1565bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman}
1575bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman
158cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallstatic bool HandleConversionToBool(const Expr* E, bool& Result,
159cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall                                   EvalInfo &Info) {
1602ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (E->getType()->isIntegralOrEnumerationType()) {
1614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
1624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateInteger(E, IntResult, Info))
1634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = IntResult != 0;
1654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
1664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else if (E->getType()->isRealFloatingType()) {
1674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APFloat FloatResult(0.0);
1684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateFloat(E, FloatResult, Info))
1694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = !FloatResult.isZero();
1714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
172a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->hasPointerRepresentation()) {
173efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue PointerResult;
1744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluatePointer(E, PointerResult, Info))
1754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1765bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman    return EvalPointerValueAsBool(PointerResult, Result);
177a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->isAnyComplexType()) {
178f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue ComplexResult;
179a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (!EvaluateComplex(E, ComplexResult, Info))
180a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      return false;
181a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (ComplexResult.isComplexFloat()) {
182a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = !ComplexResult.getComplexFloatReal().isZero() ||
183a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               !ComplexResult.getComplexFloatImag().isZero();
184a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    } else {
185a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = ComplexResult.getComplexIntReal().getBoolValue() ||
186a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               ComplexResult.getComplexIntImag().getBoolValue();
187a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    }
188a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    return true;
1894efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
1904efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
1924efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
1934efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleFloatToIntCast(QualType DestType, QualType SrcType,
195a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                   APFloat &Value, ASTContext &Ctx) {
196a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
197a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Determine whether we are converting to unsigned or signed.
198a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool DestSigned = DestType->isSignedIntegerType();
1991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
200a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // FIXME: Warning for overflow.
201a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  uint64_t Space[4];
202a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
203a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  (void)Value.convertToInteger(Space, DestWidth, DestSigned,
204a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                               llvm::APFloat::rmTowardZero, &ignored);
205a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return APSInt(llvm::APInt(DestWidth, 4, Space), !DestSigned);
206a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
207a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleFloatToFloatCast(QualType DestType, QualType SrcType,
209a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                      APFloat &Value, ASTContext &Ctx) {
210a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
211a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result = Value;
2121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Result.convert(Ctx.getFloatTypeSemantics(DestType),
213a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                 APFloat::rmNearestTiesToEven, &ignored);
214a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
215a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
216a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleIntToIntCast(QualType DestType, QualType SrcType,
218a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                 APSInt &Value, ASTContext &Ctx) {
219a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
220a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APSInt Result = Value;
221a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Figure out if this is a truncate, extend or noop cast.
222a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // If the input is signed, do a sign extend, noop, or truncate.
223a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.extOrTrunc(DestWidth);
224a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.setIsUnsigned(DestType->isUnsignedIntegerType());
225a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
226a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
227a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleIntToFloatCast(QualType DestType, QualType SrcType,
229a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    APSInt &Value, ASTContext &Ctx) {
230a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
231a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result(Ctx.getFloatTypeSemantics(DestType), 1);
232a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.convertFromAPInt(Value, Value.isSigned(),
233a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                          APFloat::rmNearestTiesToEven);
234a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
235a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
236a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
237c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumpnamespace {
238770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass HasSideEffect
239c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  : public StmtVisitor<HasSideEffect, bool> {
240c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  EvalInfo &Info;
241c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumppublic:
242c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
243c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  HasSideEffect(EvalInfo &info) : Info(info) {}
244c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
245c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // Unhandled nodes conservatively default to having side effects.
246c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStmt(Stmt *S) {
247c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return true;
248c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
249c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
250c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
251c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitDeclRefExpr(DeclRefExpr *E) {
252df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
253c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
254c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return false;
255c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
256c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // We don't want to evaluate BlockExprs multiple times, as they generate
257c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // a ton of code.
258c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBlockExpr(BlockExpr *E) { return true; }
259c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitPredefinedExpr(PredefinedExpr *E) { return false; }
260c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E)
261c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getInitializer()); }
262c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitMemberExpr(MemberExpr *E) { return Visit(E->getBase()); }
263c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitIntegerLiteral(IntegerLiteral *E) { return false; }
264c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitFloatingLiteral(FloatingLiteral *E) { return false; }
265c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStringLiteral(StringLiteral *E) { return false; }
266c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCharacterLiteral(CharacterLiteral *E) { return false; }
267c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) { return false; }
268c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E)
269980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    { return Visit(E->getLHS()) || Visit(E->getRHS()); }
270c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitChooseExpr(ChooseExpr *E)
271c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
272c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCastExpr(CastExpr *E) { return Visit(E->getSubExpr()); }
273c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBinAssign(BinaryOperator *E) { return true; }
2743f0147e161df4725ff15fbb731f4f727afcc229fMike Stump  bool VisitCompoundAssignOperator(BinaryOperator *E) { return true; }
275980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  bool VisitBinaryOperator(BinaryOperator *E)
276980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  { return Visit(E->getLHS()) || Visit(E->getRHS()); }
277c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreInc(UnaryOperator *E) { return true; }
278c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostInc(UnaryOperator *E) { return true; }
279c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreDec(UnaryOperator *E) { return true; }
280c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostDec(UnaryOperator *E) { return true; }
281c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryDeref(UnaryOperator *E) {
282df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
283c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
284980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    return Visit(E->getSubExpr());
285c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
286c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryOperator(UnaryOperator *E) { return Visit(E->getSubExpr()); }
287363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner
288363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  // Has side effects if any element does.
289363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  bool VisitInitListExpr(InitListExpr *E) {
290363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    for (unsigned i = 0, e = E->getNumInits(); i != e; ++i)
291363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner      if (Visit(E->getInit(i))) return true;
292363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    return false;
293363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  }
294c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump};
295c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
296c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump} // end anonymous namespace
297c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
2984efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
2994efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// LValue Evaluation
3004efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
3014efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmannamespace {
302770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass LValueExprEvaluator
303efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<LValueExprEvaluator, bool> {
3044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  EvalInfo &Info;
305efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
306efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
307efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
308efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
309efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
310efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
311efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
3124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanpublic:
3131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
314efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValueExprEvaluator(EvalInfo &info, LValue &Result) :
315efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info(info), Result(Result) {}
3164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
317efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
318efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
321efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
322efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitDeclRefExpr(DeclRefExpr *E);
323efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitPredefinedExpr(PredefinedExpr *E) { return Success(E); }
324efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
325efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitMemberExpr(MemberExpr *E);
326efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStringLiteral(StringLiteral *E) { return Success(E); }
327efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCEncodeExpr(ObjCEncodeExpr *E) { return Success(E); }
328efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E);
329efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryDeref(UnaryOperator *E);
330efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
331ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
332efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(const ChooseExpr *E)
333ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
33426bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
335efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr *E) {
33626bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    switch (E->getCastKind()) {
33726bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    default:
338efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
33926bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
34026bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    case CastExpr::CK_NoOp:
34126bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson      return Visit(E->getSubExpr());
34226bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    }
34326bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson  }
344ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: __real__, __imag__
3454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman};
3464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman} // end anonymous namespace
3474efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
348efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr* E, LValue& Result, EvalInfo &Info) {
349efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return LValueExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
3504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
352efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitDeclRefExpr(DeclRefExpr *E) {
35350c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  if (isa<FunctionDecl>(E->getDecl())) {
354efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
35550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  } else if (VarDecl* VD = dyn_cast<VarDecl>(E->getDecl())) {
35650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman    if (!VD->getType()->isReferenceType())
357efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
358761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // Reference parameters can refer to anything even if they have an
359761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // "initializer" in the form of a default argument.
360761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    if (isa<ParmVarDecl>(VD))
361761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth      return false;
362d933a0198f3ccce9c73bf2951625315b911d37bfEli Friedman    // FIXME: Check whether VD might be overridden!
36331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (const Expr *Init = VD->getAnyInitializer())
364cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor      return Visit(const_cast<Expr *>(Init));
36550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  }
36650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman
367efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
36835873c49adad211ff466e34342a52665742794f5Anders Carlsson}
36935873c49adad211ff466e34342a52665742794f5Anders Carlsson
370efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
371efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Success(E);
3724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
374efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitMemberExpr(MemberExpr *E) {
3754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  QualType Ty;
3764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->isArrow()) {
377efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E->getBase(), Result, Info))
378efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3796217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek    Ty = E->getBase()->getType()->getAs<PointerType>()->getPointeeType();
3804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else {
381efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!Visit(E->getBase()))
382efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3834efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Ty = E->getBase()->getType();
3844efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3854efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3866217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek  RecordDecl *RD = Ty->getAs<RecordType>()->getDecl();
3874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
38886f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor
38986f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
39086f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  if (!FD) // FIXME: deal with other kinds of member expressions
391efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3922be586108bb401019647791feca19ea03fd477ceEli Friedman
3932be586108bb401019647791feca19ea03fd477ceEli Friedman  if (FD->getType()->isReferenceType())
394efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3952be586108bb401019647791feca19ea03fd477ceEli Friedman
3964efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  // FIXME: This is linear time.
39744b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  unsigned i = 0;
39817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (RecordDecl::field_iterator Field = RD->field_begin(),
39917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis                               FieldEnd = RD->field_end();
40044b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor       Field != FieldEnd; (void)++Field, ++i) {
40144b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    if (*Field == FD)
4024efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      break;
4034efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
4044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
405efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += CharUnits::fromQuantity(RL.getFieldOffset(i) / 8);
406efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
4084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
409efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
4103068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluatePointer(E->getBase(), Result, Info))
411efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4133068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APSInt Index;
4143068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluateInteger(E->getIdx(), Index, Info))
415efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4163068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
417199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(E->getType());
418efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += Index.getSExtValue() * ElementSize;
419efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4203068d117951a8df54bae9db039b56201ab10962bAnders Carlsson}
4214efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
422efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitUnaryDeref(UnaryOperator *E) {
423efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluatePointer(E->getSubExpr(), Result, Info);
424e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman}
425e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman
4264efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
427f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Pointer Evaluation
428f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
429f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
430c754aa62643e66ab967ca32ae8b0b3fc419bba25Anders Carlssonnamespace {
431770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass PointerExprEvaluator
432efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<PointerExprEvaluator, bool> {
43387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
434efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
435efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
436efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
437efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
438efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
439efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
440efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
4412bad1687fe6f00e10767a691a33b070b151902b6Anders Carlssonpublic:
4421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
443efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  PointerExprEvaluator(EvalInfo &info, LValue &Result)
444efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    : Info(info), Result(Result) {}
445f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
446efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
447efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4482bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
4492bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
450efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
4512bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
452efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBinaryOperator(const BinaryOperator *E);
453efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr* E);
454efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
4552217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman      { return Visit(E->getSubExpr()); }
456efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryAddrOf(const UnaryOperator *E);
457efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCStringLiteral(ObjCStringLiteral *E)
458efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
459efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitAddrLabelExpr(AddrLabelExpr *E)
460efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
461efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCallExpr(CallExpr *E);
462efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBlockExpr(BlockExpr *E) {
463b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump    if (!E->hasBlockDeclRefExprs())
464efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
465efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
466b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump  }
467efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitImplicitValueInitExpr(ImplicitValueInitExpr *E)
468efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
469efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitConditionalOperator(ConditionalOperator *E);
470efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(ChooseExpr *E)
4716e8ed16ffef02b82995a90bdcf10ffff7d63839aSebastian Redl      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
472efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E)
473efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
474ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: @protocol, @selector
4752bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson};
476f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
4772bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
478efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr* E, LValue& Result, EvalInfo &Info) {
4797db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->hasPointerRepresentation());
480efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return PointerExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
481f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
482650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
483efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
484650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() != BinaryOperator::Add &&
485650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson      E->getOpcode() != BinaryOperator::Sub)
486efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
488650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *PExp = E->getLHS();
489650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *IExp = E->getRHS();
490650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (IExp->getType()->isPointerType())
491f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    std::swap(PExp, IExp);
4921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
493efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluatePointer(PExp, Result, Info))
494efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
496efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  llvm::APSInt Offset;
497efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluateInteger(IExp, Offset, Info))
498efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
499efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  int64_t AdditionalOffset
500efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    = Offset.isSigned() ? Offset.getSExtValue()
501efdb83e26f9a1fd2566afe54461216cd84814d42John McCall                        : static_cast<int64_t>(Offset.getZExtValue());
502650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
503e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  // Compute the new offset in the appropriate width.
504e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar
505e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  QualType PointeeType =
506e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar    PExp->getType()->getAs<PointerType>()->getPointeeType();
507efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  CharUnits SizeOfPointee;
5081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5094d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  // Explicitly handle GNU void* and function pointer arithmetic extensions.
5104d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  if (PointeeType->isVoidType() || PointeeType->isFunctionType())
511efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = CharUnits::One();
5124d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  else
513efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = Info.Ctx.getTypeSizeInChars(PointeeType);
5144efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
515650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() == BinaryOperator::Add)
516efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset += AdditionalOffset * SizeOfPointee;
517650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  else
518efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset -= AdditionalOffset * SizeOfPointee;
5194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
520efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
521650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson}
5224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
523efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
524efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluateLValue(E->getSubExpr(), Result, Info);
5254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
5261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
527650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
528efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCastExpr(CastExpr* E) {
52909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  Expr* SubExpr = E->getSubExpr();
530650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
53109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  switch (E->getCastKind()) {
53209a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  default:
53309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    break;
53409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
53509a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_Unknown: {
53609a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    // FIXME: The handling for CK_Unknown is ugly/shouldn't be necessary!
53709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
53809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    // Check for pointer->pointer cast
53909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    if (SubExpr->getType()->isPointerType() ||
54009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isObjCObjectPointerType() ||
54109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isNullPtrType() ||
54209a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isBlockPointerType())
54309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      return Visit(SubExpr);
54409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
5452ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType()) {
546efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      APValue Value;
547efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
54809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        break;
54909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
550efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      if (Value.isInt()) {
551efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
552efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Base = 0;
553efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
554efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        return true;
555efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      } else {
556efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Base = Value.getLValueBase();
557efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Offset = Value.getLValueOffset();
558efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        return true;
55909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      }
56009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    }
56109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    break;
562650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
5631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
56409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_NoOp:
56509a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_BitCast:
56609a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_AnyPointerToObjCPointerCast:
56709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_AnyPointerToBlockPointerCast:
56809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    return Visit(SubExpr);
56909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
57009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_IntegralToPointer: {
571efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    APValue Value;
572efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
57309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      break;
57469ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
575efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (Value.isInt()) {
576efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
577efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = 0;
578efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
579efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
580efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    } else {
581efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      // Cast is of an lvalue, no need to change value.
582efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = Value.getLValueBase();
583efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = Value.getLValueOffset();
584efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
585650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    }
586650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
58709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_ArrayToPointerDecay:
588efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  case CastExpr::CK_FunctionToPointerDecay:
589efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return EvaluateLValue(SubExpr, Result, Info);
5904efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
5914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
592efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
5931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump}
594650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
595efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCallExpr(CallExpr *E) {
5961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (E->isBuiltinCall(Info.Ctx) ==
5970d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___CFStringMakeConstantString ||
5980d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall      E->isBuiltinCall(Info.Ctx) ==
5990d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___NSStringMakeConstantString)
600efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
601efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
6023941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman}
6033941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman
604efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
6054efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool BoolResult;
6064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
607efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
6084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
6094efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
610efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Visit(EvalExpr);
6114efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
612f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
613f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
61459b5da6d853b4368b984700315adf7b37de05764Nate Begeman// Vector Evaluation
61559b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
61659b5da6d853b4368b984700315adf7b37de05764Nate Begeman
61759b5da6d853b4368b984700315adf7b37de05764Nate Begemannamespace {
618770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramer  class VectorExprEvaluator
61959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  : public StmtVisitor<VectorExprEvaluator, APValue> {
62059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    EvalInfo &Info;
62191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue GetZeroVector(QualType VecType);
62259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  public:
6231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
62459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    VectorExprEvaluator(EvalInfo &info) : Info(info) {}
6251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
62659b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitStmt(Stmt *S) {
62759b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return APValue();
62859b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
6291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
63091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitParenExpr(ParenExpr *E)
63191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        { return Visit(E->getSubExpr()); }
63291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryExtension(const UnaryOperator *E)
63391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryPlus(const UnaryOperator *E)
63591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryReal(const UnaryOperator *E)
63791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E)
63991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return GetZeroVector(E->getType()); }
64059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCastExpr(const CastExpr* E);
64159b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
64259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitInitListExpr(const InitListExpr *E);
64391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitConditionalOperator(const ConditionalOperator *E);
644ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    APValue VisitChooseExpr(const ChooseExpr *E)
645ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
64691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryImag(const UnaryOperator *E);
64791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    // FIXME: Missing: unary -, unary ~, binary add/sub/mul/div,
6482217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman    //                 binary comparisons, binary and/or/xor,
64991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //                 shufflevector, ExtVectorElementExpr
65091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //        (Note that these require implementing conversions
65191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //         between vector types.)
65259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  };
65359b5da6d853b4368b984700315adf7b37de05764Nate Begeman} // end anonymous namespace
65459b5da6d853b4368b984700315adf7b37de05764Nate Begeman
65559b5da6d853b4368b984700315adf7b37de05764Nate Begemanstatic bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
65659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (!E->getType()->isVectorType())
65759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return false;
65859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  Result = VectorExprEvaluator(Info).Visit(const_cast<Expr*>(E));
65959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return !Result.isUninit();
66059b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
66159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
66259b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue VectorExprEvaluator::VisitCastExpr(const CastExpr* E) {
663183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VTy = E->getType()->getAs<VectorType>();
664c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  QualType EltTy = VTy->getElementType();
665c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned NElts = VTy->getNumElements();
666c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned EltWidth = Info.Ctx.getTypeSize(EltTy);
6671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
66859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  const Expr* SE = E->getSubExpr();
669e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  QualType SETy = SE->getType();
670e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  APValue Result = APValue();
67159b5da6d853b4368b984700315adf7b37de05764Nate Begeman
672e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  // Check for vector->vector bitcast and scalar->vector splat.
673e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  if (SETy->isVectorType()) {
67459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return this->Visit(const_cast<Expr*>(SE));
675e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isIntegerType()) {
676e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APSInt IntResult;
677d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateInteger(SE, IntResult, Info))
678d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
679d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(IntResult);
680e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isRealFloatingType()) {
681e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APFloat F(0.0);
682d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateFloat(SE, F, Info))
683d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
684d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(F);
685d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar  } else
686c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue();
68759b5da6d853b4368b984700315adf7b37de05764Nate Begeman
688c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to ExtVector, convert the scalar to the element type
689c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // and splat it to all elements.
690c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  if (E->getType()->isExtVectorType()) {
691c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType() && Result.isInt())
692c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToIntCast(EltTy, SETy, Result.getInt(),
693c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                          Info.Ctx));
694c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isIntegerType())
695c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToIntCast(EltTy, SETy, Result.getFloat(),
696c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
697c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType() && Result.isInt())
698c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToFloatCast(EltTy, SETy, Result.getInt(),
699c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
700c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
701c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToFloatCast(EltTy, SETy, Result.getFloat(),
702c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                              Info.Ctx));
703c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
704c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
705c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
706c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    // Splat and create vector APValue.
707c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    llvm::SmallVector<APValue, 4> Elts(NElts, Result);
708c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue(&Elts[0], Elts.size());
709e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  }
710c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
711c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to regular gcc-style vector type, bitcast the scalar
712c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // to the vector. To construct the APValue vector initializer, bitcast the
713c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // initializing value to an APInt, and shift out the bits pertaining to each
714c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // element.
715c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  APSInt Init;
716c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  Init = Result.isInt() ? Result.getInt() : Result.getFloat().bitcastToAPInt();
7171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
718c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  llvm::SmallVector<APValue, 4> Elts;
719c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  for (unsigned i = 0; i != NElts; ++i) {
720c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    APSInt Tmp = Init;
721c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    Tmp.extOrTrunc(EltWidth);
7221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
723c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType())
724c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(Tmp));
725c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
726c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(APFloat(Tmp)));
727c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
728c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
729c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
730c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    Init >>= EltWidth;
731c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  }
732c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  return APValue(&Elts[0], Elts.size());
73359b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
73459b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7351eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
73659b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
73759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return this->Visit(const_cast<Expr*>(E->getInitializer()));
73859b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
73959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7401eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
74159b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
742183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = E->getType()->getAs<VectorType>();
74359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  unsigned NumInits = E->getNumInits();
74491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  unsigned NumElements = VT->getNumElements();
7451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
74659b5da6d853b4368b984700315adf7b37de05764Nate Begeman  QualType EltTy = VT->getElementType();
74759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  llvm::SmallVector<APValue, 4> Elements;
74859b5da6d853b4368b984700315adf7b37de05764Nate Begeman
749a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // If a vector is initialized with a single element, that value
750a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // becomes every element of the vector, not just the first.
751a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // This is the behavior described in the IBM AltiVec documentation.
752a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  if (NumInits == 1) {
753a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    APValue InitValue;
75459b5da6d853b4368b984700315adf7b37de05764Nate Begeman    if (EltTy->isIntegerType()) {
75559b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APSInt sInt(32);
756a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateInteger(E->getInit(0), sInt, Info))
757a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
758a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(sInt);
75959b5da6d853b4368b984700315adf7b37de05764Nate Begeman    } else {
76059b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APFloat f(0.0);
761a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateFloat(E->getInit(0), f, Info))
762a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
763a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(f);
764a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
765a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
766a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      Elements.push_back(InitValue);
767a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
768a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  } else {
769a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
770a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (EltTy->isIntegerType()) {
771a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APSInt sInt(32);
772a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
773a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateInteger(E->getInit(i), sInt, Info))
774a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
775a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
776a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          sInt = Info.Ctx.MakeIntValue(0, EltTy);
777a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
778a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(sInt));
77991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      } else {
780a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APFloat f(0.0);
781a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
782a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateFloat(E->getInit(i), f, Info))
783a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
784a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
785a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          f = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy));
786a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
787a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(f));
78891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      }
78959b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
79059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  }
79159b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return APValue(&Elements[0], Elements.size());
79259b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
79359b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7941eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
79591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanVectorExprEvaluator::GetZeroVector(QualType T) {
796183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = T->getAs<VectorType>();
79791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  QualType EltTy = VT->getElementType();
79891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue ZeroElement;
79991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EltTy->isIntegerType())
80091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement = APValue(Info.Ctx.MakeIntValue(0, EltTy));
80191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  else
80291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement =
80391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        APValue(APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy)));
80491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
80591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  llvm::SmallVector<APValue, 4> Elements(VT->getNumElements(), ZeroElement);
80691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue(&Elements[0], Elements.size());
80791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
80891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
80991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
81091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  bool BoolResult;
81191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
81291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return APValue();
81391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
81591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue Result;
81791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EvaluateVector(EvalExpr, Result, Info))
81891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return Result;
81991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue();
82091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
82191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
82391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
82491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    Info.EvalResult.HasSideEffects = true;
82591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return GetZeroVector(E->getType());
82691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
82791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82859b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
829f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Integer Evaluation
830f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
831f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
832f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnernamespace {
833770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass IntExprEvaluator
834b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  : public StmtVisitor<IntExprEvaluator, bool> {
83587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
83630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue &Result;
837f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerpublic:
83830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  IntExprEvaluator(EvalInfo &info, APValue &result)
83987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    : Info(info), Result(result) {}
840f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
8413f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  bool Success(const llvm::APSInt &SI, const Expr *E) {
8422ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8432ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
84430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.isSigned() == E->getType()->isSignedIntegerType() &&
8453f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
84630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8473f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
84830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(SI);
8493f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
8503f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
8513f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar
852131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(const llvm::APInt &I, const Expr *E) {
8532ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8542ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
85530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8563f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
85730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(APSInt(I));
85830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result.getInt().setIsUnsigned(E->getType()->isUnsignedIntegerType());
859131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
860131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
861131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
862131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(uint64_t Value, const Expr *E) {
8632ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8642ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
86530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
866131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
867131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
868131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
86982206e267ce6cc709797127616f64672d255b310Anders Carlsson  bool Error(SourceLocation L, diag::kind D, const Expr *E) {
87032fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    // Take the first error.
87154da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    if (Info.EvalResult.Diag == 0) {
87254da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.DiagLoc = L;
87354da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.Diag = D;
87482206e267ce6cc709797127616f64672d255b310Anders Carlsson      Info.EvalResult.DiagExpr = E;
87532fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    }
87654176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    return false;
8777a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner  }
8781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
879f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
880f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //                            Visitor Methods
881f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
8821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
88332fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitStmt(Stmt *) {
88432fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    assert(0 && "This should be called on integers, stmts are not integers");
88532fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    return false;
88632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  }
8871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
88832fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitExpr(Expr *E) {
8890e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
890f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  }
8911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
892b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
893f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
8944c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitIntegerLiteral(const IntegerLiteral *E) {
895131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
8964c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
8974c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitCharacterLiteral(const CharacterLiteral *E) {
898131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
8994c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
9004c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
901ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // Per gcc docs "this built-in function ignores top level
902ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // qualifiers".  We need to use the canonical version to properly
903ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // be able to strip CRV qualifiers from the type.
904ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T0 = Info.Ctx.getCanonicalType(E->getArgType1());
905ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T1 = Info.Ctx.getCanonicalType(E->getArgType2());
9061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return Success(Info.Ctx.typesAreCompatible(T0.getUnqualifiedType(),
907131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                                               T1.getUnqualifiedType()),
908131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                   E);
9094c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
910043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
911043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool CheckReferencedDecl(const Expr *E, const Decl *D);
912043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitDeclRefExpr(const DeclRefExpr *E) {
913043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return CheckReferencedDecl(E, E->getDecl());
914043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
915043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitMemberExpr(const MemberExpr *E) {
916043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (CheckReferencedDecl(E, E->getMemberDecl())) {
917043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      // Conservatively assume a MemberExpr will have side-effects
918043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      Info.EvalResult.HasSideEffects = true;
919043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      return true;
920043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    }
921043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return false;
922043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
923043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
924c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  bool VisitCallExpr(CallExpr *E);
925b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitBinaryOperator(const BinaryOperator *E);
9268ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  bool VisitOffsetOfExpr(const OffsetOfExpr *E);
927b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitUnaryOperator(const UnaryOperator *E);
928ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes  bool VisitConditionalOperator(const ConditionalOperator *E);
929f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
930a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool VisitCastExpr(CastExpr* E);
9310518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  bool VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
9320518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
9333068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
934131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9353068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9373f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  bool VisitGNUNullExpr(const GNUNullExpr *E) {
938131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9393f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  }
9401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
941ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
942131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9433068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9443068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
945664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
946664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    return Success(0, E);
947664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  }
948664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
94964b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
9505e03f9ea8174ae588c5e69ec6b5ef4c68f8fd766Douglas Gregor    return Success(E->EvaluateTrait(Info.Ctx), E);
95164b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  }
95264b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl
953ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E) {
954ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    return Visit(E->getChosenSubExpr(Info.Ctx));
955ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  }
956ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
957722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  bool VisitUnaryReal(const UnaryOperator *E);
958664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitUnaryImag(const UnaryOperator *E);
959664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
960fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattnerprivate:
9618b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfExpr(const Expr *E);
9628b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfType(QualType T);
96342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  static QualType GetObjectType(const Expr *E);
96442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  bool TryEvaluateBuiltinObjectSize(CallExpr *E);
965664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  // FIXME: Missing: array subscript of vector, member of vector
966f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner};
967f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
968f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
96969ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateIntegerOrLValue(const Expr* E, APValue &Result, EvalInfo &Info) {
9702ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
97169ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  return IntExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
97269ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar}
97369ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
97469ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
9752ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
9767db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall
97730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue Val;
97869ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  if (!EvaluateIntegerOrLValue(E, Val, Info) || !Val.isInt())
97930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
98030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  Result = Val.getInt();
98130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  return true;
982f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
983f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
984043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedmanbool IntExprEvaluator::CheckReferencedDecl(const Expr* E, const Decl* D) {
9854c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Enums are integer constant exprs.
98629a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman  if (const EnumConstantDecl *ECD = dyn_cast<EnumConstantDecl>(D))
98729a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman    return Success(ECD->getInitVal(), E);
988b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
989b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  // In C++, const, non-volatile integers initialized with ICEs are ICEs.
990e1646da3bb0fe97b372e5fe8cefc537b22048fc4Eli Friedman  // In C, they can also be folded, although they are not ICEs.
991cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor  if (Info.Ctx.getCanonicalType(E->getType()).getCVRQualifiers()
992cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor                                                        == Qualifiers::Const) {
993f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
994f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson    if (isa<ParmVarDecl>(D))
995f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson      return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
996f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
997043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
99831310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl      if (const Expr *Init = VD->getAnyInitializer()) {
999c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (APValue *V = VD->getEvaluatedValue()) {
1000c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          if (V->isInt())
1001c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman            return Success(V->getInt(), E);
1002c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1003c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        }
1004c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1005c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (VD->isEvaluatingValue())
1006c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1007c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1008c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatingValue();
1009c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
101078d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        if (Visit(const_cast<Expr*>(Init))) {
101178d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          // Cache the evaluated value in the variable declaration.
1012c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          VD->setEvaluatedValue(Result);
101378d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          return true;
101478d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        }
101578d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor
1016c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatedValue(APValue());
101778d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        return false;
101878d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor      }
1019b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    }
1020b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  }
1021b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
10224c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Otherwise, random variable references are not constants.
10230e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson  return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
10244c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
10254c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner
1026a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
1027a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// as GCC.
1028a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattnerstatic int EvaluateBuiltinClassifyType(const CallExpr *E) {
1029a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // The following enum mimics the values returned by GCC.
10307c80bd64032e610c0dbd74fc0ef6ea334447f2fdSebastian Redl  // FIXME: Does GCC differ between lvalue and rvalue references here?
1031a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  enum gcc_type_class {
1032a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    no_type_class = -1,
1033a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    void_type_class, integer_type_class, char_type_class,
1034a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    enumeral_type_class, boolean_type_class,
1035a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    pointer_type_class, reference_type_class, offset_type_class,
1036a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    real_type_class, complex_type_class,
1037a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    function_type_class, method_type_class,
1038a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    record_type_class, union_type_class,
1039a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    array_type_class, string_type_class,
1040a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    lang_type_class
1041a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  };
10421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // If no argument was supplied, default to "no_type_class". This isn't
1044a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // ideal, however it is what gcc does.
1045a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (E->getNumArgs() == 0)
1046a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return no_type_class;
10471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1048a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  QualType ArgTy = E->getArg(0)->getType();
1049a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (ArgTy->isVoidType())
1050a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return void_type_class;
1051a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isEnumeralType())
1052a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return enumeral_type_class;
1053a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isBooleanType())
1054a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return boolean_type_class;
1055a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isCharType())
1056a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return string_type_class; // gcc doesn't appear to use char_type_class
1057a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isIntegerType())
1058a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return integer_type_class;
1059a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isPointerType())
1060a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return pointer_type_class;
1061a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isReferenceType())
1062a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return reference_type_class;
1063a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isRealType())
1064a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return real_type_class;
1065a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isComplexType())
1066a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return complex_type_class;
1067a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isFunctionType())
1068a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return function_type_class;
1069fb87b89fc9eb103e19fb8e4b925c23f0bd091b99Douglas Gregor  else if (ArgTy->isStructureOrClassType())
1070a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return record_type_class;
1071a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1072a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1073a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isArrayType())
1074a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return array_type_class;
1075a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1076a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1077a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else  // FIXME: offset_type_class, method_type_class, & lang_type_class?
1078a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
1079a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  return -1;
1080a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner}
1081a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
108242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Retrieves the "underlying object type" of the given expression,
108342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// as used by __builtin_object_size.
108442c8f87eb60958170c46767273bf93e6c96125bfJohn McCallQualType IntExprEvaluator::GetObjectType(const Expr *E) {
108542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
108642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
108742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->getType();
108842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  } else if (isa<CompoundLiteralExpr>(E)) {
108942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return E->getType();
109042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
109142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return QualType();
109342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
109442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109542c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool IntExprEvaluator::TryEvaluateBuiltinObjectSize(CallExpr *E) {
109642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // TODO: Perhaps we should let LLVM lower this?
109742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  LValue Base;
109842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!EvaluatePointer(E->getArg(0), Base, Info))
109942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
110042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // If we can prove the base is null, lower to zero now.
110242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *LVBase = Base.getLValueBase();
110342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!LVBase) return Success(0, E);
110442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  QualType T = GetObjectType(LVBase);
110642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (T.isNull() ||
110742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isIncompleteType() ||
110842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !T->isObjectType() ||
110942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isVariablyModifiedType() ||
111042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isDependentType())
111142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
111242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Size = Info.Ctx.getTypeSizeInChars(T);
111442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Offset = Base.getLValueOffset();
111542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!Offset.isNegative() && Offset <= Size)
111742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size -= Offset;
111842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  else
111942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size = CharUnits::Zero();
112042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return Success(Size.getQuantity(), E);
112142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
112242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1123c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedmanbool IntExprEvaluator::VisitCallExpr(CallExpr *E) {
11243c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
1125019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  default:
11260e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
112764eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
112864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  case Builtin::BI__builtin_object_size: {
112942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (TryEvaluateBuiltinObjectSize(E))
113042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
113164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1132b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // If evaluating the argument has side-effects we can't determine
1133b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // the size of the object and lower it to unknown now.
1134393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian    if (E->getArg(0)->HasSideEffects(Info.Ctx)) {
11353f27b384801de26ce7efaa395699b42719372f24Benjamin Kramer      if (E->getArg(1)->EvaluateAsInt(Info.Ctx).getZExtValue() <= 1)
1136cf184655319cf7a5b811067cff9d26a5741fd161Chris Lattner        return Success(-1ULL, E);
113764eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump      return Success(0, E);
113864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    }
1139c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
114064eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
114164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  }
114264eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1143019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_classify_type:
1144131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(EvaluateBuiltinClassifyType(E), E);
11451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11464bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson  case Builtin::BI__builtin_constant_p:
1147019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // __builtin_constant_p always has one operand: it returns true if that
1148019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // operand can be folded, false otherwise.
1149131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
115021fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner
115121fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  case Builtin::BI__builtin_eh_return_data_regno: {
115221fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    int Operand = E->getArg(0)->EvaluateAsInt(Info.Ctx).getZExtValue();
115321fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    Operand = Info.Ctx.Target.getEHDataRegisterNumber(Operand);
115421fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    return Success(Operand, E);
115521fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  }
1156c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman
1157c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  case Builtin::BI__builtin_expect:
1158c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman    return Visit(E->getArg(0));
1159019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
11604c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
1161f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
1162b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
1163a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == BinaryOperator::Comma) {
1164027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    if (!Visit(E->getRHS()))
1165027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson      return false;
11664fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
116733ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
116833ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // conservatively mark it.
116933ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
117033ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      Info.EvalResult.HasSideEffects = true;
1171c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner
1172027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    return true;
1173a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1174a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1175a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->isLogicalOp()) {
1176a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // These need to be handled specially because the operands aren't
1177a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // necessarily integral
1178fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    bool lhsResult, rhsResult;
11791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1180fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
118151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // We were able to evaluate the LHS, see if we can get away with not
118251fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
118333ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      if (lhsResult == (E->getOpcode() == BinaryOperator::LOr))
11843f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        return Success(lhsResult, E);
1185a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1186fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
11874bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        if (E->getOpcode() == BinaryOperator::LOr)
1188131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult || rhsResult, E);
11894bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        else
1190131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult && rhsResult, E);
11914bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
11924bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson    } else {
1193fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
11944bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // We can't evaluate the LHS; however, sometimes the result
11954bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
11961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        if (rhsResult == (E->getOpcode() == BinaryOperator::LOr) ||
1197fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson            !rhsResult == (E->getOpcode() == BinaryOperator::LAnd)) {
1198131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          // Since we weren't able to evaluate the left hand side, it
1199fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          // must have had side effects.
1200fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          Info.EvalResult.HasSideEffects = true;
1201131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
1202131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(rhsResult, E);
12034bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        }
12044bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
1205a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    }
1206a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1207a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1208c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner  }
120954176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
1210286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType LHSTy = E->getLHS()->getType();
1211286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType RHSTy = E->getRHS()->getType();
12124087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12134087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  if (LHSTy->isAnyComplexType()) {
12144087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    assert(RHSTy->isAnyComplexType() && "Invalid comparison");
1215f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LHS, RHS;
12164087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12174087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getLHS(), LHS, Info))
12184087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12194087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12204087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getRHS(), RHS, Info))
12214087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12224087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12234087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (LHS.isComplexFloat()) {
12241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_r =
12254087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
12261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_i =
12274087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
12284087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12294087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
1230131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((CR_r == APFloat::cmpEqual &&
1231131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        CR_i == APFloat::cmpEqual), E);
1232131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
1233131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
1234131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid complex comparison.");
12351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        return Success(((CR_r == APFloat::cmpGreaterThan ||
1236fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpLessThan ||
1237fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpUnordered) ||
12381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                        (CR_i == APFloat::cmpGreaterThan ||
1239fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpLessThan ||
1240fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpUnordered)), E);
1241131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12424087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    } else {
12434087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
1244131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
1245131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
1246131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
1247131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
1248131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid compex comparison.");
1249131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
1250131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
1251131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12524087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    }
12534087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  }
12541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1255286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (LHSTy->isRealFloatingType() &&
1256286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      RHSTy->isRealFloatingType()) {
1257286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat RHS(0.0), LHS(0.0);
12581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1259286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getRHS(), RHS, Info))
1260286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1262286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getLHS(), LHS, Info))
1263286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1265286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat::cmpResult CR = LHS.compare(RHS);
1266529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson
1267286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    switch (E->getOpcode()) {
1268286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    default:
1269286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      assert(0 && "Invalid binary operator!");
1270286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LT:
1271131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan, E);
1272286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GT:
1273131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan, E);
1274286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LE:
1275131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
1276286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GE:
12771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
1278131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     E);
1279286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::EQ:
1280131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpEqual, E);
1281286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::NE:
12821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan
1283fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpLessThan
1284fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpUnordered, E);
1285286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    }
1286286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  }
12871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1288ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman  if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
1289ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman    if (E->getOpcode() == BinaryOperator::Sub || E->isEqualityOp()) {
1290efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue LHSValue;
12913068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
12923068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1293a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1294efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue RHSValue;
12953068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
12963068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1297a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
12985bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // Reject any bases from the normal codepath; we special-case comparisons
12995bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // to null.
13005bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      if (LHSValue.getLValueBase()) {
13015bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13025bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1303a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (RHSValue.getLValueBase() || !RHSValue.getLValueOffset().isZero())
13045bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13055bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13065bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(LHSValue, bres))
13075bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13085bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        return Success(bres ^ (E->getOpcode() == BinaryOperator::EQ), E);
13095bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      } else if (RHSValue.getLValueBase()) {
13105bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13115bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1312a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (LHSValue.getLValueBase() || !LHSValue.getLValueOffset().isZero())
13135bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13145bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13155bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(RHSValue, bres))
13165bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13175bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        return Success(bres ^ (E->getOpcode() == BinaryOperator::EQ), E);
13185bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      }
1319a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1320ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      if (E->getOpcode() == BinaryOperator::Sub) {
13214992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType Type = E->getLHS()->getType();
13224992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType ElementType = Type->getAs<PointerType>()->getPointeeType();
13233068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
1324a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits ElementSize = CharUnits::One();
1325ce1bca73aef9bbf6359ab8420278203dda81a054Eli Friedman        if (!ElementType->isVoidType() && !ElementType->isFunctionType())
1326a73058324197b7bdfd19307965954f626e26199dKen Dyck          ElementSize = Info.Ctx.getTypeSizeInChars(ElementType);
1327a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1328a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits Diff = LHSValue.getLValueOffset() -
1329a73058324197b7bdfd19307965954f626e26199dKen Dyck                             RHSValue.getLValueOffset();
1330a73058324197b7bdfd19307965954f626e26199dKen Dyck        return Success(Diff / ElementSize, E);
1331ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1332ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      bool Result;
1333ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      if (E->getOpcode() == BinaryOperator::EQ) {
1334ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() == RHSValue.getLValueOffset();
1335267c0ab1b9a15768f3f15abbfc40ce344751c78bEli Friedman      } else {
1336ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() != RHSValue.getLValueOffset();
1337ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1338ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      return Success(Result, E);
13393068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    }
13403068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
13412ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!LHSTy->isIntegralOrEnumerationType() ||
13422ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor      !RHSTy->isIntegralOrEnumerationType()) {
1343a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // We can't continue from here for non-integral types, and they
1344a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // could potentially confuse the following operations.
1345a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1346a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1347a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1348a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  // The LHS of a constant expr is always evaluated and needed.
134930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Visit(E->getLHS()))
1350a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false; // error in subexpression.
1351d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman
135242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APValue RHSVal;
135342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!EvaluateIntegerOrLValue(E->getRHS(), RHSVal, Info))
135430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
135542edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
135642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like (unsigned long)&a + 4.
135742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->isAdditiveOp() && Result.isLValue() && RHSVal.isInt()) {
1358a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = Result.getLValueOffset();
1359a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits AdditionalOffset = CharUnits::fromQuantity(
1360a73058324197b7bdfd19307965954f626e26199dKen Dyck                                     RHSVal.getInt().getZExtValue());
136142edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    if (E->getOpcode() == BinaryOperator::Add)
1362a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset += AdditionalOffset;
136342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    else
1364a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset -= AdditionalOffset;
1365a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(Result.getLValueBase(), Offset);
136642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
136742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
136842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
136942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like 4 + (unsigned long)&a
137042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->getOpcode() == BinaryOperator::Add &&
137142edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman        RHSVal.isLValue() && Result.isInt()) {
1372a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = RHSVal.getLValueOffset();
1373a73058324197b7bdfd19307965954f626e26199dKen Dyck    Offset += CharUnits::fromQuantity(Result.getInt().getZExtValue());
1374a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(RHSVal.getLValueBase(), Offset);
137542edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
137642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
137742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
137842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // All the following cases expect both operands to be an integer
137942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!Result.isInt() || !RHSVal.isInt())
1380b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner    return false;
1381a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
138242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APSInt& RHS = RHSVal.getInt();
138342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
1384a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  switch (E->getOpcode()) {
138532fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  default:
13860e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
138730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Mul: return Success(Result.getInt() * RHS, E);
138830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Add: return Success(Result.getInt() + RHS, E);
138930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Sub: return Success(Result.getInt() - RHS, E);
139030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::And: return Success(Result.getInt() & RHS, E);
139130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Xor: return Success(Result.getInt() ^ RHS, E);
139230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Or:  return Success(Result.getInt() | RHS, E);
1393a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Div:
139454176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
13950e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
139630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() / RHS, E);
1397a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Rem:
139854176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
13990e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
140030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() % RHS, E);
14013f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shl: {
140254176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // FIXME: Warn about out of range shift amounts!
14031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
140430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
140530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() << SA, E);
14063f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14073f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shr: {
14081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
140930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
141030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() >> SA, E);
14113f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
141330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LT: return Success(Result.getInt() < RHS, E);
141430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GT: return Success(Result.getInt() > RHS, E);
141530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LE: return Success(Result.getInt() <= RHS, E);
141630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GE: return Success(Result.getInt() >= RHS, E);
141730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::EQ: return Success(Result.getInt() == RHS, E);
141830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::NE: return Success(Result.getInt() != RHS, E);
1419b11e77836dd0867955c5abf32baf1c3e6c7f81e1Eli Friedman  }
1420a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
1421a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
1422ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopesbool IntExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
1423a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  bool Cond;
1424a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  if (!HandleConversionToBool(E->getCond(), Cond, Info))
1425ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes    return false;
1426ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
1427a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
1428ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes}
1429ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
14308b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfType(QualType T) {
14315d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14325d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14335d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14345d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14355d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = T->getAs<ReferenceType>())
14365d484e8cf710207010720589d89602233de61d01Sebastian Redl    T = Ref->getPointeeType();
14375d484e8cf710207010720589d89602233de61d01Sebastian Redl
1438e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the alignment.
1439e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
144018857644059c45da6776f1a288eec7b4cf3a844aDouglas Gregor
14412be586108bb401019647791feca19ea03fd477ceEli Friedman  // __alignof is defined to return the preferred alignment.
14428b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  return CharUnits::fromQuantity(
14438b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      Info.Ctx.getPreferredTypeAlign(T.getTypePtr()) / CharSize);
1444e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1445e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14468b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
1447af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  E = E->IgnoreParens();
1448af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1449af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // alignof decl is always accepted, even if it doesn't make sense: we default
14501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // to 1 in those cases.
1451af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
14528b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(DRE->getDecl(),
14538b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1454a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1455af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
14568b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
14578b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1458af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1459e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return GetAlignOfType(E->getType());
1460e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1461e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
1462e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14630518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// VisitSizeAlignOfExpr - Evaluate a sizeof or alignof with a result as the
14640518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// expression's type.
14650518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redlbool IntExprEvaluator::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
1466e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Handle alignof separately.
1467e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!E->isSizeOf()) {
1468e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    if (E->isArgumentType())
14698b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfType(E->getArgumentType()).getQuantity(), E);
1470e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    else
14718b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfExpr(E->getArgumentExpr()).getQuantity(), E);
1472e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  }
1473a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
14740518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType SrcTy = E->getTypeOfArgument();
14755d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14765d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14775d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14785d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14795d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
14805d484e8cf710207010720589d89602233de61d01Sebastian Redl    SrcTy = Ref->getPointeeType();
14810518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
1482131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
1483131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // extension.
1484131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (SrcTy->isVoidType() || SrcTy->isFunctionType())
1485131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(1, E);
1486a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1487fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner  // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
1488e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!SrcTy->isConstantSizeType())
1489fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner    return false;
1490f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
1491e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the size.
1492199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  return Success(Info.Ctx.getTypeSizeInChars(SrcTy).getQuantity(), E);
1493fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner}
1494fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
14958ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregorbool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *E) {
14968ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  CharUnits Result;
14978ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  unsigned n = E->getNumComponents();
14988ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  OffsetOfExpr* OOE = const_cast<OffsetOfExpr*>(E);
14998ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  if (n == 0)
15008ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    return false;
15018ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  QualType CurrentType = E->getTypeSourceInfo()->getType();
15028ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  for (unsigned i = 0; i != n; ++i) {
15038ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    OffsetOfExpr::OffsetOfNode ON = OOE->getComponent(i);
15048ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    switch (ON.getKind()) {
15058ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Array: {
15068ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
15078ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      APSInt IdxResult;
15088ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!EvaluateInteger(Idx, IdxResult, Info))
15098ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15108ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
15118ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!AT)
15128ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15138ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = AT->getElementType();
15148ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
15158ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Result += IdxResult.getSExtValue() * ElementSize;
15168ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        break;
15178ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15188ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Field: {
15208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      FieldDecl *MemberDecl = ON.getField();
15218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
15228ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!RT)
15238ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15248ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      RecordDecl *RD = RT->getDecl();
15258ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
15268ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      unsigned i = 0;
15278ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      // FIXME: It would be nice if we didn't have to loop here!
15288ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      for (RecordDecl::field_iterator Field = RD->field_begin(),
15298ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor                                      FieldEnd = RD->field_end();
15308ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor           Field != FieldEnd; (void)++Field, ++i) {
15318ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        if (*Field == MemberDecl)
15328ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor          break;
15338ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      }
1534cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      assert(i < RL.getFieldCount() && "offsetof field in wrong type");
1535cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1536cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                           RL.getFieldOffset(i) / Info.Ctx.getCharWidth());
15378ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = MemberDecl->getType().getNonReferenceType();
15388ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      break;
15398ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15408ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15418ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Identifier:
15428ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      llvm_unreachable("dependent __builtin_offsetof");
1543cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      return false;
1544cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1545cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Base: {
1546cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CXXBaseSpecifier *BaseSpec = ON.getBase();
1547cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (BaseSpec->isVirtual())
1548cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1549cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1550cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the layout of the class whose base we are looking into.
1551cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
1552cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!RT)
1553cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1554cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      RecordDecl *RD = RT->getDecl();
1555cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
1556cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1557cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the base class itself.
1558cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CurrentType = BaseSpec->getType();
1559cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *BaseRT = CurrentType->getAs<RecordType>();
1560cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!BaseRT)
1561cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1562cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1563cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Add the offset to the base.
1564cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1565cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                RL.getBaseClassOffset(cast<CXXRecordDecl>(BaseRT->getDecl()))
1566cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                                        / Info.Ctx.getCharWidth());
1567cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      break;
1568cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    }
15698ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15708ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  }
15718ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  return Success(Result.getQuantity(), E);
15728ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor}
15738ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1574b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
15754c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Special case unary operators that do not need their subexpression
15764c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // evaluated.  offsetof/sizeof/alignof are all special.
157735183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman  if (E->isOffsetOfOp()) {
157835183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    // The AST for offsetof is defined in such a way that we can just
157935183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    // directly Evaluate it as an l-value.
1580efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
158135183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    if (!EvaluateLValue(E->getSubExpr(), LV, Info))
15828ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
158335183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    if (LV.getLValueBase())
15848ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
1585a73058324197b7bdfd19307965954f626e26199dKen Dyck    return Success(LV.getLValueOffset().getQuantity(), E);
158635183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman  }
15878ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1588a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == UnaryOperator::LNot) {
1589a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // LNot's operand isn't necessarily an integer, so we handle it specially.
1590a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    bool bres;
1591a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1592a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      return false;
1593131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(!bres, E);
1594a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1595a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
15964fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar  // Only handle integral operations...
15972ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getSubExpr()->getType()->isIntegralOrEnumerationType())
15984fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar    return false;
15994fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar
160087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  // Get the operand value into 'Result'.
160187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!Visit(E->getSubExpr()))
160275a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    return false;
1603a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
160475a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  switch (E->getOpcode()) {
16054c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  default:
160675a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
160775a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // See C99 6.6p3.
16080e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
160975a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Extension:
16104c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // FIXME: Should extension allow i-c-e extension expressions in its scope?
16114c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // If so, we could clear the diagnostic ID.
16123f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
161375a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Plus:
16141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The result is always just the subexpr.
16153f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
161675a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Minus:
161730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
161830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(-Result.getInt(), E);
161975a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Not:
162030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
162130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(~Result.getInt(), E);
162206a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson  }
1623a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
16241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1625732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// HandleCast - This is used to evaluate implicit or explicit casts where the
1626732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// result type is integer.
1627a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarbool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
162882206e267ce6cc709797127616f64672d255b310Anders Carlsson  Expr *SubExpr = E->getSubExpr();
162982206e267ce6cc709797127616f64672d255b310Anders Carlsson  QualType DestType = E->getType();
1630b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  QualType SrcType = SubExpr->getType();
163182206e267ce6cc709797127616f64672d255b310Anders Carlsson
16324efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (DestType->isBooleanType()) {
16334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    bool BoolResult;
16344efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!HandleConversionToBool(SubExpr, BoolResult, Info))
16354efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1636131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(BoolResult, E);
16374efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
16384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1639a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  // Handle simple integer->integer casts.
16402ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SrcType->isIntegralOrEnumerationType()) {
1641732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner    if (!Visit(SubExpr))
1642b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
1643a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1644be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!Result.isInt()) {
1645be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      // Only allow casts of lvalues if they are lossless.
1646be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
1647be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    }
164830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
1649dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType,
165030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar                                      Result.getInt(), Info.Ctx), E);
1651732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  }
16521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1653732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  // FIXME: Clean this up!
1654b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (SrcType->isPointerType()) {
1655efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
165687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (!EvaluatePointer(SubExpr, LV, Info))
1657b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
16584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1659dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    if (LV.getLValueBase()) {
1660dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      // Only allow based lvalue casts if they are lossless.
1661dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
1662dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar        return false;
1663dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar
1664efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LV.moveInto(Result);
1665dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      return true;
1666dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    }
16674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1668a73058324197b7bdfd19307965954f626e26199dKen Dyck    APSInt AsInt = Info.Ctx.MakeIntValue(LV.getLValueOffset().getQuantity(),
1669a73058324197b7bdfd19307965954f626e26199dKen Dyck                                         SrcType);
1670dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType, AsInt, Info.Ctx), E);
16712bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
16724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1673be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  if (SrcType->isArrayType() || SrcType->isFunctionType()) {
1674be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // This handles double-conversion cases, where there's both
1675be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // an l-value promotion and an implicit conversion to int.
1676efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
1677be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!EvaluateLValue(SubExpr, LV, Info))
1678be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1679be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1680be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(Info.Ctx.VoidPtrTy))
1681be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1682be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1683efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result);
1684be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    return true;
1685be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  }
1686be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
16871725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  if (SrcType->isAnyComplexType()) {
1688f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue C;
16891725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (!EvaluateComplex(SubExpr, C, Info))
16901725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return false;
16911725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (C.isComplexFloat())
16921725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleFloatToIntCast(DestType, SrcType,
16931725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                          C.getComplexFloatReal(), Info.Ctx),
16941725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                     E);
16951725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    else
16961725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleIntToIntCast(DestType, SrcType,
16971725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                        C.getComplexIntReal(), Info.Ctx), E);
16981725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  }
16992217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle vectors
17002217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1701b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (!SrcType->isRealFloatingType())
17020e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1703732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1704d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat F(0.0);
1705d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(SubExpr, F, Info))
17060e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
17071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1708b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
1709a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
17102bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
1711722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedmanbool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1712722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
1713f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1714722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1715722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1716722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntReal(), E);
1717722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1718722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1719722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  return Visit(E->getSubExpr());
1720722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman}
1721722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1722664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedmanbool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1723722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isComplexIntegerType()) {
1724f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1725722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1726722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1727722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntImag(), E);
1728722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1729722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1730664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
1731664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    Info.EvalResult.HasSideEffects = true;
1732664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  return Success(0, E);
1733664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman}
1734664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
1735f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1736d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman// Float Evaluation
1737d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1738d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1739d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmannamespace {
1740770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass FloatExprEvaluator
1741d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  : public StmtVisitor<FloatExprEvaluator, bool> {
1742d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  EvalInfo &Info;
1743d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat &Result;
1744d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanpublic:
1745d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  FloatExprEvaluator(EvalInfo &info, APFloat &result)
1746d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    : Info(info), Result(result) {}
1747d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1748d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitStmt(Stmt *S) {
1749d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1750d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1751d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1752d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
1753019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  bool VisitCallExpr(const CallExpr *E);
1754d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
17555db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  bool VisitUnaryOperator(const UnaryOperator *E);
1756d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
1757d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitFloatingLiteral(const FloatingLiteral *E);
17584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCastExpr(CastExpr *E);
1759ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
176067f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool VisitConditionalOperator(ConditionalOperator *E);
17612217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1762ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
1763ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
1764ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
1765ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
1766abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryReal(const UnaryOperator *E);
1767abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryImag(const UnaryOperator *E);
1768ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
1769abd3a857ace59100305790545d1baae5877b8945John McCall  // FIXME: Missing: array subscript of vector, member of vector,
1770abd3a857ace59100305790545d1baae5877b8945John McCall  //                 ImplicitValueInitExpr
1771d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman};
1772d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman} // end anonymous namespace
1773d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1774d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
17757db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isRealFloatingType());
1776d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
1777d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1778d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1779db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCallstatic bool TryEvaluateBuiltinNaN(ASTContext &Context,
1780db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  QualType ResultTy,
1781db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  const Expr *Arg,
1782db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  bool SNaN,
1783db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  llvm::APFloat &Result) {
1784db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
1785db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (!S) return false;
1786db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1787db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
1788db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1789db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  llvm::APInt fill;
1790db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1791db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  // Treat empty strings as if they were zero.
1792db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (S->getString().empty())
1793db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    fill = llvm::APInt(32, 0);
1794db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else if (S->getString().getAsInteger(0, fill))
1795db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return false;
1796db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1797db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (SNaN)
1798db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getSNaN(Sem, false, &fill);
1799db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else
1800db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getQNaN(Sem, false, &fill);
1801db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  return true;
1802db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall}
1803db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1804019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattnerbool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
18053c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
180634a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner  default: return false;
1807019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_val:
1808019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_valf:
1809019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_vall:
1810019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inf:
1811019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inff:
18127cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  case Builtin::BI__builtin_infl: {
18137cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar    const llvm::fltSemantics &Sem =
18147cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar      Info.Ctx.getFloatTypeSemantics(E->getType());
181534a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    Result = llvm::APFloat::getInf(Sem);
181634a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    return true;
18177cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  }
18181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1819db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nans:
1820db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansf:
1821db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansl:
1822db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1823db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 true, Result);
1824db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
18259e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nan:
18269e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanf:
18279e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanl:
18284572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump    // If this is __builtin_nan() turn this into a nan, otherwise we
18299e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // can't constant fold it.
1830db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1831db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 false, Result);
18325db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18335db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabs:
18345db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsf:
18355db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsl:
18365db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info))
18375db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18395db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (Result.isNegative())
18405db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      Result.changeSign();
18415db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18425db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysign:
18441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysignf:
18455db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignl: {
18465db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    APFloat RHS(0.);
18475db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info) ||
18485db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar        !EvaluateFloat(E->getArg(1), RHS, Info))
18495db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18505db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.copySign(RHS);
18515db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18525db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
1853019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
1854019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner}
1855019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1856abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1857abd3a857ace59100305790545d1baae5877b8945John McCall  ComplexValue CV;
1858abd3a857ace59100305790545d1baae5877b8945John McCall  if (!EvaluateComplex(E->getSubExpr(), CV, Info))
1859abd3a857ace59100305790545d1baae5877b8945John McCall    return false;
1860abd3a857ace59100305790545d1baae5877b8945John McCall  Result = CV.FloatReal;
1861abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1862abd3a857ace59100305790545d1baae5877b8945John McCall}
1863abd3a857ace59100305790545d1baae5877b8945John McCall
1864abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1865abd3a857ace59100305790545d1baae5877b8945John McCall  ComplexValue CV;
1866abd3a857ace59100305790545d1baae5877b8945John McCall  if (!EvaluateComplex(E->getSubExpr(), CV, Info))
1867abd3a857ace59100305790545d1baae5877b8945John McCall    return false;
1868abd3a857ace59100305790545d1baae5877b8945John McCall  Result = CV.FloatImag;
1869abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1870abd3a857ace59100305790545d1baae5877b8945John McCall}
1871abd3a857ace59100305790545d1baae5877b8945John McCall
18725db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbarbool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
1873a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes  if (E->getOpcode() == UnaryOperator::Deref)
1874a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes    return false;
1875a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes
18765db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  if (!EvaluateFloat(E->getSubExpr(), Result, Info))
18775db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return false;
18785db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18795db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  switch (E->getOpcode()) {
18805db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  default: return false;
18811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case UnaryOperator::Plus:
18825db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18835db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case UnaryOperator::Minus:
18845db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.changeSign();
18855db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18865db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
18875db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar}
1888019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1889d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
18907f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  if (E->getOpcode() == BinaryOperator::Comma) {
18917f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!EvaluateFloat(E->getRHS(), Result, Info))
18927f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      return false;
18937f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
18947f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
18957f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // conservatively mark it.
18967f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
18977f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      Info.EvalResult.HasSideEffects = true;
18987f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
18997f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    return true;
19007f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  }
19017f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
1902d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // FIXME: Diagnostics?  I really don't understand how the warnings
1903d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // and errors are supposed to work.
19045db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  APFloat RHS(0.0);
1905d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getLHS(), Result, Info))
1906d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1907d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getRHS(), RHS, Info))
1908d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1909d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1910d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  switch (E->getOpcode()) {
1911d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  default: return false;
1912d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Mul:
1913d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.multiply(RHS, APFloat::rmNearestTiesToEven);
1914d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1915d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Add:
1916d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.add(RHS, APFloat::rmNearestTiesToEven);
1917d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1918d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Sub:
1919d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.subtract(RHS, APFloat::rmNearestTiesToEven);
1920d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1921d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Div:
1922d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.divide(RHS, APFloat::rmNearestTiesToEven);
1923d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1924d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1925d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1926d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1927d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
1928d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  Result = E->getValue();
1929d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return true;
1930d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1931d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
19324efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCastExpr(CastExpr *E) {
19334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* SubExpr = E->getSubExpr();
19341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19352ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SubExpr->getType()->isIntegralOrEnumerationType()) {
19364efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
19373f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    if (!EvaluateInteger(SubExpr, IntResult, Info))
19384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
19391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
1940a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                  IntResult, Info.Ctx);
19414efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19434efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
19444efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!Visit(SubExpr))
19454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1946a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
1947a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    Result, Info.Ctx);
19484efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19494efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19502217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle complex types
19514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
19524efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
19534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19544efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1955ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregorbool FloatExprEvaluator::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19564efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
19574efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return true;
19584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19594efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
196067f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedmanbool FloatExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
196167f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool Cond;
196267f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  if (!HandleConversionToBool(E->getCond(), Cond, Info))
196367f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman    return false;
196467f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
196567f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
196667f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman}
196767f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
1968d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1969a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar// Complex Evaluation (for float and integer)
19709ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
19719ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
19729ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonnamespace {
1973770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass ComplexExprEvaluator
1974f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  : public StmtVisitor<ComplexExprEvaluator, bool> {
19759ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  EvalInfo &Info;
1976f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue &Result;
19771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19789ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonpublic:
1979f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
1980f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    : Info(info), Result(Result) {}
19811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19829ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
19839ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //                            Visitor Methods
19849ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
19859ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1986f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitStmt(Stmt *S) {
1987f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
19889ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
19891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1990f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
19919ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1992f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitImaginaryLiteral(ImaginaryLiteral *E) {
1993a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    Expr* SubExpr = E->getSubExpr();
1994a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1995a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (SubExpr->getType()->isRealFloatingType()) {
1996f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.makeComplexFloat();
1997f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APFloat &Imag = Result.FloatImag;
1998f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateFloat(SubExpr, Imag, Info))
1999f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2001f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.FloatReal = APFloat(Imag.getSemantics());
2002f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2003a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
20041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      assert(SubExpr->getType()->isIntegerType() &&
2005a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar             "Unexpected imaginary literal.");
2006a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
2007f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.makeComplexInt();
2008f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APSInt &Imag = Result.IntImag;
2009f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateInteger(SubExpr, Imag, Info))
2010f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2012f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
2013f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2014a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
20159ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
20169ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2017f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitCastExpr(CastExpr *E) {
2018ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    Expr* SubExpr = E->getSubExpr();
2019183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall    QualType EltType = E->getType()->getAs<ComplexType>()->getElementType();
2020a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    QualType SubType = SubExpr->getType();
2021ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2022a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    if (SubType->isRealFloatingType()) {
2023f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APFloat &Real = Result.FloatReal;
2024f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateFloat(SubExpr, Real, Info))
2025f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20261725f683432715e5afe34d476024bd6f16eac3fcEli Friedman
20271725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2028f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2029f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Real = HandleFloatToFloatCast(EltType, SubType, Real, Info.Ctx);
2030f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatImag = APFloat(Real.getSemantics());
2031f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20321725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2033f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2034f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntReal = HandleFloatToIntCast(EltType, SubType, Real, Info.Ctx);
2035f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntImag = APSInt(Result.IntReal.getBitWidth(),
2036f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                !Result.IntReal.isSigned());
2037f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20381725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2039a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    } else if (SubType->isIntegerType()) {
2040f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APSInt &Real = Result.IntReal;
2041f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateInteger(SubExpr, Real, Info))
2042f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
2043a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
20441725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2045f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2046f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatReal
2047f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          = HandleIntToFloatCast(EltType, SubType, Real, Info.Ctx);
2048f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatImag = APFloat(Result.FloatReal.getSemantics());
2049f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20501725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2051f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2052f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Real = HandleIntToIntCast(EltType, SubType, Real, Info.Ctx);
2053f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
2054f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20551725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2056183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall    } else if (const ComplexType *CT = SubType->getAs<ComplexType>()) {
2057f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!Visit(SubExpr))
2058f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
2059a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2060a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      QualType SrcType = CT->getElementType();
2061a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2062f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (Result.isComplexFloat()) {
2063a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
2064f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexFloat();
2065f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatReal = HandleFloatToFloatCast(EltType, SrcType,
2066f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Result.FloatReal,
2067f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Info.Ctx);
2068f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatImag = HandleFloatToFloatCast(EltType, SrcType,
2069f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Result.FloatImag,
2070f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Info.Ctx);
2071f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2072a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
2073f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexInt();
2074f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntReal = HandleFloatToIntCast(EltType, SrcType,
2075f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Result.FloatReal,
2076f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Info.Ctx);
2077f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntImag = HandleFloatToIntCast(EltType, SrcType,
2078f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Result.FloatImag,
2079f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Info.Ctx);
2080f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2081a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
2082a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      } else {
2083f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        assert(Result.isComplexInt() && "Invalid evaluate result.");
2084a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
2085f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexFloat();
2086f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatReal = HandleIntToFloatCast(EltType, SrcType,
2087f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Result.IntReal,
2088f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Info.Ctx);
2089f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatImag = HandleIntToFloatCast(EltType, SrcType,
2090f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Result.IntImag,
2091f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Info.Ctx);
2092f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2093a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
2094f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexInt();
2095f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntReal = HandleIntToIntCast(EltType, SrcType,
2096f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Result.IntReal,
2097f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Info.Ctx);
2098f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntImag = HandleIntToIntCast(EltType, SrcType,
2099f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Result.IntImag,
2100f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Info.Ctx);
2101f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2102a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
2103a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      }
2104ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    }
2105ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2106ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    // FIXME: Handle more casts.
2107f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2108ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
21091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2110f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitBinaryOperator(const BinaryOperator *E);
2111f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitChooseExpr(const ChooseExpr *E)
2112ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
2113f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
2114ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
2115ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME Missing: unary +/-/~, binary div, ImplicitValueInitExpr,
21162217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  //                conditional ?:, comma
21179ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson};
21189ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson} // end anonymous namespace
21199ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2120f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallstatic bool EvaluateComplex(const Expr *E, ComplexValue &Result,
2121f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                            EvalInfo &Info) {
21227db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isAnyComplexType());
2123f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return ComplexExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
21249ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson}
21259ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2126f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallbool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
2127f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  if (!Visit(E->getLHS()))
2128f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
21291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2130f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue RHS;
2131a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getRHS(), RHS, Info))
2132f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2133a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
21343f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
21353f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         "Invalid operands to binary operator.");
2136ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  switch (E->getOpcode()) {
2137f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  default: return false;
2138ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Add:
2139a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2140a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
2141a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2142a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
2143a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2144a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2145a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() += RHS.getComplexIntReal();
2146a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() += RHS.getComplexIntImag();
2147a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21483f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
2149ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Sub:
2150a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2151a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
2152a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2153a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
2154a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2155a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2156a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() -= RHS.getComplexIntReal();
2157a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() -= RHS.getComplexIntImag();
2158a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21593f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
21603f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  case BinaryOperator::Mul:
21613f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    if (Result.isComplexFloat()) {
2162f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
21633f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_r = LHS.getComplexFloatReal();
21643f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_i = LHS.getComplexFloatImag();
21653f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_r = RHS.getComplexFloatReal();
21663f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_i = RHS.getComplexFloatImag();
21671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
21683f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat Tmp = LHS_r;
21693f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
21703f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal() = Tmp;
21713f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
21723f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
21733f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
21743f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
21753f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_r;
21763f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
21773f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag() = Tmp;
21783f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
21793f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
21803f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
21813f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    } else {
2182f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
21831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntReal() =
21843f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
21853f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntImag());
21861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntImag() =
21873f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
21883f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntReal());
21893f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    }
21903f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
2191ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
2192ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2193f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return true;
2194ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson}
2195ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
21969ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
21976ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner// Top level Expr::Evaluate method.
2198f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
2199f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
220042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Evaluate - Return true if this is a constant which we can fold using
220142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// any crazy technique (that has nothing to do with language standards) that
220242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// we want to.  If this function returns true, it returns the folded constant
220342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// in Result.
220442c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool Expr::Evaluate(EvalResult &Result, ASTContext &Ctx) const {
220542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *E = this;
220642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo Info(Ctx, Result);
2207efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (E->getType()->isVectorType()) {
2208efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateVector(E, Info.EvalResult.Val, Info))
220959b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return false;
2210efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isIntegerType()) {
2211efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!IntExprEvaluator(Info, Info.EvalResult.Val).Visit(const_cast<Expr*>(E)))
22126dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22130f2b692bb10be35fdc60d0a72a847bdd73124670John McCall    if (Result.Val.isLValue() && !IsGlobalLValue(Result.Val.getLValueBase()))
22140f2b692bb10be35fdc60d0a72a847bdd73124670John McCall      return false;
2215efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->hasPointerRepresentation()) {
2216efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
2217efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E, LV, Info))
22186dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
2219e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    if (!IsGlobalLValue(LV.Base))
222042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return false;
2221efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Info.EvalResult.Val);
2222efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isRealFloatingType()) {
2223efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    llvm::APFloat F(0.0);
2224efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateFloat(E, F, Info))
22256dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2227efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info.EvalResult.Val = APValue(F);
2228efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isAnyComplexType()) {
2229efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    ComplexValue C;
2230efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateComplex(E, C, Info))
2231660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump      return false;
2232efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    C.moveInto(Info.EvalResult.Val);
2233660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  } else
2234660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump    return false;
2235660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2236660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  return true;
2237660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump}
2238660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2239cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallbool Expr::EvaluateAsBooleanCondition(bool &Result, ASTContext &Ctx) const {
2240cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalResult Scratch;
2241cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalInfo Info(Ctx, Scratch);
2242cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
2243cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  return HandleConversionToBool(this, Result, Info);
2244cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall}
2245cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
22461b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlssonbool Expr::EvaluateAsLValue(EvalResult &Result, ASTContext &Ctx) const {
22471b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson  EvalInfo Info(Ctx, Result);
22481b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson
2249efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue LV;
225042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (EvaluateLValue(this, LV, Info) &&
225142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !Result.HasSideEffects &&
2252e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara      IsGlobalLValue(LV.Base)) {
2253e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    LV.moveInto(Result.Val);
2254e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    return true;
2255e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  }
2256e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  return false;
2257e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara}
2258e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2259e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarabool Expr::EvaluateAsAnyLValue(EvalResult &Result, ASTContext &Ctx) const {
2260e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  EvalInfo Info(Ctx, Result);
2261e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2262e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  LValue LV;
2263e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (EvaluateLValue(this, LV, Info)) {
2264efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result.Val);
2265efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
2266efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
2267efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
2268b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman}
2269b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman
22706ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// isEvaluatable - Call Evaluate to see if this expression can be constant
227145b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner/// folded, but discard the result.
227245b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattnerbool Expr::isEvaluatable(ASTContext &Ctx) const {
22734fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  EvalResult Result;
22744fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  return Evaluate(Result, Ctx) && !Result.HasSideEffects;
227545b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner}
227651fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
2277393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanianbool Expr::HasSideEffects(ASTContext &Ctx) const {
2278393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  Expr::EvalResult Result;
2279393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  EvalInfo Info(Ctx, Result);
2280393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  return HasSideEffect(Info).Visit(const_cast<Expr*>(this));
2281393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian}
2282393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian
228351fe996231b1d7199f76e4005ff4c943d5deeecdAnders CarlssonAPSInt Expr::EvaluateAsInt(ASTContext &Ctx) const {
22841c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  EvalResult EvalResult;
22851c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  bool Result = Evaluate(EvalResult, Ctx);
2286f185319f25efd6094870f287030270fad26085baDaniel Dunbar  Result = Result;
228751fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson  assert(Result && "Could not evaluate expression");
22881c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
228951fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
22901c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  return EvalResult.Val.getInt();
229151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson}
2292d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2293e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara bool Expr::EvalResult::isGlobalLValue() const {
2294e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   assert(Val.isLValue());
2295e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   return IsGlobalLValue(Val.getLValueBase());
2296e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara }
2297e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2298e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2299d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// isIntegerConstantExpr - this recursive routine will test if an expression is
2300d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// an integer constant expression.
2301d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2302d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
2303d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// comma, etc
2304d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall///
2305d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Handle offsetof.  Two things to do:  Handle GCC's __builtin_offsetof
2306d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// to support gcc 4.0+  and handle the idiom GCC recognizes with a null pointer
2307d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// cast+dereference.
2308d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2309d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// CheckICE - This function does the fundamental ICE checking: the returned
2310d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// ICEDiag contains a Val of 0, 1, or 2, and a possibly null SourceLocation.
2311d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Note that to reduce code duplication, this helper does no evaluation
2312d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// itself; the caller checks whether the expression is evaluatable, and
2313d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// in the rare cases where CheckICE actually cares about the evaluated
2314d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// value, it calls into Evalute.
2315d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//
2316d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Meanings of Val:
2317d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 0: This expression is an ICE if it can be evaluated by Evaluate.
2318d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 1: This expression is not an ICE, but if it isn't evaluated, it's
2319d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    a legal subexpression for an ICE. This return value is used to handle
2320d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    the comma operator in C99 mode.
2321d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 2: This expression is not an ICE, and is not a legal subexpression for one.
2322d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2323d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstruct ICEDiag {
2324d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  unsigned Val;
2325d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  SourceLocation Loc;
2326d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2327d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  public:
2328d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag(unsigned v, SourceLocation l) : Val(v), Loc(l) {}
2329d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag() : Val(0) {}
2330d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall};
2331d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2332d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallICEDiag NoDiag() { return ICEDiag(); }
2333d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2334d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckEvalInICE(const Expr* E, ASTContext &Ctx) {
2335d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Expr::EvalResult EVResult;
2336d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2337d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      !EVResult.Val.isInt()) {
2338d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2339d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2340d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return NoDiag();
2341d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2342d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2343d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckICE(const Expr* E, ASTContext &Ctx) {
2344d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!E->isValueDependent() && "Should not see value dependent exprs!");
23452ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getType()->isIntegralOrEnumerationType()) {
2346d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2347d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2348d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2349d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  switch (E->getStmtClass()) {
2350d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define STMT(Node, Base) case Expr::Node##Class:
2351d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define EXPR(Node, Base)
2352d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#include "clang/AST/StmtNodes.inc"
2353d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::PredefinedExprClass:
2354d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::FloatingLiteralClass:
2355d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImaginaryLiteralClass:
2356d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StringLiteralClass:
2357d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ArraySubscriptExprClass:
2358d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::MemberExprClass:
2359d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundAssignOperatorClass:
2360d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundLiteralExprClass:
2361d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ExtVectorElementExprClass:
2362d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::InitListExprClass:
2363d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DesignatedInitExprClass:
2364d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitValueInitExprClass:
2365d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenListExprClass:
2366d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::VAArgExprClass:
2367d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::AddrLabelExprClass:
2368d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StmtExprClass:
2369d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXMemberCallExprClass:
2370d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDynamicCastExprClass:
2371d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTypeidExprClass:
2372d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNullPtrLiteralExprClass:
2373d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThisExprClass:
2374d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThrowExprClass:
2375d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNewExprClass:
2376d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDeleteExprClass:
2377d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXPseudoDestructorExprClass:
2378d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedLookupExprClass:
2379d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DependentScopeDeclRefExprClass:
2380d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstructExprClass:
2381d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindTemporaryExprClass:
2382d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindReferenceExprClass:
2383d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXExprWithTemporariesClass:
2384d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTemporaryObjectExprClass:
2385d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXUnresolvedConstructExprClass:
2386d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDependentScopeMemberExprClass:
2387d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedMemberExprClass:
2388d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCStringLiteralClass:
2389d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCEncodeExprClass:
2390d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCMessageExprClass:
2391d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSelectorExprClass:
2392d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCProtocolExprClass:
2393d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIvarRefExprClass:
2394d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCPropertyRefExprClass:
2395d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCImplicitSetterGetterRefExprClass:
2396d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSuperExprClass:
2397d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIsaExprClass:
2398d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ShuffleVectorExprClass:
2399d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockExprClass:
2400d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockDeclRefExprClass:
2401d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::NoStmtClass:
2402d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2403d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2404d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::GNUNullExprClass:
2405d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // GCC considers the GNU __null value to be an integral constant expression.
2406d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2407d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2408d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenExprClass:
2409d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
2410d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::IntegerLiteralClass:
2411d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CharacterLiteralClass:
2412d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBoolLiteralExprClass:
2413ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  case Expr::CXXScalarValueInitExprClass:
2414d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::TypesCompatibleExprClass:
2415d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryTypeTraitExprClass:
2416d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2417d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CallExprClass:
2418d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXOperatorCallExprClass: {
2419d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const CallExpr *CE = cast<CallExpr>(E);
2420d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CE->isBuiltinCall(Ctx))
2421d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2422d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2423d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2424d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DeclRefExprClass:
2425d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<EnumConstantDecl>(cast<DeclRefExpr>(E)->getDecl()))
2426d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2427d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Ctx.getLangOptions().CPlusPlus &&
2428d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        E->getType().getCVRQualifiers() == Qualifiers::Const) {
2429d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
2430d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2431d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Parameter variables are never constants.  Without this check,
2432d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // getAnyInitializer() can find a default argument, which leads
2433d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // to chaos.
2434d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (isa<ParmVarDecl>(D))
2435d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2436d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2437d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // C++ 7.1.5.1p2
2438d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   A variable of non-volatile const-qualified integral or enumeration
2439d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   type initialized by an ICE can be used in ICEs.
2440d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (const VarDecl *Dcl = dyn_cast<VarDecl>(D)) {
2441d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Qualifiers Quals = Ctx.getCanonicalType(Dcl->getType()).getQualifiers();
2442d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Quals.hasVolatile() || !Quals.hasConst())
2443d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2444d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2445d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Look for a declaration of this variable that has an initializer.
2446d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const VarDecl *ID = 0;
2447d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const Expr *Init = Dcl->getAnyInitializer(ID);
2448d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Init) {
2449d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (ID->isInitKnownICE()) {
2450d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // We have already checked whether this subexpression is an
2451d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // integral constant expression.
2452d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (ID->isInitICE())
2453d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return NoDiag();
2454d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            else
2455d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2456d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2457d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2458d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // It's an ICE whether or not the definition we found is
2459d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // out-of-line.  See DR 721 and the discussion in Clang PR
2460d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // 6206 for details.
2461d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2462d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (Dcl->isCheckingICE()) {
2463d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2464d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2465d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2466d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setCheckingICE();
2467d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          ICEDiag Result = CheckICE(Init, Ctx);
2468d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // Cache the result of the ICE test.
2469d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setInitKnownICE(Result.Val == 0);
2470d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return Result;
2471d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2472d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2473d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2474d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2475d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryOperatorClass: {
2476d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const UnaryOperator *Exp = cast<UnaryOperator>(E);
2477d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
2478d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PostInc:
2479d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PostDec:
2480d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PreInc:
2481d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PreDec:
2482d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::AddrOf:
2483d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Deref:
2484d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2485d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Extension:
2486d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::LNot:
2487d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Plus:
2488d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Minus:
2489d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Not:
2490d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Real:
2491d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Imag:
2492d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(Exp->getSubExpr(), Ctx);
2493d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::OffsetOf:
2494d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      break;
2495d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2496d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2497d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // OffsetOf falls through here.
2498d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2499d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::OffsetOfExprClass: {
2500d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Note that per C99, offsetof must be an ICE. And AFAIK, using
2501d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Evaluate matches the proposed gcc behavior for cases like
2502d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // "offsetof(struct s{int x[4];}, x[!.0])".  This doesn't affect
2503d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // compliance: we should warn earlier for offsetof expressions with
2504d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // array subscripts that aren't ICEs, and if the array subscripts
2505d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // are ICEs, the value of the offsetof must be an integer constant.
2506d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2507d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2508d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::SizeOfAlignOfExprClass: {
2509d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const SizeOfAlignOfExpr *Exp = cast<SizeOfAlignOfExpr>(E);
2510d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->isSizeOf() && Exp->getTypeOfArgument()->isVariableArrayType())
2511d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2512d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2513d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2514d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BinaryOperatorClass: {
2515d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const BinaryOperator *Exp = cast<BinaryOperator>(E);
2516d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
2517d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::PtrMemD:
2518d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::PtrMemI:
2519d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Assign:
2520d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::MulAssign:
2521d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::DivAssign:
2522d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::RemAssign:
2523d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::AddAssign:
2524d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::SubAssign:
2525d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::ShlAssign:
2526d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::ShrAssign:
2527d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::AndAssign:
2528d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::XorAssign:
2529d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::OrAssign:
2530d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2531d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2532d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Mul:
2533d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Div:
2534d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Rem:
2535d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Add:
2536d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Sub:
2537d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Shl:
2538d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Shr:
2539d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LT:
2540d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::GT:
2541d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LE:
2542d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::GE:
2543d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::EQ:
2544d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::NE:
2545d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::And:
2546d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Xor:
2547d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Or:
2548d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Comma: {
2549d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2550d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2551d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (Exp->getOpcode() == BinaryOperator::Div ||
2552d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Exp->getOpcode() == BinaryOperator::Rem) {
2553d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Evaluate gives an error for undefined Div/Rem, so make sure
2554d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // we don't evaluate one.
2555d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (LHSResult.Val != 2 && RHSResult.Val != 2) {
2556d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          llvm::APSInt REval = Exp->getRHS()->EvaluateAsInt(Ctx);
2557d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval == 0)
2558d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2559d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval.isSigned() && REval.isAllOnesValue()) {
2560d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            llvm::APSInt LEval = Exp->getLHS()->EvaluateAsInt(Ctx);
2561d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (LEval.isMinSignedValue())
2562d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(1, E->getLocStart());
2563d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2564d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2565d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2566d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (Exp->getOpcode() == BinaryOperator::Comma) {
2567d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Ctx.getLangOptions().C99) {
2568d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
2569d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // if it isn't evaluated.
2570d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (LHSResult.Val == 0 && RHSResult.Val == 0)
2571d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2572d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        } else {
2573d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // In both C89 and C++, commas in ICEs are illegal.
2574d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2575d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2576d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2577d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2578d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2579d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2580d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2581d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LAnd:
2582d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LOr: {
2583d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2584d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2585d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val == 0 && RHSResult.Val == 1) {
2586d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Rare case where the RHS has a comma "side-effect"; we need
2587d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // to actually check the condition to see whether the side
2588d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // with the comma is evaluated.
2589d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if ((Exp->getOpcode() == BinaryOperator::LAnd) !=
2590d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            (Exp->getLHS()->EvaluateAsInt(Ctx) == 0))
2591d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return RHSResult;
2592d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2593d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2594d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2595d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2596d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2597d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2598d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2599d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2600d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2601d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitCastExprClass:
2602d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CStyleCastExprClass:
2603d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXFunctionalCastExprClass:
2604d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXStaticCastExprClass:
2605d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXReinterpretCastExprClass:
2606d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstCastExprClass: {
2607d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
26082ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType())
2609d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(SubExpr, Ctx);
2610d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<FloatingLiteral>(SubExpr->IgnoreParens()))
2611d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2612d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2613d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2614d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ConditionalOperatorClass: {
2615d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
2616d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // If the condition (ignoring parens) is a __builtin_constant_p call,
2617d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // then only the true side is actually considered in an integer constant
2618d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // expression, and it is fully evaluated.  This is an important GNU
2619d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // extension.  See GCC PR38377 for discussion.
2620d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (const CallExpr *CallCE
2621d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
2622d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (CallCE->isBuiltinCall(Ctx) == Builtin::BI__builtin_constant_p) {
2623d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Expr::EvalResult EVResult;
2624d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2625d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            !EVResult.Val.isInt()) {
2626d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2627d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2628d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2629d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2630d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
2631d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
2632d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
2633d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 2)
2634d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2635d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 2)
2636d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return TrueResult;
2637d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (FalseResult.Val == 2)
2638d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2639d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 1)
2640d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2641d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 0 && FalseResult.Val == 0)
2642d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2643d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Rare case where the diagnostics depend on which side is evaluated
2644d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Note that if we get here, CondResult is 0, and at least one of
2645d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // TrueResult and FalseResult is non-zero.
2646d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->getCond()->EvaluateAsInt(Ctx) == 0) {
2647d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2648d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2649d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return TrueResult;
2650d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2651d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDefaultArgExprClass:
2652d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
2653d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ChooseExprClass: {
2654d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(Ctx), Ctx);
2655d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2656d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2657d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2658d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  // Silence a GCC warning
2659d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return ICEDiag(2, E->getLocStart());
2660d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2661d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2662d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallbool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
2663d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall                                 SourceLocation *Loc, bool isEvaluated) const {
2664d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag d = CheckICE(this, Ctx);
2665d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (d.Val != 0) {
2666d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Loc) *Loc = d.Loc;
2667d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return false;
2668d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2669d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  EvalResult EvalResult;
2670d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!Evaluate(EvalResult, Ctx))
2671d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    llvm_unreachable("ICE cannot be evaluated!");
2672d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!EvalResult.HasSideEffects && "ICE with side effects!");
2673d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(EvalResult.Val.isInt() && "ICE that isn't integer!");
2674d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Result = EvalResult.Val.getInt();
2675d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return true;
2676d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2677