ExprConstant.cpp revision 2e156225a29407a50dd19041aa5750171ad44ea3
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
3402de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case 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) {
4842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() != BO_Add &&
4852de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      E->getOpcode() != BO_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
5152de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_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
5352de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case 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
5642de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_NoOp:
5652de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_BitCast:
5662de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_LValueBitCast:
5672de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_AnyPointerToObjCPointerCast:
5682de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_AnyPointerToBlockPointerCast:
56909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    return Visit(SubExpr);
57009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
5712de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_IntegralToPointer: {
572efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    APValue Value;
573efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
57409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      break;
57569ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
576efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (Value.isInt()) {
577efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
578efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = 0;
579efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
580efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
581efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    } else {
582efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      // Cast is of an lvalue, no need to change value.
583efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Base = Value.getLValueBase();
584efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      Result.Offset = Value.getLValueOffset();
585efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return true;
586650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson    }
587650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
5882de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_ArrayToPointerDecay:
5892de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case CK_FunctionToPointerDecay:
590efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return EvaluateLValue(SubExpr, Result, Info);
5914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
5924efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
593efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
5941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump}
595650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
596efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCallExpr(CallExpr *E) {
5971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (E->isBuiltinCall(Info.Ctx) ==
5980d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___CFStringMakeConstantString ||
5990d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall      E->isBuiltinCall(Info.Ctx) ==
6000d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall        Builtin::BI__builtin___NSStringMakeConstantString)
601efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
602efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
6033941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman}
6043941b18b8e441c8c466efecd557de60b9a32d10bEli Friedman
605efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
6064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool BoolResult;
6074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
608efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
6094efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
6104efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
611efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Visit(EvalExpr);
6124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
613f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
614f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
61559b5da6d853b4368b984700315adf7b37de05764Nate Begeman// Vector Evaluation
61659b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
61759b5da6d853b4368b984700315adf7b37de05764Nate Begeman
61859b5da6d853b4368b984700315adf7b37de05764Nate Begemannamespace {
619770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramer  class VectorExprEvaluator
62059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  : public StmtVisitor<VectorExprEvaluator, APValue> {
62159b5da6d853b4368b984700315adf7b37de05764Nate Begeman    EvalInfo &Info;
62291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue GetZeroVector(QualType VecType);
62359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  public:
6241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
62559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    VectorExprEvaluator(EvalInfo &info) : Info(info) {}
6261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
62759b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitStmt(Stmt *S) {
62859b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return APValue();
62959b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
6301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
63191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitParenExpr(ParenExpr *E)
63291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        { return Visit(E->getSubExpr()); }
63391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryExtension(const UnaryOperator *E)
63491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryPlus(const UnaryOperator *E)
63691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryReal(const UnaryOperator *E)
63891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return Visit(E->getSubExpr()); }
63991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E)
64091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      { return GetZeroVector(E->getType()); }
64159b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCastExpr(const CastExpr* E);
64259b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
64359b5da6d853b4368b984700315adf7b37de05764Nate Begeman    APValue VisitInitListExpr(const InitListExpr *E);
64491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitConditionalOperator(const ConditionalOperator *E);
645ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    APValue VisitChooseExpr(const ChooseExpr *E)
646ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
64791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    APValue VisitUnaryImag(const UnaryOperator *E);
64891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    // FIXME: Missing: unary -, unary ~, binary add/sub/mul/div,
6492217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman    //                 binary comparisons, binary and/or/xor,
65091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //                 shufflevector, ExtVectorElementExpr
65191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //        (Note that these require implementing conversions
65291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    //         between vector types.)
65359b5da6d853b4368b984700315adf7b37de05764Nate Begeman  };
65459b5da6d853b4368b984700315adf7b37de05764Nate Begeman} // end anonymous namespace
65559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
65659b5da6d853b4368b984700315adf7b37de05764Nate Begemanstatic bool EvaluateVector(const Expr* E, APValue& Result, EvalInfo &Info) {
65759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  if (!E->getType()->isVectorType())
65859b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return false;
65959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  Result = VectorExprEvaluator(Info).Visit(const_cast<Expr*>(E));
66059b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return !Result.isUninit();
66159b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
66259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
66359b5da6d853b4368b984700315adf7b37de05764Nate BegemanAPValue VectorExprEvaluator::VisitCastExpr(const CastExpr* E) {
664183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VTy = E->getType()->getAs<VectorType>();
665c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  QualType EltTy = VTy->getElementType();
666c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned NElts = VTy->getNumElements();
667c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  unsigned EltWidth = Info.Ctx.getTypeSize(EltTy);
6681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
66959b5da6d853b4368b984700315adf7b37de05764Nate Begeman  const Expr* SE = E->getSubExpr();
670e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  QualType SETy = SE->getType();
671e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  APValue Result = APValue();
67259b5da6d853b4368b984700315adf7b37de05764Nate Begeman
673e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  // Check for vector->vector bitcast and scalar->vector splat.
674e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  if (SETy->isVectorType()) {
67559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    return this->Visit(const_cast<Expr*>(SE));
676e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isIntegerType()) {
677e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APSInt IntResult;
678d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateInteger(SE, IntResult, Info))
679d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
680d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(IntResult);
681e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  } else if (SETy->isRealFloatingType()) {
682e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman    APFloat F(0.0);
683d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    if (!EvaluateFloat(SE, F, Info))
684d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar      return APValue();
685d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar    Result = APValue(F);
686d906dc7605b2d6352e16c27a5d3fb5f1a1b44ac8Daniel Dunbar  } else
687c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue();
68859b5da6d853b4368b984700315adf7b37de05764Nate Begeman
689c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to ExtVector, convert the scalar to the element type
690c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // and splat it to all elements.
691c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  if (E->getType()->isExtVectorType()) {
692c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType() && Result.isInt())
693c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToIntCast(EltTy, SETy, Result.getInt(),
694c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                          Info.Ctx));
695c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isIntegerType())
696c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToIntCast(EltTy, SETy, Result.getFloat(),
697c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
698c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType() && Result.isInt())
699c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleIntToFloatCast(EltTy, SETy, Result.getInt(),
700c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                            Info.Ctx));
701c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
702c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Result = APValue(HandleFloatToFloatCast(EltTy, SETy, Result.getFloat(),
703c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman                                              Info.Ctx));
704c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
705c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
706c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
707c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    // Splat and create vector APValue.
708c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    llvm::SmallVector<APValue, 4> Elts(NElts, Result);
709c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    return APValue(&Elts[0], Elts.size());
710e8c9e9218f215ec6089f12b076c7b9d310fd5194Nate Begeman  }
711c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
712c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // For casts of a scalar to regular gcc-style vector type, bitcast the scalar
713c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // to the vector. To construct the APValue vector initializer, bitcast the
714c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // initializing value to an APInt, and shift out the bits pertaining to each
715c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  // element.
716c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  APSInt Init;
717c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  Init = Result.isInt() ? Result.getInt() : Result.getFloat().bitcastToAPInt();
7181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
719c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  llvm::SmallVector<APValue, 4> Elts;
720c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  for (unsigned i = 0; i != NElts; ++i) {
721c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    APSInt Tmp = Init;
722c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    Tmp.extOrTrunc(EltWidth);
7231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
724c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    if (EltTy->isIntegerType())
725c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(Tmp));
726c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else if (EltTy->isRealFloatingType())
727c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      Elts.push_back(APValue(APFloat(Tmp)));
728c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    else
729c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman      return APValue();
730c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman
731c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman    Init >>= EltWidth;
732c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  }
733c0b8b19bd8056d6b5d831623a0825cce150f4507Nate Begeman  return APValue(&Elts[0], Elts.size());
73459b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
73559b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7361eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
73759b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
73859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return this->Visit(const_cast<Expr*>(E->getInitializer()));
73959b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
74059b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7411eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
74259b5da6d853b4368b984700315adf7b37de05764Nate BegemanVectorExprEvaluator::VisitInitListExpr(const InitListExpr *E) {
743183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = E->getType()->getAs<VectorType>();
74459b5da6d853b4368b984700315adf7b37de05764Nate Begeman  unsigned NumInits = E->getNumInits();
74591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  unsigned NumElements = VT->getNumElements();
7461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
74759b5da6d853b4368b984700315adf7b37de05764Nate Begeman  QualType EltTy = VT->getElementType();
74859b5da6d853b4368b984700315adf7b37de05764Nate Begeman  llvm::SmallVector<APValue, 4> Elements;
74959b5da6d853b4368b984700315adf7b37de05764Nate Begeman
750a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // If a vector is initialized with a single element, that value
751a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // becomes every element of the vector, not just the first.
752a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  // This is the behavior described in the IBM AltiVec documentation.
753a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  if (NumInits == 1) {
754a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    APValue InitValue;
75559b5da6d853b4368b984700315adf7b37de05764Nate Begeman    if (EltTy->isIntegerType()) {
75659b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APSInt sInt(32);
757a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateInteger(E->getInit(0), sInt, Info))
758a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
759a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(sInt);
76059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    } else {
76159b5da6d853b4368b984700315adf7b37de05764Nate Begeman      llvm::APFloat f(0.0);
762a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (!EvaluateFloat(E->getInit(0), f, Info))
763a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        return APValue();
764a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      InitValue = APValue(f);
765a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
766a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
767a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      Elements.push_back(InitValue);
768a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    }
769a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall  } else {
770a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall    for (unsigned i = 0; i < NumElements; i++) {
771a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall      if (EltTy->isIntegerType()) {
772a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APSInt sInt(32);
773a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
774a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateInteger(E->getInit(i), sInt, Info))
775a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
776a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
777a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          sInt = Info.Ctx.MakeIntValue(0, EltTy);
778a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
779a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(sInt));
78091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      } else {
781a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        llvm::APFloat f(0.0);
782a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        if (i < NumInits) {
783a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          if (!EvaluateFloat(E->getInit(i), f, Info))
784a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall            return APValue();
785a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        } else {
786a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall          f = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy));
787a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        }
788a7d6c221558fa8117aa1d1fd41cf0046c5c9fadeJohn McCall        Elements.push_back(APValue(f));
78991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman      }
79059b5da6d853b4368b984700315adf7b37de05764Nate Begeman    }
79159b5da6d853b4368b984700315adf7b37de05764Nate Begeman  }
79259b5da6d853b4368b984700315adf7b37de05764Nate Begeman  return APValue(&Elements[0], Elements.size());
79359b5da6d853b4368b984700315adf7b37de05764Nate Begeman}
79459b5da6d853b4368b984700315adf7b37de05764Nate Begeman
7951eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpAPValue
79691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanVectorExprEvaluator::GetZeroVector(QualType T) {
797183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const VectorType *VT = T->getAs<VectorType>();
79891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  QualType EltTy = VT->getElementType();
79991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue ZeroElement;
80091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EltTy->isIntegerType())
80191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement = APValue(Info.Ctx.MakeIntValue(0, EltTy));
80291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  else
80391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    ZeroElement =
80491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman        APValue(APFloat::getZero(Info.Ctx.getFloatTypeSemantics(EltTy)));
80591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
80691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  llvm::SmallVector<APValue, 4> Elements(VT->getNumElements(), ZeroElement);
80791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue(&Elements[0], Elements.size());
80891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
80991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
81191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  bool BoolResult;
81291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!HandleConversionToBool(E->getCond(), BoolResult, Info))
81391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return APValue();
81491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  Expr* EvalExpr = BoolResult ? E->getTrueExpr() : E->getFalseExpr();
81691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
81791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  APValue Result;
81891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (EvaluateVector(EvalExpr, Result, Info))
81991110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    return Result;
82091110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return APValue();
82191110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
82291110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82391110ee24e3475e0a3a38938c7b98439b5cf0b0eEli FriedmanAPValue VectorExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
82491110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
82591110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman    Info.EvalResult.HasSideEffects = true;
82691110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman  return GetZeroVector(E->getType());
82791110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman}
82891110ee24e3475e0a3a38938c7b98439b5cf0b0eEli Friedman
82959b5da6d853b4368b984700315adf7b37de05764Nate Begeman//===----------------------------------------------------------------------===//
830f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Integer Evaluation
831f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
832f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
833f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnernamespace {
834770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass IntExprEvaluator
835b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  : public StmtVisitor<IntExprEvaluator, bool> {
83687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
83730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue &Result;
838f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerpublic:
83930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  IntExprEvaluator(EvalInfo &info, APValue &result)
84087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    : Info(info), Result(result) {}
841f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
8423f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  bool Success(const llvm::APSInt &SI, const Expr *E) {
8432ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8442ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
84530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.isSigned() == E->getType()->isSignedIntegerType() &&
8463f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
84730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(SI.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8483f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
84930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(SI);
8503f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
8513f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
8523f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar
853131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(const llvm::APInt &I, const Expr *E) {
8542ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8552ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
85630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    assert(I.getBitWidth() == Info.Ctx.getIntWidth(E->getType()) &&
8573f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar           "Invalid evaluation result.");
85830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(APSInt(I));
85930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result.getInt().setIsUnsigned(E->getType()->isUnsignedIntegerType());
860131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
861131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
862131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
863131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  bool Success(uint64_t Value, const Expr *E) {
8642ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    assert(E->getType()->isIntegralOrEnumerationType() &&
8652ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor           "Invalid evaluation result.");
86630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    Result = APValue(Info.Ctx.MakeIntValue(Value, E->getType()));
867131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return true;
868131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  }
869131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
87082206e267ce6cc709797127616f64672d255b310Anders Carlsson  bool Error(SourceLocation L, diag::kind D, const Expr *E) {
87132fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    // Take the first error.
87254da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson    if (Info.EvalResult.Diag == 0) {
87354da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.DiagLoc = L;
87454da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson      Info.EvalResult.Diag = D;
87582206e267ce6cc709797127616f64672d255b310Anders Carlsson      Info.EvalResult.DiagExpr = E;
87632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    }
87754176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    return false;
8787a76778e218da57a3b10c80066a0a938f28987b6Chris Lattner  }
8791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
880f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
881f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //                            Visitor Methods
882f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  //===--------------------------------------------------------------------===//
8831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
88432fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitStmt(Stmt *) {
88532fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    assert(0 && "This should be called on integers, stmts are not integers");
88632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner    return false;
88732fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  }
8881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
88932fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  bool VisitExpr(Expr *E) {
8900e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
891f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner  }
8921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
893b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
894f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
8954c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitIntegerLiteral(const IntegerLiteral *E) {
896131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
8974c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
8984c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitCharacterLiteral(const CharacterLiteral *E) {
899131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9004c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
9014c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  bool VisitTypesCompatibleExpr(const TypesCompatibleExpr *E) {
902ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // Per gcc docs "this built-in function ignores top level
903ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // qualifiers".  We need to use the canonical version to properly
904ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    // be able to strip CRV qualifiers from the type.
905ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T0 = Info.Ctx.getCanonicalType(E->getArgType1());
906ac620decb68aad1a2cf6c0c191b56d78981d9aaaDaniel Dunbar    QualType T1 = Info.Ctx.getCanonicalType(E->getArgType2());
9071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return Success(Info.Ctx.typesAreCompatible(T0.getUnqualifiedType(),
908131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                                               T1.getUnqualifiedType()),
909131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                   E);
9104c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  }
911043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
912043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool CheckReferencedDecl(const Expr *E, const Decl *D);
913043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitDeclRefExpr(const DeclRefExpr *E) {
914043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return CheckReferencedDecl(E, E->getDecl());
915043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
916043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  bool VisitMemberExpr(const MemberExpr *E) {
917043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (CheckReferencedDecl(E, E->getMemberDecl())) {
918043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      // Conservatively assume a MemberExpr will have side-effects
919043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      Info.EvalResult.HasSideEffects = true;
920043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman      return true;
921043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    }
922043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    return false;
923043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman  }
924043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman
925c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  bool VisitCallExpr(CallExpr *E);
926b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitBinaryOperator(const BinaryOperator *E);
9278ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  bool VisitOffsetOfExpr(const OffsetOfExpr *E);
928b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner  bool VisitUnaryOperator(const UnaryOperator *E);
929ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes  bool VisitConditionalOperator(const ConditionalOperator *E);
930f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
931a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool VisitCastExpr(CastExpr* E);
9320518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  bool VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E);
9330518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
9343068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  bool VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
935131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getValue(), E);
9363068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9383f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  bool VisitGNUNullExpr(const GNUNullExpr *E) {
939131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9403f70456b8adb0405ef2a47d51f9fc2d5937ae8aeAnders Carlsson  }
9411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
942ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
943131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(0, E);
9443068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
9453068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
946664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
947664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    return Success(0, E);
948664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  }
949664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
95064b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  bool VisitUnaryTypeTraitExpr(const UnaryTypeTraitExpr *E) {
9515e03f9ea8174ae588c5e69ec6b5ef4c68f8fd766Douglas Gregor    return Success(E->EvaluateTrait(Info.Ctx), E);
95264b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl  }
95364b45f7e0d3167f040841ac2920aead7f080730dSebastian Redl
954ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E) {
955ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    return Visit(E->getChosenSubExpr(Info.Ctx));
956ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  }
957ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
958722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  bool VisitUnaryReal(const UnaryOperator *E);
959664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  bool VisitUnaryImag(const UnaryOperator *E);
960664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
961fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattnerprivate:
9628b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfExpr(const Expr *E);
9638b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  CharUnits GetAlignOfType(QualType T);
96442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  static QualType GetObjectType(const Expr *E);
96542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  bool TryEvaluateBuiltinObjectSize(CallExpr *E);
966664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  // FIXME: Missing: array subscript of vector, member of vector
967f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner};
968f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
969f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
97069ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateIntegerOrLValue(const Expr* E, APValue &Result, EvalInfo &Info) {
9712ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
97269ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  return IntExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
97369ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar}
97469ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar
97569ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbarstatic bool EvaluateInteger(const Expr* E, APSInt &Result, EvalInfo &Info) {
9762ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  assert(E->getType()->isIntegralOrEnumerationType());
9777db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall
97830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  APValue Val;
97969ab26a8623141f35e86817cfc6e0fbe7639a40fDaniel Dunbar  if (!EvaluateIntegerOrLValue(E, Val, Info) || !Val.isInt())
98030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
98130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  Result = Val.getInt();
98230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  return true;
983f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
984f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
985043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedmanbool IntExprEvaluator::CheckReferencedDecl(const Expr* E, const Decl* D) {
9864c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Enums are integer constant exprs.
98729a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman  if (const EnumConstantDecl *ECD = dyn_cast<EnumConstantDecl>(D))
98829a7f3342c3c6dd15d914c61ae22246c36d51ce7Eli Friedman    return Success(ECD->getInitVal(), E);
989b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
990b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  // In C++, const, non-volatile integers initialized with ICEs are ICEs.
991e1646da3bb0fe97b372e5fe8cefc537b22048fc4Eli Friedman  // In C, they can also be folded, although they are not ICEs.
992cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor  if (Info.Ctx.getCanonicalType(E->getType()).getCVRQualifiers()
993cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor                                                        == Qualifiers::Const) {
994f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
995f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson    if (isa<ParmVarDecl>(D))
996f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson      return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
997f6b6025a4e26bb0295a57e0efece73253038f278Anders Carlsson
998043097507f99b1156bfd8bad41e7d5166ae4b9b6Eli Friedman    if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
99931310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl      if (const Expr *Init = VD->getAnyInitializer()) {
1000c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (APValue *V = VD->getEvaluatedValue()) {
1001c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          if (V->isInt())
1002c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman            return Success(V->getInt(), E);
1003c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1004c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        }
1005c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1006c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        if (VD->isEvaluatingValue())
1007c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1008c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1009c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatingValue();
1010c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman
1011a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman        Expr::EvalResult EResult;
1012a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman        if (Init->Evaluate(EResult, Info.Ctx) && !EResult.HasSideEffects &&
1013a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman            EResult.Val.isInt()) {
101478d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          // Cache the evaluated value in the variable declaration.
1015a7dedf7004e8ac3f947d6817370b2778fa648c2bEli Friedman          Result = EResult.Val;
1016c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          VD->setEvaluatedValue(Result);
101778d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          return true;
101878d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        }
101978d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor
1020c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatedValue(APValue());
102178d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor      }
1022b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    }
1023b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  }
1024b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
10254c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Otherwise, random variable references are not constants.
10260e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson  return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
10274c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
10284c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner
1029a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
1030a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// as GCC.
1031a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattnerstatic int EvaluateBuiltinClassifyType(const CallExpr *E) {
1032a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // The following enum mimics the values returned by GCC.
10337c80bd64032e610c0dbd74fc0ef6ea334447f2fdSebastian Redl  // FIXME: Does GCC differ between lvalue and rvalue references here?
1034a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  enum gcc_type_class {
1035a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    no_type_class = -1,
1036a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    void_type_class, integer_type_class, char_type_class,
1037a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    enumeral_type_class, boolean_type_class,
1038a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    pointer_type_class, reference_type_class, offset_type_class,
1039a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    real_type_class, complex_type_class,
1040a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    function_type_class, method_type_class,
1041a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    record_type_class, union_type_class,
1042a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    array_type_class, string_type_class,
1043a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    lang_type_class
1044a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  };
10451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // If no argument was supplied, default to "no_type_class". This isn't
1047a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // ideal, however it is what gcc does.
1048a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (E->getNumArgs() == 0)
1049a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return no_type_class;
10501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1051a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  QualType ArgTy = E->getArg(0)->getType();
1052a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (ArgTy->isVoidType())
1053a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return void_type_class;
1054a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isEnumeralType())
1055a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return enumeral_type_class;
1056a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isBooleanType())
1057a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return boolean_type_class;
1058a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isCharType())
1059a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return string_type_class; // gcc doesn't appear to use char_type_class
1060a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isIntegerType())
1061a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return integer_type_class;
1062a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isPointerType())
1063a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return pointer_type_class;
1064a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isReferenceType())
1065a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return reference_type_class;
1066a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isRealType())
1067a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return real_type_class;
1068a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isComplexType())
1069a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return complex_type_class;
1070a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isFunctionType())
1071a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return function_type_class;
1072fb87b89fc9eb103e19fb8e4b925c23f0bd091b99Douglas Gregor  else if (ArgTy->isStructureOrClassType())
1073a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return record_type_class;
1074a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1075a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1076a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isArrayType())
1077a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return array_type_class;
1078a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1079a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1080a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else  // FIXME: offset_type_class, method_type_class, & lang_type_class?
1081a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
1082a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  return -1;
1083a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner}
1084a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
108542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Retrieves the "underlying object type" of the given expression,
108642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// as used by __builtin_object_size.
108742c8f87eb60958170c46767273bf93e6c96125bfJohn McCallQualType IntExprEvaluator::GetObjectType(const Expr *E) {
108842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
108942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
109042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->getType();
109142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  } else if (isa<CompoundLiteralExpr>(E)) {
109242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return E->getType();
109342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
109442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return QualType();
109642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
109742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109842c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool IntExprEvaluator::TryEvaluateBuiltinObjectSize(CallExpr *E) {
109942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // TODO: Perhaps we should let LLVM lower this?
110042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  LValue Base;
110142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!EvaluatePointer(E->getArg(0), Base, Info))
110242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
110342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // If we can prove the base is null, lower to zero now.
110542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *LVBase = Base.getLValueBase();
110642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!LVBase) return Success(0, E);
110742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  QualType T = GetObjectType(LVBase);
110942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (T.isNull() ||
111042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isIncompleteType() ||
11111357869bc5983cdfbc986db1f3d18265bb34cb0eEli Friedman      T->isFunctionType() ||
111242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isVariablyModifiedType() ||
111342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isDependentType())
111442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
111542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Size = Info.Ctx.getTypeSizeInChars(T);
111742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Offset = Base.getLValueOffset();
111842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!Offset.isNegative() && Offset <= Size)
112042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size -= Offset;
112142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  else
112242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size = CharUnits::Zero();
112342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return Success(Size.getQuantity(), E);
112442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
112542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1126c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedmanbool IntExprEvaluator::VisitCallExpr(CallExpr *E) {
11273c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
1128019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  default:
11290e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
113064eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
113164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  case Builtin::BI__builtin_object_size: {
113242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (TryEvaluateBuiltinObjectSize(E))
113342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
113464eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1135b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // If evaluating the argument has side-effects we can't determine
1136b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // the size of the object and lower it to unknown now.
1137393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian    if (E->getArg(0)->HasSideEffects(Info.Ctx)) {
11383f27b384801de26ce7efaa395699b42719372f24Benjamin Kramer      if (E->getArg(1)->EvaluateAsInt(Info.Ctx).getZExtValue() <= 1)
1139cf184655319cf7a5b811067cff9d26a5741fd161Chris Lattner        return Success(-1ULL, E);
114064eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump      return Success(0, E);
114164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    }
1142c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
114364eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
114464eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  }
114564eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1146019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_classify_type:
1147131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(EvaluateBuiltinClassifyType(E), E);
11481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11494bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson  case Builtin::BI__builtin_constant_p:
1150019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // __builtin_constant_p always has one operand: it returns true if that
1151019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // operand can be folded, false otherwise.
1152131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
115321fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner
115421fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  case Builtin::BI__builtin_eh_return_data_regno: {
115521fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    int Operand = E->getArg(0)->EvaluateAsInt(Info.Ctx).getZExtValue();
115621fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    Operand = Info.Ctx.Target.getEHDataRegisterNumber(Operand);
115721fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    return Success(Operand, E);
115821fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  }
1159c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman
1160c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  case Builtin::BI__builtin_expect:
1161c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman    return Visit(E->getArg(0));
11625726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11635726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor  case Builtin::BIstrlen:
11645726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor  case Builtin::BI__builtin_strlen:
11655726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    // As an extension, we support strlen() and __builtin_strlen() as constant
11665726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    // expressions when the argument is a string literal.
11675726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    if (StringLiteral *S
11685726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor               = dyn_cast<StringLiteral>(E->getArg(0)->IgnoreParenImpCasts())) {
11695726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      // The string literal may have embedded null characters. Find the first
11705726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      // one and truncate there.
11715726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      llvm::StringRef Str = S->getString();
11725726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      llvm::StringRef::size_type Pos = Str.find(0);
11735726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      if (Pos != llvm::StringRef::npos)
11745726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor        Str = Str.substr(0, Pos);
11755726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11765726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor      return Success(Str.size(), E);
11775726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    }
11785726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor
11795726d405e71f11feaaf0c8f518abe26e909537a4Douglas Gregor    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
1180019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
11814c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
1182f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
1183b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
11842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Comma) {
1185027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    if (!Visit(E->getRHS()))
1186027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson      return false;
11874fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
118833ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
118933ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // conservatively mark it.
119033ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
119133ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      Info.EvalResult.HasSideEffects = true;
1192c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner
1193027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    return true;
1194a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1195a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1196a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->isLogicalOp()) {
1197a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // These need to be handled specially because the operands aren't
1198a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // necessarily integral
1199fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    bool lhsResult, rhsResult;
12001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1201fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
120251fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // We were able to evaluate the LHS, see if we can get away with not
120351fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
12042de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (lhsResult == (E->getOpcode() == BO_LOr))
12053f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        return Success(lhsResult, E);
1206a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1207fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
12082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if (E->getOpcode() == BO_LOr)
1209131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult || rhsResult, E);
12104bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        else
1211131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult && rhsResult, E);
12124bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
12134bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson    } else {
1214fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
12154bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // We can't evaluate the LHS; however, sometimes the result
12164bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
12172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if (rhsResult == (E->getOpcode() == BO_LOr) ||
12182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall            !rhsResult == (E->getOpcode() == BO_LAnd)) {
1219131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          // Since we weren't able to evaluate the left hand side, it
1220fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          // must have had side effects.
1221fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          Info.EvalResult.HasSideEffects = true;
1222131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
1223131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(rhsResult, E);
12244bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        }
12254bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
1226a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    }
1227a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1228a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1229c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner  }
123054176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
1231286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType LHSTy = E->getLHS()->getType();
1232286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType RHSTy = E->getRHS()->getType();
12334087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12344087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  if (LHSTy->isAnyComplexType()) {
12354087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    assert(RHSTy->isAnyComplexType() && "Invalid comparison");
1236f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LHS, RHS;
12374087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12384087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getLHS(), LHS, Info))
12394087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12404087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12414087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getRHS(), RHS, Info))
12424087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12434087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12444087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (LHS.isComplexFloat()) {
12451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_r =
12464087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
12471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_i =
12484087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
12494087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12502de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ)
1251131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((CR_r == APFloat::cmpEqual &&
1252131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        CR_i == APFloat::cmpEqual), E);
1253131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
12542de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        assert(E->getOpcode() == BO_NE &&
1255131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid complex comparison.");
12561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        return Success(((CR_r == APFloat::cmpGreaterThan ||
1257fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpLessThan ||
1258fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpUnordered) ||
12591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                        (CR_i == APFloat::cmpGreaterThan ||
1260fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpLessThan ||
1261fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpUnordered)), E);
1262131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12634087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    } else {
12642de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ)
1265131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
1266131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
1267131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
12682de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        assert(E->getOpcode() == BO_NE &&
1269131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid compex comparison.");
1270131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
1271131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
1272131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12734087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    }
12744087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  }
12751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1276286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (LHSTy->isRealFloatingType() &&
1277286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      RHSTy->isRealFloatingType()) {
1278286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat RHS(0.0), LHS(0.0);
12791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1280286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getRHS(), RHS, Info))
1281286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1283286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getLHS(), LHS, Info))
1284286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1286286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat::cmpResult CR = LHS.compare(RHS);
1287529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson
1288286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    switch (E->getOpcode()) {
1289286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    default:
1290286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      assert(0 && "Invalid binary operator!");
12912de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LT:
1292131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan, E);
12932de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GT:
1294131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan, E);
12952de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LE:
1296131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
12972de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GE:
12981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
1299131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     E);
13002de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_EQ:
1301131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpEqual, E);
13022de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_NE:
13031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan
1304fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpLessThan
1305fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpUnordered, E);
1306286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    }
1307286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  }
13081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1309ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman  if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
13102de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    if (E->getOpcode() == BO_Sub || E->isEqualityOp()) {
1311efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue LHSValue;
13123068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
13133068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1314a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1315efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue RHSValue;
13163068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
13173068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1318a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
13195bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // Reject any bases from the normal codepath; we special-case comparisons
13205bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // to null.
13215bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      if (LHSValue.getLValueBase()) {
13225bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13235bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1324a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (RHSValue.getLValueBase() || !RHSValue.getLValueOffset().isZero())
13255bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13265bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13275bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(LHSValue, bres))
13285bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13292de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        return Success(bres ^ (E->getOpcode() == BO_EQ), E);
13305bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      } else if (RHSValue.getLValueBase()) {
13315bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13325bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1333a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (LHSValue.getLValueBase() || !LHSValue.getLValueOffset().isZero())
13345bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13355bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13365bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(RHSValue, bres))
13375bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13382de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        return Success(bres ^ (E->getOpcode() == BO_EQ), E);
13395bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      }
1340a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
13412de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_Sub) {
13424992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType Type = E->getLHS()->getType();
13434992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType ElementType = Type->getAs<PointerType>()->getPointeeType();
13443068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
1345a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits ElementSize = CharUnits::One();
1346ce1bca73aef9bbf6359ab8420278203dda81a054Eli Friedman        if (!ElementType->isVoidType() && !ElementType->isFunctionType())
1347a73058324197b7bdfd19307965954f626e26199dKen Dyck          ElementSize = Info.Ctx.getTypeSizeInChars(ElementType);
1348a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1349a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits Diff = LHSValue.getLValueOffset() -
1350a73058324197b7bdfd19307965954f626e26199dKen Dyck                             RHSValue.getLValueOffset();
1351a73058324197b7bdfd19307965954f626e26199dKen Dyck        return Success(Diff / ElementSize, E);
1352ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1353ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      bool Result;
13542de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (E->getOpcode() == BO_EQ) {
1355ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() == RHSValue.getLValueOffset();
1356267c0ab1b9a15768f3f15abbfc40ce344751c78bEli Friedman      } else {
1357ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() != RHSValue.getLValueOffset();
1358ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1359ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      return Success(Result, E);
13603068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    }
13613068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
13622ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!LHSTy->isIntegralOrEnumerationType() ||
13632ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor      !RHSTy->isIntegralOrEnumerationType()) {
1364a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // We can't continue from here for non-integral types, and they
1365a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // could potentially confuse the following operations.
1366a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1367a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1368a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1369a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  // The LHS of a constant expr is always evaluated and needed.
137030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Visit(E->getLHS()))
1371a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false; // error in subexpression.
1372d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman
137342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APValue RHSVal;
137442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!EvaluateIntegerOrLValue(E->getRHS(), RHSVal, Info))
137530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
137642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
137742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like (unsigned long)&a + 4.
137842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->isAdditiveOp() && Result.isLValue() && RHSVal.isInt()) {
1379a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = Result.getLValueOffset();
1380a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits AdditionalOffset = CharUnits::fromQuantity(
1381a73058324197b7bdfd19307965954f626e26199dKen Dyck                                     RHSVal.getInt().getZExtValue());
13822de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    if (E->getOpcode() == BO_Add)
1383a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset += AdditionalOffset;
138442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    else
1385a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset -= AdditionalOffset;
1386a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(Result.getLValueBase(), Offset);
138742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
138842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
138942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
139042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like 4 + (unsigned long)&a
13912de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Add &&
139242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman        RHSVal.isLValue() && Result.isInt()) {
1393a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = RHSVal.getLValueOffset();
1394a73058324197b7bdfd19307965954f626e26199dKen Dyck    Offset += CharUnits::fromQuantity(Result.getInt().getZExtValue());
1395a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(RHSVal.getLValueBase(), Offset);
139642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
139742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
139842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
139942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // All the following cases expect both operands to be an integer
140042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!Result.isInt() || !RHSVal.isInt())
1401b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner    return false;
1402a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
140342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APSInt& RHS = RHSVal.getInt();
140442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
1405a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  switch (E->getOpcode()) {
140632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  default:
14070e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
14082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul: return Success(Result.getInt() * RHS, E);
14092de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add: return Success(Result.getInt() + RHS, E);
14102de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub: return Success(Result.getInt() - RHS, E);
14112de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_And: return Success(Result.getInt() & RHS, E);
14122de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Xor: return Success(Result.getInt() ^ RHS, E);
14132de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Or:  return Success(Result.getInt() | RHS, E);
14142de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Div:
141554176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
14160e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
141730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() / RHS, E);
14182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Rem:
141954176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
14200e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
142130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() % RHS, E);
14222de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Shl: {
142354176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // FIXME: Warn about out of range shift amounts!
14241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
142530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
142630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() << SA, E);
14273f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14282de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Shr: {
14291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
143030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
143130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() >> SA, E);
14323f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
14342de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_LT: return Success(Result.getInt() < RHS, E);
14352de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_GT: return Success(Result.getInt() > RHS, E);
14362de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_LE: return Success(Result.getInt() <= RHS, E);
14372de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_GE: return Success(Result.getInt() >= RHS, E);
14382de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_EQ: return Success(Result.getInt() == RHS, E);
14392de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_NE: return Success(Result.getInt() != RHS, E);
1440b11e77836dd0867955c5abf32baf1c3e6c7f81e1Eli Friedman  }
1441a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
1442a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
1443ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopesbool IntExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
1444a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  bool Cond;
1445a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  if (!HandleConversionToBool(E->getCond(), Cond, Info))
1446ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes    return false;
1447ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
1448a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
1449ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes}
1450ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
14518b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfType(QualType T) {
14525d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14535d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14545d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14555d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14565d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = T->getAs<ReferenceType>())
14575d484e8cf710207010720589d89602233de61d01Sebastian Redl    T = Ref->getPointeeType();
14585d484e8cf710207010720589d89602233de61d01Sebastian Redl
1459e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the alignment.
1460e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
146118857644059c45da6776f1a288eec7b4cf3a844aDouglas Gregor
14622be586108bb401019647791feca19ea03fd477ceEli Friedman  // __alignof is defined to return the preferred alignment.
14638b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  return CharUnits::fromQuantity(
14648b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      Info.Ctx.getPreferredTypeAlign(T.getTypePtr()) / CharSize);
1465e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1466e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14678b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
1468af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  E = E->IgnoreParens();
1469af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1470af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // alignof decl is always accepted, even if it doesn't make sense: we default
14711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // to 1 in those cases.
1472af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
14738b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(DRE->getDecl(),
14748b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1475a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1476af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
14778b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
14788b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1479af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1480e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return GetAlignOfType(E->getType());
1481e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1482e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
1483e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14840518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// VisitSizeAlignOfExpr - Evaluate a sizeof or alignof with a result as the
14850518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// expression's type.
14860518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redlbool IntExprEvaluator::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
1487e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Handle alignof separately.
1488e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!E->isSizeOf()) {
1489e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    if (E->isArgumentType())
14908b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfType(E->getArgumentType()).getQuantity(), E);
1491e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    else
14928b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfExpr(E->getArgumentExpr()).getQuantity(), E);
1493e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  }
1494a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
14950518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType SrcTy = E->getTypeOfArgument();
14965d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14975d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14985d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14995d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
15005d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
15015d484e8cf710207010720589d89602233de61d01Sebastian Redl    SrcTy = Ref->getPointeeType();
15020518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
1503131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
1504131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // extension.
1505131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (SrcTy->isVoidType() || SrcTy->isFunctionType())
1506131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(1, E);
1507a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1508fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner  // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
1509e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!SrcTy->isConstantSizeType())
1510fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner    return false;
1511f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
1512e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the size.
1513199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  return Success(Info.Ctx.getTypeSizeInChars(SrcTy).getQuantity(), E);
1514fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner}
1515fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
15168ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregorbool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *E) {
15178ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  CharUnits Result;
15188ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  unsigned n = E->getNumComponents();
15198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  OffsetOfExpr* OOE = const_cast<OffsetOfExpr*>(E);
15208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  if (n == 0)
15218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    return false;
15228ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  QualType CurrentType = E->getTypeSourceInfo()->getType();
15238ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  for (unsigned i = 0; i != n; ++i) {
15248ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    OffsetOfExpr::OffsetOfNode ON = OOE->getComponent(i);
15258ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    switch (ON.getKind()) {
15268ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Array: {
15278ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
15288ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      APSInt IdxResult;
15298ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!EvaluateInteger(Idx, IdxResult, Info))
15308ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15318ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
15328ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!AT)
15338ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15348ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = AT->getElementType();
15358ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
15368ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Result += IdxResult.getSExtValue() * ElementSize;
15378ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        break;
15388ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15398ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15408ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Field: {
15418ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      FieldDecl *MemberDecl = ON.getField();
15428ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
15438ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!RT)
15448ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15458ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      RecordDecl *RD = RT->getDecl();
15468ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
15478ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      unsigned i = 0;
15488ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      // FIXME: It would be nice if we didn't have to loop here!
15498ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      for (RecordDecl::field_iterator Field = RD->field_begin(),
15508ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor                                      FieldEnd = RD->field_end();
15518ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor           Field != FieldEnd; (void)++Field, ++i) {
15528ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        if (*Field == MemberDecl)
15538ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor          break;
15548ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      }
1555cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      assert(i < RL.getFieldCount() && "offsetof field in wrong type");
1556cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1557cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                           RL.getFieldOffset(i) / Info.Ctx.getCharWidth());
15588ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = MemberDecl->getType().getNonReferenceType();
15598ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      break;
15608ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15618ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15628ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Identifier:
15638ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      llvm_unreachable("dependent __builtin_offsetof");
1564cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      return false;
1565cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1566cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Base: {
1567cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CXXBaseSpecifier *BaseSpec = ON.getBase();
1568cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (BaseSpec->isVirtual())
1569cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1570cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1571cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the layout of the class whose base we are looking into.
1572cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
1573cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!RT)
1574cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1575cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      RecordDecl *RD = RT->getDecl();
1576cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
1577cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1578cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the base class itself.
1579cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CurrentType = BaseSpec->getType();
1580cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *BaseRT = CurrentType->getAs<RecordType>();
1581cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!BaseRT)
1582cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1583cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1584cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Add the offset to the base.
1585cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1586cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                RL.getBaseClassOffset(cast<CXXRecordDecl>(BaseRT->getDecl()))
1587cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                                        / Info.Ctx.getCharWidth());
1588cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      break;
1589cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    }
15908ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15918ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  }
15928ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  return Success(Result.getQuantity(), E);
15938ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor}
15948ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1595b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
15962de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == UO_LNot) {
1597a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // LNot's operand isn't necessarily an integer, so we handle it specially.
1598a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    bool bres;
1599a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1600a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      return false;
1601131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(!bres, E);
1602a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1603a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
16044fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar  // Only handle integral operations...
16052ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getSubExpr()->getType()->isIntegralOrEnumerationType())
16064fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar    return false;
16074fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar
160887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  // Get the operand value into 'Result'.
160987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!Visit(E->getSubExpr()))
161075a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    return false;
1611a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
161275a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  switch (E->getOpcode()) {
16134c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  default:
161475a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
161575a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // See C99 6.6p3.
16160e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
16172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Extension:
16184c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // FIXME: Should extension allow i-c-e extension expressions in its scope?
16194c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // If so, we could clear the diagnostic ID.
16203f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
16212de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Plus:
16221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The result is always just the subexpr.
16233f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
16242de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Minus:
162530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
162630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(-Result.getInt(), E);
16272de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Not:
162830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
162930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(~Result.getInt(), E);
163006a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson  }
1631a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
16321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1633732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// HandleCast - This is used to evaluate implicit or explicit casts where the
1634732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// result type is integer.
1635a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarbool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
163682206e267ce6cc709797127616f64672d255b310Anders Carlsson  Expr *SubExpr = E->getSubExpr();
163782206e267ce6cc709797127616f64672d255b310Anders Carlsson  QualType DestType = E->getType();
1638b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  QualType SrcType = SubExpr->getType();
163982206e267ce6cc709797127616f64672d255b310Anders Carlsson
16404efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (DestType->isBooleanType()) {
16414efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    bool BoolResult;
16424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!HandleConversionToBool(SubExpr, BoolResult, Info))
16434efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1644131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(BoolResult, E);
16454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
16464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1647a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  // Handle simple integer->integer casts.
16482ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SrcType->isIntegralOrEnumerationType()) {
1649732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner    if (!Visit(SubExpr))
1650b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
1651a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1652be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!Result.isInt()) {
1653be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      // Only allow casts of lvalues if they are lossless.
1654be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
1655be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    }
165630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
1657dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType,
165830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar                                      Result.getInt(), Info.Ctx), E);
1659732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  }
16601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1661732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  // FIXME: Clean this up!
1662b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (SrcType->isPointerType()) {
1663efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
166487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (!EvaluatePointer(SubExpr, LV, Info))
1665b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
16664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1667dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    if (LV.getLValueBase()) {
1668dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      // Only allow based lvalue casts if they are lossless.
1669dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
1670dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar        return false;
1671dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar
1672efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LV.moveInto(Result);
1673dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      return true;
1674dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    }
16754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1676a73058324197b7bdfd19307965954f626e26199dKen Dyck    APSInt AsInt = Info.Ctx.MakeIntValue(LV.getLValueOffset().getQuantity(),
1677a73058324197b7bdfd19307965954f626e26199dKen Dyck                                         SrcType);
1678dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType, AsInt, Info.Ctx), E);
16792bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
16804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1681be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  if (SrcType->isArrayType() || SrcType->isFunctionType()) {
1682be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // This handles double-conversion cases, where there's both
1683be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // an l-value promotion and an implicit conversion to int.
1684efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
1685be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!EvaluateLValue(SubExpr, LV, Info))
1686be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1687be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1688be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(Info.Ctx.VoidPtrTy))
1689be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1690be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1691efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result);
1692be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    return true;
1693be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  }
1694be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
16951725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  if (SrcType->isAnyComplexType()) {
1696f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue C;
16971725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (!EvaluateComplex(SubExpr, C, Info))
16981725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return false;
16991725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (C.isComplexFloat())
17001725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleFloatToIntCast(DestType, SrcType,
17011725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                          C.getComplexFloatReal(), Info.Ctx),
17021725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                     E);
17031725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    else
17041725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleIntToIntCast(DestType, SrcType,
17051725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                        C.getComplexIntReal(), Info.Ctx), E);
17061725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  }
17072217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle vectors
17082217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1709b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (!SrcType->isRealFloatingType())
17100e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1711732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1712d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat F(0.0);
1713d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(SubExpr, F, Info))
17140e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
17151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1716b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
1717a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
17182bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
1719722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedmanbool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1720722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
1721f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1722722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1723722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1724722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntReal(), E);
1725722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1726722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1727722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  return Visit(E->getSubExpr());
1728722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman}
1729722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1730664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedmanbool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1731722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isComplexIntegerType()) {
1732f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1733722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1734722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1735722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntImag(), E);
1736722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1737722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1738664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
1739664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    Info.EvalResult.HasSideEffects = true;
1740664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  return Success(0, E);
1741664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman}
1742664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
1743f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1744d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman// Float Evaluation
1745d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1746d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1747d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmannamespace {
1748770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass FloatExprEvaluator
1749d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  : public StmtVisitor<FloatExprEvaluator, bool> {
1750d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  EvalInfo &Info;
1751d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat &Result;
1752d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanpublic:
1753d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  FloatExprEvaluator(EvalInfo &info, APFloat &result)
1754d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    : Info(info), Result(result) {}
1755d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1756d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitStmt(Stmt *S) {
1757d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1758d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1759d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1760d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
1761019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  bool VisitCallExpr(const CallExpr *E);
1762d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
17635db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  bool VisitUnaryOperator(const UnaryOperator *E);
1764d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
1765d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitFloatingLiteral(const FloatingLiteral *E);
17664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCastExpr(CastExpr *E);
1767ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
176867f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool VisitConditionalOperator(ConditionalOperator *E);
17692217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1770ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
1771ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
1772ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
1773ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
1774abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryReal(const UnaryOperator *E);
1775abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryImag(const UnaryOperator *E);
1776ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
1777abd3a857ace59100305790545d1baae5877b8945John McCall  // FIXME: Missing: array subscript of vector, member of vector,
1778abd3a857ace59100305790545d1baae5877b8945John McCall  //                 ImplicitValueInitExpr
1779d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman};
1780d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman} // end anonymous namespace
1781d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1782d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
17837db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isRealFloatingType());
1784d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
1785d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1786d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1787db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCallstatic bool TryEvaluateBuiltinNaN(ASTContext &Context,
1788db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  QualType ResultTy,
1789db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  const Expr *Arg,
1790db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  bool SNaN,
1791db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  llvm::APFloat &Result) {
1792db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
1793db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (!S) return false;
1794db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1795db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
1796db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1797db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  llvm::APInt fill;
1798db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1799db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  // Treat empty strings as if they were zero.
1800db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (S->getString().empty())
1801db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    fill = llvm::APInt(32, 0);
1802db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else if (S->getString().getAsInteger(0, fill))
1803db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return false;
1804db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1805db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (SNaN)
1806db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getSNaN(Sem, false, &fill);
1807db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else
1808db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getQNaN(Sem, false, &fill);
1809db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  return true;
1810db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall}
1811db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1812019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattnerbool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
18133c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
181434a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner  default: return false;
1815019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_val:
1816019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_valf:
1817019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_vall:
1818019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inf:
1819019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inff:
18207cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  case Builtin::BI__builtin_infl: {
18217cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar    const llvm::fltSemantics &Sem =
18227cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar      Info.Ctx.getFloatTypeSemantics(E->getType());
182334a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    Result = llvm::APFloat::getInf(Sem);
182434a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    return true;
18257cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  }
18261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1827db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nans:
1828db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansf:
1829db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansl:
1830db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1831db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 true, Result);
1832db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
18339e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nan:
18349e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanf:
18359e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanl:
18364572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump    // If this is __builtin_nan() turn this into a nan, otherwise we
18379e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // can't constant fold it.
1838db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1839db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 false, Result);
18405db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18415db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabs:
18425db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsf:
18435db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsl:
18445db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info))
18455db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18475db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (Result.isNegative())
18485db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      Result.changeSign();
18495db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18505db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysign:
18521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysignf:
18535db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignl: {
18545db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    APFloat RHS(0.);
18555db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info) ||
18565db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar        !EvaluateFloat(E->getArg(1), RHS, Info))
18575db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18585db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.copySign(RHS);
18595db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18605db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
1861019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
1862019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner}
1863019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1864abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
186543efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
186643efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    ComplexValue CV;
186743efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), CV, Info))
186843efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman      return false;
186943efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Result = CV.FloatReal;
187043efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    return true;
187143efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  }
187243efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman
187343efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  return Visit(E->getSubExpr());
1874abd3a857ace59100305790545d1baae5877b8945John McCall}
1875abd3a857ace59100305790545d1baae5877b8945John McCall
1876abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
187743efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
187843efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    ComplexValue CV;
187943efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), CV, Info))
188043efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman      return false;
188143efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Result = CV.FloatImag;
188243efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    return true;
188343efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  }
188443efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman
188543efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
188643efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman    Info.EvalResult.HasSideEffects = true;
188743efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  const llvm::fltSemantics &Sem = Info.Ctx.getFloatTypeSemantics(E->getType());
188843efa31fe601bb7c3f132f02246dc3c903d9f361Eli Friedman  Result = llvm::APFloat::getZero(Sem);
1889abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1890abd3a857ace59100305790545d1baae5877b8945John McCall}
1891abd3a857ace59100305790545d1baae5877b8945John McCall
18925db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbarbool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
18932de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == UO_Deref)
1894a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes    return false;
1895a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes
18965db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  if (!EvaluateFloat(E->getSubExpr(), Result, Info))
18975db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return false;
18985db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18995db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  switch (E->getOpcode()) {
19005db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  default: return false;
19012de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Plus:
19025db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
19032de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case UO_Minus:
19045db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.changeSign();
19055db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
19065db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
19075db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar}
1908019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1909d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
19102de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  if (E->getOpcode() == BO_Comma) {
19117f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!EvaluateFloat(E->getRHS(), Result, Info))
19127f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      return false;
19137f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
19147f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
19157f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // conservatively mark it.
19167f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
19177f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      Info.EvalResult.HasSideEffects = true;
19187f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
19197f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    return true;
19207f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  }
19217f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
1922d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // FIXME: Diagnostics?  I really don't understand how the warnings
1923d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // and errors are supposed to work.
19245db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  APFloat RHS(0.0);
1925d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getLHS(), Result, Info))
1926d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1927d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getRHS(), RHS, Info))
1928d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1929d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1930d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  switch (E->getOpcode()) {
1931d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  default: return false;
19322de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul:
1933d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.multiply(RHS, APFloat::rmNearestTiesToEven);
1934d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
19352de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add:
1936d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.add(RHS, APFloat::rmNearestTiesToEven);
1937d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
19382de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub:
1939d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.subtract(RHS, APFloat::rmNearestTiesToEven);
1940d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
19412de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Div:
1942d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.divide(RHS, APFloat::rmNearestTiesToEven);
1943d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1944d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1945d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1946d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1947d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
1948d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  Result = E->getValue();
1949d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return true;
1950d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1951d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
19524efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCastExpr(CastExpr *E) {
19534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* SubExpr = E->getSubExpr();
19541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19552ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SubExpr->getType()->isIntegralOrEnumerationType()) {
19564efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
19573f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    if (!EvaluateInteger(SubExpr, IntResult, Info))
19584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
19591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
1960a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                  IntResult, Info.Ctx);
19614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
19644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!Visit(SubExpr))
19654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1966a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
1967a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    Result, Info.Ctx);
19684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19702217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle complex types
19714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
19724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
19734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1975ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregorbool FloatExprEvaluator::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
19774efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return true;
19784efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19794efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
198067f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedmanbool FloatExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
198167f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool Cond;
198267f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  if (!HandleConversionToBool(E->getCond(), Cond, Info))
198367f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman    return false;
198467f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
198567f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
198667f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman}
198767f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
1988d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1989a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar// Complex Evaluation (for float and integer)
19909ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
19919ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
19929ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonnamespace {
1993770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass ComplexExprEvaluator
1994f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  : public StmtVisitor<ComplexExprEvaluator, bool> {
19959ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  EvalInfo &Info;
1996f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue &Result;
19971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19989ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonpublic:
1999f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
2000f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    : Info(info), Result(Result) {}
20011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20029ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
20039ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //                            Visitor Methods
20049ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
20059ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2006f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitStmt(Stmt *S) {
2007f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
20089ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
20091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2010f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
20119ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2012b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitImaginaryLiteral(ImaginaryLiteral *E);
2013a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
2014b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitCastExpr(CastExpr *E);
2015b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2016b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
2017b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
2018b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
2019b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
2020b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    { return Visit(E->getSubExpr()); }
2021b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  // FIXME Missing: unary +/-/~, binary div, ImplicitValueInitExpr,
2022b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  //                conditional ?:, comma
2023b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman};
2024b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman} // end anonymous namespace
20251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2026b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanstatic bool EvaluateComplex(const Expr *E, ComplexValue &Result,
2027b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                            EvalInfo &Info) {
2028b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  assert(E->getType()->isAnyComplexType());
2029b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  return ComplexExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
2030b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman}
2031b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2032b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanbool ComplexExprEvaluator::VisitImaginaryLiteral(ImaginaryLiteral *E) {
2033b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  Expr* SubExpr = E->getSubExpr();
2034b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2035b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
2036b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.makeComplexFloat();
2037b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APFloat &Imag = Result.FloatImag;
2038b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateFloat(SubExpr, Imag, Info))
2039b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2040b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2041b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.FloatReal = APFloat(Imag.getSemantics());
2042b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    return true;
2043b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  } else {
2044b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    assert(SubExpr->getType()->isIntegerType() &&
2045b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman           "Unexpected imaginary literal.");
2046b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2047b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.makeComplexInt();
2048b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APSInt &Imag = Result.IntImag;
2049b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateInteger(SubExpr, Imag, Info))
2050b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2051b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2052b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
2053b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    return true;
2054b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  }
2055b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman}
2056b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2057b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedmanbool ComplexExprEvaluator::VisitCastExpr(CastExpr *E) {
2058b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  Expr* SubExpr = E->getSubExpr();
2059b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  QualType EltType = E->getType()->getAs<ComplexType>()->getElementType();
2060b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  QualType SubType = SubExpr->getType();
2061b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2062b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  if (SubType->isRealFloatingType()) {
2063b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APFloat &Real = Result.FloatReal;
2064b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateFloat(SubExpr, Real, Info))
2065b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2066b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman
2067b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (EltType->isRealFloatingType()) {
2068b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.makeComplexFloat();
2069b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Real = HandleFloatToFloatCast(EltType, SubType, Real, Info.Ctx);
2070b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.FloatImag = APFloat(Real.getSemantics());
2071f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2072a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2073f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.makeComplexInt();
2074b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.IntReal = HandleFloatToIntCast(EltType, SubType, Real, Info.Ctx);
2075b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.IntImag = APSInt(Result.IntReal.getBitWidth(),
2076b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                              !Result.IntReal.isSigned());
2077f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2078a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
2079b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  } else if (SubType->isIntegerType()) {
2080b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    APSInt &Real = Result.IntReal;
2081b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!EvaluateInteger(SubExpr, Real, Info))
2082b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
20839ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2084b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (EltType->isRealFloatingType()) {
2085b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.makeComplexFloat();
2086b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.FloatReal
2087b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        = HandleIntToFloatCast(EltType, SubType, Real, Info.Ctx);
2088b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.FloatImag = APFloat(Result.FloatReal.getSemantics());
2089b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return true;
2090b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    } else {
2091b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.makeComplexInt();
2092b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Real = HandleIntToIntCast(EltType, SubType, Real, Info.Ctx);
2093b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
2094b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return true;
2095b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    }
2096b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  } else if (const ComplexType *CT = SubType->getAs<ComplexType>()) {
2097b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (!Visit(SubExpr))
2098b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      return false;
2099ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2100b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    QualType SrcType = CT->getElementType();
21011725f683432715e5afe34d476024bd6f16eac3fcEli Friedman
2102b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    if (Result.isComplexFloat()) {
21031725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2104f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2105b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.FloatReal = HandleFloatToFloatCast(EltType, SrcType,
2106b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                  Result.FloatReal,
2107b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                  Info.Ctx);
2108b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.FloatImag = HandleFloatToFloatCast(EltType, SrcType,
2109b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                  Result.FloatImag,
2110b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                  Info.Ctx);
2111f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
21121725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2113f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2114b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.IntReal = HandleFloatToIntCast(EltType, SrcType,
2115b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                              Result.FloatReal,
2116b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                              Info.Ctx);
2117b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.IntImag = HandleFloatToIntCast(EltType, SrcType,
2118b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                              Result.FloatImag,
2119b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                              Info.Ctx);
2120f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
21211725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2122b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman    } else {
2123b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman      assert(Result.isComplexInt() && "Invalid evaluate result.");
21241725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2125f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2126b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.FloatReal = HandleIntToFloatCast(EltType, SrcType,
2127b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                Result.IntReal,
2128b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                Info.Ctx);
2129b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.FloatImag = HandleIntToFloatCast(EltType, SrcType,
2130b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                Result.IntImag,
2131b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                                Info.Ctx);
2132f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
21331725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2134f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2135b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.IntReal = HandleIntToIntCast(EltType, SrcType,
2136b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                            Result.IntReal,
2137b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                            Info.Ctx);
2138b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman        Result.IntImag = HandleIntToIntCast(EltType, SrcType,
2139b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                            Result.IntImag,
2140b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman                                            Info.Ctx);
2141f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
21421725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2143ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    }
2144ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
21451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2146b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  // FIXME: Handle more casts.
2147b2dc7f59fe4c762cba73badc3bbc6f356fcd7b5bEli Friedman  return false;
21489ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson}
21499ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2150f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallbool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
2151f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  if (!Visit(E->getLHS()))
2152f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
21531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2154f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue RHS;
2155a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getRHS(), RHS, Info))
2156f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2157a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
21583f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
21593f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         "Invalid operands to binary operator.");
2160ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  switch (E->getOpcode()) {
2161f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  default: return false;
21622de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Add:
2163a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2164a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
2165a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2166a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
2167a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2168a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2169a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() += RHS.getComplexIntReal();
2170a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() += RHS.getComplexIntImag();
2171a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21723f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
21732de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Sub:
2174a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2175a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
2176a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2177a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
2178a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2179a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2180a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() -= RHS.getComplexIntReal();
2181a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() -= RHS.getComplexIntImag();
2182a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21833f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
21842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall  case BO_Mul:
21853f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    if (Result.isComplexFloat()) {
2186f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
21873f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_r = LHS.getComplexFloatReal();
21883f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_i = LHS.getComplexFloatImag();
21893f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_r = RHS.getComplexFloatReal();
21903f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_i = RHS.getComplexFloatImag();
21911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
21923f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat Tmp = LHS_r;
21933f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
21943f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal() = Tmp;
21953f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
21963f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
21973f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
21983f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
21993f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_r;
22003f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
22013f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag() = Tmp;
22023f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
22033f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
22043f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
22053f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    } else {
2206f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
22071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntReal() =
22083f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
22093f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntImag());
22101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntImag() =
22113f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
22123f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntReal());
22133f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    }
22143f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
2215ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
2216ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2217f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return true;
2218ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson}
2219ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
22209ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
22216ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner// Top level Expr::Evaluate method.
2222f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
2223f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
222442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Evaluate - Return true if this is a constant which we can fold using
222542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// any crazy technique (that has nothing to do with language standards) that
222642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// we want to.  If this function returns true, it returns the folded constant
222742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// in Result.
222842c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool Expr::Evaluate(EvalResult &Result, ASTContext &Ctx) const {
222942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *E = this;
223042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo Info(Ctx, Result);
2231efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (E->getType()->isVectorType()) {
2232efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateVector(E, Info.EvalResult.Val, Info))
223359b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return false;
2234efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isIntegerType()) {
2235efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!IntExprEvaluator(Info, Info.EvalResult.Val).Visit(const_cast<Expr*>(E)))
22366dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22370f2b692bb10be35fdc60d0a72a847bdd73124670John McCall    if (Result.Val.isLValue() && !IsGlobalLValue(Result.Val.getLValueBase()))
22380f2b692bb10be35fdc60d0a72a847bdd73124670John McCall      return false;
2239efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->hasPointerRepresentation()) {
2240efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
2241efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E, LV, Info))
22426dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
2243e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    if (!IsGlobalLValue(LV.Base))
224442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return false;
2245efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Info.EvalResult.Val);
2246efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isRealFloatingType()) {
2247efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    llvm::APFloat F(0.0);
2248efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateFloat(E, F, Info))
22496dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2251efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info.EvalResult.Val = APValue(F);
2252efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isAnyComplexType()) {
2253efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    ComplexValue C;
2254efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateComplex(E, C, Info))
2255660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump      return false;
2256efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    C.moveInto(Info.EvalResult.Val);
2257660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  } else
2258660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump    return false;
2259660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2260660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  return true;
2261660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump}
2262660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2263cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallbool Expr::EvaluateAsBooleanCondition(bool &Result, ASTContext &Ctx) const {
2264cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalResult Scratch;
2265cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalInfo Info(Ctx, Scratch);
2266cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
2267cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  return HandleConversionToBool(this, Result, Info);
2268cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall}
2269cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
22701b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlssonbool Expr::EvaluateAsLValue(EvalResult &Result, ASTContext &Ctx) const {
22711b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson  EvalInfo Info(Ctx, Result);
22721b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson
2273efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue LV;
227442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (EvaluateLValue(this, LV, Info) &&
227542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !Result.HasSideEffects &&
2276e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara      IsGlobalLValue(LV.Base)) {
2277e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    LV.moveInto(Result.Val);
2278e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    return true;
2279e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  }
2280e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  return false;
2281e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara}
2282e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2283e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarabool Expr::EvaluateAsAnyLValue(EvalResult &Result, ASTContext &Ctx) const {
2284e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  EvalInfo Info(Ctx, Result);
2285e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2286e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  LValue LV;
2287e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (EvaluateLValue(this, LV, Info)) {
2288efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result.Val);
2289efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
2290efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
2291efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
2292b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman}
2293b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman
22946ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// isEvaluatable - Call Evaluate to see if this expression can be constant
229545b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner/// folded, but discard the result.
229645b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattnerbool Expr::isEvaluatable(ASTContext &Ctx) const {
22974fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  EvalResult Result;
22984fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  return Evaluate(Result, Ctx) && !Result.HasSideEffects;
229945b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner}
230051fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
2301393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanianbool Expr::HasSideEffects(ASTContext &Ctx) const {
2302393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  Expr::EvalResult Result;
2303393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  EvalInfo Info(Ctx, Result);
2304393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  return HasSideEffect(Info).Visit(const_cast<Expr*>(this));
2305393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian}
2306393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian
230751fe996231b1d7199f76e4005ff4c943d5deeecdAnders CarlssonAPSInt Expr::EvaluateAsInt(ASTContext &Ctx) const {
23081c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  EvalResult EvalResult;
23091c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  bool Result = Evaluate(EvalResult, Ctx);
2310f185319f25efd6094870f287030270fad26085baDaniel Dunbar  Result = Result;
231151fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson  assert(Result && "Could not evaluate expression");
23121c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
231351fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
23141c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  return EvalResult.Val.getInt();
231551fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson}
2316d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2317e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara bool Expr::EvalResult::isGlobalLValue() const {
2318e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   assert(Val.isLValue());
2319e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   return IsGlobalLValue(Val.getLValueBase());
2320e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara }
2321e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2322e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2323d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// isIntegerConstantExpr - this recursive routine will test if an expression is
2324d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// an integer constant expression.
2325d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2326d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
2327d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// comma, etc
2328d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall///
2329d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Handle offsetof.  Two things to do:  Handle GCC's __builtin_offsetof
2330d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// to support gcc 4.0+  and handle the idiom GCC recognizes with a null pointer
2331d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// cast+dereference.
2332d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2333d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// CheckICE - This function does the fundamental ICE checking: the returned
2334d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// ICEDiag contains a Val of 0, 1, or 2, and a possibly null SourceLocation.
2335d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Note that to reduce code duplication, this helper does no evaluation
2336d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// itself; the caller checks whether the expression is evaluatable, and
2337d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// in the rare cases where CheckICE actually cares about the evaluated
2338d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// value, it calls into Evalute.
2339d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//
2340d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Meanings of Val:
2341d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 0: This expression is an ICE if it can be evaluated by Evaluate.
2342d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 1: This expression is not an ICE, but if it isn't evaluated, it's
2343d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    a legal subexpression for an ICE. This return value is used to handle
2344d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    the comma operator in C99 mode.
2345d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 2: This expression is not an ICE, and is not a legal subexpression for one.
2346d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
23473c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmannamespace {
23483c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
2349d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstruct ICEDiag {
2350d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  unsigned Val;
2351d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  SourceLocation Loc;
2352d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2353d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  public:
2354d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag(unsigned v, SourceLocation l) : Val(v), Loc(l) {}
2355d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag() : Val(0) {}
2356d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall};
2357d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
23583c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman}
23593c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
23603c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmanstatic ICEDiag NoDiag() { return ICEDiag(); }
2361d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2362d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckEvalInICE(const Expr* E, ASTContext &Ctx) {
2363d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Expr::EvalResult EVResult;
2364d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2365d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      !EVResult.Val.isInt()) {
2366d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2367d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2368d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return NoDiag();
2369d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2370d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2371d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckICE(const Expr* E, ASTContext &Ctx) {
2372d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!E->isValueDependent() && "Should not see value dependent exprs!");
23732ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getType()->isIntegralOrEnumerationType()) {
2374d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2375d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2376d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2377d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  switch (E->getStmtClass()) {
2378d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define STMT(Node, Base) case Expr::Node##Class:
2379d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define EXPR(Node, Base)
2380d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#include "clang/AST/StmtNodes.inc"
2381d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::PredefinedExprClass:
2382d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::FloatingLiteralClass:
2383d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImaginaryLiteralClass:
2384d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StringLiteralClass:
2385d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ArraySubscriptExprClass:
2386d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::MemberExprClass:
2387d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundAssignOperatorClass:
2388d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundLiteralExprClass:
2389d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ExtVectorElementExprClass:
2390d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::InitListExprClass:
2391d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DesignatedInitExprClass:
2392d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitValueInitExprClass:
2393d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenListExprClass:
2394d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::VAArgExprClass:
2395d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::AddrLabelExprClass:
2396d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StmtExprClass:
2397d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXMemberCallExprClass:
2398d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDynamicCastExprClass:
2399d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTypeidExprClass:
24009be88403e965cc49af76c9d33d818781d44b333eFrancois Pichet  case Expr::CXXUuidofExprClass:
2401d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNullPtrLiteralExprClass:
2402d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThisExprClass:
2403d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThrowExprClass:
2404d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNewExprClass:
2405d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDeleteExprClass:
2406d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXPseudoDestructorExprClass:
2407d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedLookupExprClass:
2408d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DependentScopeDeclRefExprClass:
2409d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstructExprClass:
2410d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindTemporaryExprClass:
2411d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXExprWithTemporariesClass:
2412d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTemporaryObjectExprClass:
2413d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXUnresolvedConstructExprClass:
2414d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDependentScopeMemberExprClass:
2415d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedMemberExprClass:
2416d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCStringLiteralClass:
2417d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCEncodeExprClass:
2418d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCMessageExprClass:
2419d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSelectorExprClass:
2420d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCProtocolExprClass:
2421d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIvarRefExprClass:
2422d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCPropertyRefExprClass:
2423d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCImplicitSetterGetterRefExprClass:
2424d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSuperExprClass:
2425d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIsaExprClass:
2426d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ShuffleVectorExprClass:
2427d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockExprClass:
2428d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockDeclRefExprClass:
2429d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::NoStmtClass:
2430d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2431d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2432d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::GNUNullExprClass:
2433d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // GCC considers the GNU __null value to be an integral constant expression.
2434d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2435d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2436d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenExprClass:
2437d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
2438d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::IntegerLiteralClass:
2439d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CharacterLiteralClass:
2440d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBoolLiteralExprClass:
2441ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  case Expr::CXXScalarValueInitExprClass:
2442d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::TypesCompatibleExprClass:
2443d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryTypeTraitExprClass:
24442e156225a29407a50dd19041aa5750171ad44ea3Sebastian Redl  case Expr::CXXNoexceptExprClass:
2445d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2446d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CallExprClass:
24476cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  case Expr::CXXOperatorCallExprClass: {
2448d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const CallExpr *CE = cast<CallExpr>(E);
2449d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CE->isBuiltinCall(Ctx))
2450d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2451d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2452d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2453d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DeclRefExprClass:
2454d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<EnumConstantDecl>(cast<DeclRefExpr>(E)->getDecl()))
2455d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2456d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Ctx.getLangOptions().CPlusPlus &&
2457d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        E->getType().getCVRQualifiers() == Qualifiers::Const) {
2458d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
2459d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2460d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Parameter variables are never constants.  Without this check,
2461d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // getAnyInitializer() can find a default argument, which leads
2462d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // to chaos.
2463d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (isa<ParmVarDecl>(D))
2464d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2465d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2466d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // C++ 7.1.5.1p2
2467d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   A variable of non-volatile const-qualified integral or enumeration
2468d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   type initialized by an ICE can be used in ICEs.
2469d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (const VarDecl *Dcl = dyn_cast<VarDecl>(D)) {
2470d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Qualifiers Quals = Ctx.getCanonicalType(Dcl->getType()).getQualifiers();
2471d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Quals.hasVolatile() || !Quals.hasConst())
2472d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2473d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2474d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Look for a declaration of this variable that has an initializer.
2475d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const VarDecl *ID = 0;
2476d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const Expr *Init = Dcl->getAnyInitializer(ID);
2477d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Init) {
2478d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (ID->isInitKnownICE()) {
2479d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // We have already checked whether this subexpression is an
2480d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // integral constant expression.
2481d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (ID->isInitICE())
2482d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return NoDiag();
2483d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            else
2484d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2485d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2486d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2487d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // It's an ICE whether or not the definition we found is
2488d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // out-of-line.  See DR 721 and the discussion in Clang PR
2489d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // 6206 for details.
2490d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2491d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (Dcl->isCheckingICE()) {
2492d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2493d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2494d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2495d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setCheckingICE();
2496d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          ICEDiag Result = CheckICE(Init, Ctx);
2497d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // Cache the result of the ICE test.
2498d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setInitKnownICE(Result.Val == 0);
2499d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return Result;
2500d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2501d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2502d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2503d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2504d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryOperatorClass: {
2505d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const UnaryOperator *Exp = cast<UnaryOperator>(E);
2506d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
25072de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PostInc:
25082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PostDec:
25092de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PreInc:
25102de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_PreDec:
25112de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_AddrOf:
25122de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Deref:
2513d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
25142de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Extension:
25152de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_LNot:
25162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Plus:
25172de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Minus:
25182de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Not:
25192de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Real:
25202de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case UO_Imag:
2521d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(Exp->getSubExpr(), Ctx);
2522d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2523d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2524d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // OffsetOf falls through here.
2525d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2526d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::OffsetOfExprClass: {
2527d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Note that per C99, offsetof must be an ICE. And AFAIK, using
2528d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Evaluate matches the proposed gcc behavior for cases like
2529d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // "offsetof(struct s{int x[4];}, x[!.0])".  This doesn't affect
2530d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // compliance: we should warn earlier for offsetof expressions with
2531d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // array subscripts that aren't ICEs, and if the array subscripts
2532d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // are ICEs, the value of the offsetof must be an integer constant.
2533d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2534d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2535d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::SizeOfAlignOfExprClass: {
2536d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const SizeOfAlignOfExpr *Exp = cast<SizeOfAlignOfExpr>(E);
2537d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->isSizeOf() && Exp->getTypeOfArgument()->isVariableArrayType())
2538d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2539d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2540d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2541d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BinaryOperatorClass: {
2542d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const BinaryOperator *Exp = cast<BinaryOperator>(E);
2543d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
25442de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_PtrMemD:
25452de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_PtrMemI:
25462de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Assign:
25472de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_MulAssign:
25482de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_DivAssign:
25492de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_RemAssign:
25502de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_AddAssign:
25512de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_SubAssign:
25522de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_ShlAssign:
25532de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_ShrAssign:
25542de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_AndAssign:
25552de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_XorAssign:
25562de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_OrAssign:
2557d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2558d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
25592de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Mul:
25602de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Div:
25612de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Rem:
25622de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Add:
25632de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Sub:
25642de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Shl:
25652de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Shr:
25662de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LT:
25672de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GT:
25682de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LE:
25692de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_GE:
25702de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_EQ:
25712de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_NE:
25722de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_And:
25732de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Xor:
25742de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Or:
25752de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_Comma: {
2576d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2577d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
25782de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (Exp->getOpcode() == BO_Div ||
25792de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall          Exp->getOpcode() == BO_Rem) {
2580d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Evaluate gives an error for undefined Div/Rem, so make sure
2581d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // we don't evaluate one.
2582d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (LHSResult.Val != 2 && RHSResult.Val != 2) {
2583d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          llvm::APSInt REval = Exp->getRHS()->EvaluateAsInt(Ctx);
2584d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval == 0)
2585d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2586d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval.isSigned() && REval.isAllOnesValue()) {
2587d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            llvm::APSInt LEval = Exp->getLHS()->EvaluateAsInt(Ctx);
2588d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (LEval.isMinSignedValue())
2589d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(1, E->getLocStart());
2590d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2591d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2592d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
25932de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall      if (Exp->getOpcode() == BO_Comma) {
2594d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Ctx.getLangOptions().C99) {
2595d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
2596d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // if it isn't evaluated.
2597d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (LHSResult.Val == 0 && RHSResult.Val == 0)
2598d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2599d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        } else {
2600d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // In both C89 and C++, commas in ICEs are illegal.
2601d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2602d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2603d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2604d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2605d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2606d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2607d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
26082de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LAnd:
26092de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall    case BO_LOr: {
2610d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2611d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2612d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val == 0 && RHSResult.Val == 1) {
2613d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Rare case where the RHS has a comma "side-effect"; we need
2614d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // to actually check the condition to see whether the side
2615d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // with the comma is evaluated.
26162de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall        if ((Exp->getOpcode() == BO_LAnd) !=
2617d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            (Exp->getLHS()->EvaluateAsInt(Ctx) == 0))
2618d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return RHSResult;
2619d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2620d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2621d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2622d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2623d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2624d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2625d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2626d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2627d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2628d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitCastExprClass:
2629d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CStyleCastExprClass:
2630d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXFunctionalCastExprClass:
2631d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXStaticCastExprClass:
2632d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXReinterpretCastExprClass:
2633d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstCastExprClass: {
2634d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
26352ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType())
2636d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(SubExpr, Ctx);
2637d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<FloatingLiteral>(SubExpr->IgnoreParens()))
2638d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2639d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2640d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2641d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ConditionalOperatorClass: {
2642d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
2643d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // If the condition (ignoring parens) is a __builtin_constant_p call,
2644d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // then only the true side is actually considered in an integer constant
2645d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // expression, and it is fully evaluated.  This is an important GNU
2646d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // extension.  See GCC PR38377 for discussion.
2647d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (const CallExpr *CallCE
2648d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
2649d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (CallCE->isBuiltinCall(Ctx) == Builtin::BI__builtin_constant_p) {
2650d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Expr::EvalResult EVResult;
2651d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2652d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            !EVResult.Val.isInt()) {
2653d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2654d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2655d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2656d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2657d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
2658d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
2659d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
2660d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 2)
2661d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2662d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 2)
2663d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return TrueResult;
2664d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (FalseResult.Val == 2)
2665d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2666d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 1)
2667d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2668d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 0 && FalseResult.Val == 0)
2669d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2670d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Rare case where the diagnostics depend on which side is evaluated
2671d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Note that if we get here, CondResult is 0, and at least one of
2672d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // TrueResult and FalseResult is non-zero.
2673d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->getCond()->EvaluateAsInt(Ctx) == 0) {
2674d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2675d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2676d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return TrueResult;
2677d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2678d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDefaultArgExprClass:
2679d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
2680d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ChooseExprClass: {
2681d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(Ctx), Ctx);
2682d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2683d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2684d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2685d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  // Silence a GCC warning
2686d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return ICEDiag(2, E->getLocStart());
2687d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2688d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2689d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallbool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
2690d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall                                 SourceLocation *Loc, bool isEvaluated) const {
2691d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag d = CheckICE(this, Ctx);
2692d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (d.Val != 0) {
2693d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Loc) *Loc = d.Loc;
2694d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return false;
2695d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2696d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  EvalResult EvalResult;
2697d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!Evaluate(EvalResult, Ctx))
2698d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    llvm_unreachable("ICE cannot be evaluated!");
2699d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!EvalResult.HasSideEffects && "ICE with side effects!");
2700d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(EvalResult.Val.isInt() && "ICE that isn't integer!");
2701d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Result = EvalResult.Val.getInt();
2702d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return true;
2703d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2704