ExprConstant.cpp revision 3c46e8db99196179b30e7ac5c20c4efd5f3926d7
1b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner//===--- ExprConstant.cpp - Expression Constant Evaluator -----------------===//
2c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
3c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//                     The LLVM Compiler Infrastructure
4c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
5c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file is distributed under the University of Illinois Open Source
6c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// License. See LICENSE.TXT for details.
7c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
8c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
9c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
10c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson// This file implements the Expr constant evaluator.
11c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//
12c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson//===----------------------------------------------------------------------===//
13c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
14c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/APValue.h"
15c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson#include "clang/AST/ASTContext.h"
16199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck#include "clang/AST/CharUnits.h"
1719cc4abea06a9b49e0e16a50d335c064cd723572Anders Carlsson#include "clang/AST/RecordLayout.h"
180fe52e1bcaa69ba127f1bda036f057fec1f478deSeo Sanghyeon#include "clang/AST/StmtVisitor.h"
198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/TypeLoc.h"
20500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/AST/ASTDiagnostic.h"
218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor#include "clang/AST/Expr.h"
221b63e4f732dbc73d90abf886b4d21f8e3a165f6dChris Lattner#include "clang/Basic/Builtins.h"
2306a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson#include "clang/Basic/TargetInfo.h"
247462b39a9bccaf4392687831036713f09f9c0681Mike Stump#include "llvm/ADT/SmallString.h"
254572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump#include <cstring>
264572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump
27c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlssonusing namespace clang;
28f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattnerusing llvm::APSInt;
29d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanusing llvm::APFloat;
30c44eec6dd29ee9415cbd38a35deff4c8b67abb6aAnders Carlsson
3187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// EvalInfo - This is a private struct used by the evaluator to capture
3287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// information about a subexpression as it is folded.  It retains information
3387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// about the AST context, but also maintains information about the folded
3487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// expression.
3587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
3687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// If an expression could be evaluated, it is still possible it is not a C
3787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// "integer constant expression" or constant expression.  If not, this struct
3887eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// captures information about how and why not.
3987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner///
4087eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// One bit of information passed *into* the request for constant folding
4187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// indicates whether the subexpression is "evaluated" or not according to C
4287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// rules.  For example, the RHS of (0 && foo()) is not evaluated.  We can
4387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// evaluate the expression regardless of what the RHS is, but C only allows
4487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner/// certain things in certain situations.
4587eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstruct EvalInfo {
4687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  ASTContext &Ctx;
471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4854da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  /// EvalResult - Contains information about the evaluation.
4954da0496616cb898596cfd5fae1f824098bd8435Anders Carlsson  Expr::EvalResult &EvalResult;
50f0c1e4b679e15c26bffb5892e35985bf3c52f77aAnders Carlsson
5142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo(ASTContext &ctx, Expr::EvalResult& evalresult)
5242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    : Ctx(ctx), EvalResult(evalresult) {}
5387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner};
5487eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
55f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallnamespace {
56f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  struct ComplexValue {
57f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  private:
58f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool IsInt;
59f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
60f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  public:
61f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt IntReal, IntImag;
62f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat FloatReal, FloatImag;
63f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
64f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue() : FloatReal(APFloat::Bogus), FloatImag(APFloat::Bogus) {}
65f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
66f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexFloat() { IsInt = false; }
67f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexFloat() const { return !IsInt; }
68f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatReal() { return FloatReal; }
69f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APFloat &getComplexFloatImag() { return FloatImag; }
70f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
71f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void makeComplexInt() { IsInt = true; }
72f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    bool isComplexInt() const { return IsInt; }
73f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntReal() { return IntReal; }
74f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    APSInt &getComplexIntImag() { return IntImag; }
75f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall
76f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    void moveInto(APValue &v) {
77f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (isComplexFloat())
78f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(FloatReal, FloatImag);
79f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      else
80f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        v = APValue(IntReal, IntImag);
81f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    }
82f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  };
83efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
84efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  struct LValue {
85efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *Base;
86efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits Offset;
87efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
88efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Expr *getLValueBase() { return Base; }
89efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    CharUnits getLValueOffset() { return Offset; }
90efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
91efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    void moveInto(APValue &v) {
92efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      v = APValue(Base, Offset);
93efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    }
94efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  };
95f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall}
9687eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner
97efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr *E, LValue &Result, EvalInfo &Info);
98efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr *E, LValue &Result, EvalInfo &Info);
9987eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattnerstatic bool EvaluateInteger(const Expr *E, APSInt  &Result, EvalInfo &Info);
100d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattnerstatic bool EvaluateIntegerOrLValue(const Expr *E, APValue  &Result,
101d9becd1846e2c72bf6ad283faa1b048f33dd3afeChris Lattner                                    EvalInfo &Info);
102d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr *E, APFloat &Result, EvalInfo &Info);
103f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallstatic bool EvaluateComplex(const Expr *E, ComplexValue &Res, EvalInfo &Info);
104f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
105f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1064efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// Misc utilities
1074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
1084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
109e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarastatic bool IsGlobalLValue(const Expr* E) {
11042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!E) return true;
11142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
11242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
11342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (isa<FunctionDecl>(DRE->getDecl()))
11442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
11542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
11642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->hasGlobalStorage();
11742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
11842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
11942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const CompoundLiteralExpr *CLE = dyn_cast<CompoundLiteralExpr>(E))
12142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return CLE->isFileScope();
12242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
12342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return true;
12442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
12542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
126efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvalPointerValueAsBool(LValue& Value, bool& Result) {
127efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  const Expr* Base = Value.Base;
128a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1293554283157190e67918fad4221a5e6faf9317362John McCall  // A null base expression indicates a null pointer.  These are always
1303554283157190e67918fad4221a5e6faf9317362John McCall  // evaluatable, and they are false unless the offset is zero.
1313554283157190e67918fad4221a5e6faf9317362John McCall  if (!Base) {
1323554283157190e67918fad4221a5e6faf9317362John McCall    Result = !Value.Offset.isZero();
1333554283157190e67918fad4221a5e6faf9317362John McCall    return true;
1343554283157190e67918fad4221a5e6faf9317362John McCall  }
1353554283157190e67918fad4221a5e6faf9317362John McCall
13642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // Require the base expression to be a global l-value.
137e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (!IsGlobalLValue(Base)) return false;
13842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1393554283157190e67918fad4221a5e6faf9317362John McCall  // We have a non-null base expression.  These are generally known to
1403554283157190e67918fad4221a5e6faf9317362John McCall  // be true, but if it'a decl-ref to a weak symbol it can be null at
1413554283157190e67918fad4221a5e6faf9317362John McCall  // runtime.
1423554283157190e67918fad4221a5e6faf9317362John McCall  Result = true;
143a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1443554283157190e67918fad4221a5e6faf9317362John McCall  const DeclRefExpr* DeclRef = dyn_cast<DeclRefExpr>(Base);
145a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (!DeclRef)
146a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return true;
147a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1483554283157190e67918fad4221a5e6faf9317362John McCall  // If it's a weak symbol, it isn't constant-evaluable.
149a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  const ValueDecl* Decl = DeclRef->getDecl();
150a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola  if (Decl->hasAttr<WeakAttr>() ||
151a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakRefAttr>() ||
152a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola      Decl->hasAttr<WeakImportAttr>())
153a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola    return false;
154a7d3c04fcfe9d4af2f7758f46aef26b1a8f8ac09Rafael Espindola
1555bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman  return true;
1565bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman}
1575bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman
158cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallstatic bool HandleConversionToBool(const Expr* E, bool& Result,
159cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall                                   EvalInfo &Info) {
1602ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (E->getType()->isIntegralOrEnumerationType()) {
1614efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
1624efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateInteger(E, IntResult, Info))
1634efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1644efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = IntResult != 0;
1654efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
1664efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else if (E->getType()->isRealFloatingType()) {
1674efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APFloat FloatResult(0.0);
1684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluateFloat(E, FloatResult, Info))
1694efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1704efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Result = !FloatResult.isZero();
1714efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
172a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->hasPointerRepresentation()) {
173efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue PointerResult;
1744efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!EvaluatePointer(E, PointerResult, Info))
1754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1765bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman    return EvalPointerValueAsBool(PointerResult, Result);
177a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman  } else if (E->getType()->isAnyComplexType()) {
178f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue ComplexResult;
179a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (!EvaluateComplex(E, ComplexResult, Info))
180a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      return false;
181a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    if (ComplexResult.isComplexFloat()) {
182a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = !ComplexResult.getComplexFloatReal().isZero() ||
183a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               !ComplexResult.getComplexFloatImag().isZero();
184a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    } else {
185a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman      Result = ComplexResult.getComplexIntReal().getBoolValue() ||
186a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman               ComplexResult.getComplexIntImag().getBoolValue();
187a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    }
188a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman    return true;
1894efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
1904efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1914efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
1924efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
1934efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleFloatToIntCast(QualType DestType, QualType SrcType,
195a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                   APFloat &Value, ASTContext &Ctx) {
196a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
197a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Determine whether we are converting to unsigned or signed.
198a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool DestSigned = DestType->isSignedIntegerType();
1991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
200a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // FIXME: Warning for overflow.
201a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  uint64_t Space[4];
202a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
203a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  (void)Value.convertToInteger(Space, DestWidth, DestSigned,
204a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                               llvm::APFloat::rmTowardZero, &ignored);
205a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return APSInt(llvm::APInt(DestWidth, 4, Space), !DestSigned);
206a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
207a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleFloatToFloatCast(QualType DestType, QualType SrcType,
209a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                      APFloat &Value, ASTContext &Ctx) {
210a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  bool ignored;
211a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result = Value;
2121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Result.convert(Ctx.getFloatTypeSemantics(DestType),
213a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                 APFloat::rmNearestTiesToEven, &ignored);
214a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
215a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
216a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APSInt HandleIntToIntCast(QualType DestType, QualType SrcType,
218a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                 APSInt &Value, ASTContext &Ctx) {
219a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  unsigned DestWidth = Ctx.getIntWidth(DestType);
220a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APSInt Result = Value;
221a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // Figure out if this is a truncate, extend or noop cast.
222a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  // If the input is signed, do a sign extend, noop, or truncate.
223a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.extOrTrunc(DestWidth);
224a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.setIsUnsigned(DestType->isUnsignedIntegerType());
225a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
226a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
227a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic APFloat HandleIntToFloatCast(QualType DestType, QualType SrcType,
229a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    APSInt &Value, ASTContext &Ctx) {
230a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
231a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  APFloat Result(Ctx.getFloatTypeSemantics(DestType), 1);
232a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  Result.convertFromAPInt(Value, Value.isSigned(),
233a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                          APFloat::rmNearestTiesToEven);
234a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar  return Result;
235a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar}
236a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
237c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumpnamespace {
238770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass HasSideEffect
239c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  : public StmtVisitor<HasSideEffect, bool> {
240c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  EvalInfo &Info;
241c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stumppublic:
242c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
243c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  HasSideEffect(EvalInfo &info) : Info(info) {}
244c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
245c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // Unhandled nodes conservatively default to having side effects.
246c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStmt(Stmt *S) {
247c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return true;
248c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
249c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
250c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
251c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitDeclRefExpr(DeclRefExpr *E) {
252df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
253c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
254c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    return false;
255c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
256c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // We don't want to evaluate BlockExprs multiple times, as they generate
257c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  // a ton of code.
258c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBlockExpr(BlockExpr *E) { return true; }
259c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitPredefinedExpr(PredefinedExpr *E) { return false; }
260c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E)
261c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getInitializer()); }
262c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitMemberExpr(MemberExpr *E) { return Visit(E->getBase()); }
263c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitIntegerLiteral(IntegerLiteral *E) { return false; }
264c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitFloatingLiteral(FloatingLiteral *E) { return false; }
265c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitStringLiteral(StringLiteral *E) { return false; }
266c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCharacterLiteral(CharacterLiteral *E) { return false; }
267c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) { return false; }
268c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E)
269980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    { return Visit(E->getLHS()) || Visit(E->getRHS()); }
270c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitChooseExpr(ChooseExpr *E)
271c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
272c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitCastExpr(CastExpr *E) { return Visit(E->getSubExpr()); }
273c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitBinAssign(BinaryOperator *E) { return true; }
2743f0147e161df4725ff15fbb731f4f727afcc229fMike Stump  bool VisitCompoundAssignOperator(BinaryOperator *E) { return true; }
275980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  bool VisitBinaryOperator(BinaryOperator *E)
276980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump  { return Visit(E->getLHS()) || Visit(E->getRHS()); }
277c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreInc(UnaryOperator *E) { return true; }
278c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostInc(UnaryOperator *E) { return true; }
279c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPreDec(UnaryOperator *E) { return true; }
280c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryPostDec(UnaryOperator *E) { return true; }
281c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryDeref(UnaryOperator *E) {
282df317bf71653eeb235da8337b1e8e790f9653aa4Mike Stump    if (Info.Ctx.getCanonicalType(E->getType()).isVolatileQualified())
283c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump      return true;
284980ca220848d27ef55ef4c2d37423461a8ed0da3Mike Stump    return Visit(E->getSubExpr());
285c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  }
286c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump  bool VisitUnaryOperator(UnaryOperator *E) { return Visit(E->getSubExpr()); }
287363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner
288363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  // Has side effects if any element does.
289363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  bool VisitInitListExpr(InitListExpr *E) {
290363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    for (unsigned i = 0, e = E->getNumInits(); i != e; ++i)
291363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner      if (Visit(E->getInit(i))) return true;
292363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner    return false;
293363ff23cfddb51abe4ee4212a6dd3c9b534fcc8bChris Lattner  }
294c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump};
295c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
296c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump} // end anonymous namespace
297c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
2984efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
2994efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman// LValue Evaluation
3004efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
3014efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmannamespace {
302770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass LValueExprEvaluator
303efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<LValueExprEvaluator, bool> {
3044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  EvalInfo &Info;
305efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
306efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
307efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
308efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
309efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
310efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
311efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
3124efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanpublic:
3131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
314efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValueExprEvaluator(EvalInfo &info, LValue &Result) :
315efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info(info), Result(Result) {}
3164efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
317efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
318efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
321efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
322efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitDeclRefExpr(DeclRefExpr *E);
323efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitPredefinedExpr(PredefinedExpr *E) { return Success(E); }
324efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
325efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitMemberExpr(MemberExpr *E);
326efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStringLiteral(StringLiteral *E) { return Success(E); }
327efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCEncodeExpr(ObjCEncodeExpr *E) { return Success(E); }
328efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitArraySubscriptExpr(ArraySubscriptExpr *E);
329efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryDeref(UnaryOperator *E);
330efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
331ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
332efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(const ChooseExpr *E)
333ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
33426bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
335efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr *E) {
33626bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    switch (E->getCastKind()) {
33726bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    default:
338efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
33926bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson
34026bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    case CastExpr::CK_NoOp:
34126bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson      return Visit(E->getSubExpr());
34226bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson    }
34326bc220377705292a0519a71d3ea3aef68fcfec6Anders Carlsson  }
344ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: __real__, __imag__
3454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman};
3464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman} // end anonymous namespace
3474efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
348efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluateLValue(const Expr* E, LValue& Result, EvalInfo &Info) {
349efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return LValueExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
3504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3514efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
352efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitDeclRefExpr(DeclRefExpr *E) {
35350c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  if (isa<FunctionDecl>(E->getDecl())) {
354efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return Success(E);
35550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  } else if (VarDecl* VD = dyn_cast<VarDecl>(E->getDecl())) {
35650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman    if (!VD->getType()->isReferenceType())
357efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
358761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // Reference parameters can refer to anything even if they have an
359761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    // "initializer" in the form of a default argument.
360761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth    if (isa<ParmVarDecl>(VD))
361761c94e3bffef0fcb8b4bbf202fd5ee73db134f3Chandler Carruth      return false;
362d933a0198f3ccce9c73bf2951625315b911d37bfEli Friedman    // FIXME: Check whether VD might be overridden!
36331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (const Expr *Init = VD->getAnyInitializer())
364cf3293eaeb3853d12cff47e648bbe835004e929fDouglas Gregor      return Visit(const_cast<Expr *>(Init));
36550c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman  }
36650c39ea4858265f3f5f42a0c624557ce2281936bEli Friedman
367efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
36835873c49adad211ff466e34342a52665742794f5Anders Carlsson}
36935873c49adad211ff466e34342a52665742794f5Anders Carlsson
370efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
371efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return Success(E);
3724efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
3734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
374efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitMemberExpr(MemberExpr *E) {
3754efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  QualType Ty;
3764efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (E->isArrow()) {
377efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E->getBase(), Result, Info))
378efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3796217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek    Ty = E->getBase()->getType()->getAs<PointerType>()->getPointeeType();
3804efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  } else {
381efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!Visit(E->getBase()))
382efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return false;
3834efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    Ty = E->getBase()->getType();
3844efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
3854efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
3866217b80b7a1379b74cced1c076338262c3c980b3Ted Kremenek  RecordDecl *RD = Ty->getAs<RecordType>()->getDecl();
3874efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
38886f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor
38986f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  FieldDecl *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
39086f194083504938df72135b5b66bf0c5cafd9498Douglas Gregor  if (!FD) // FIXME: deal with other kinds of member expressions
391efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3922be586108bb401019647791feca19ea03fd477ceEli Friedman
3932be586108bb401019647791feca19ea03fd477ceEli Friedman  if (FD->getType()->isReferenceType())
394efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
3952be586108bb401019647791feca19ea03fd477ceEli Friedman
3964efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  // FIXME: This is linear time.
39744b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor  unsigned i = 0;
39817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (RecordDecl::field_iterator Field = RD->field_begin(),
39917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis                               FieldEnd = RD->field_end();
40044b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor       Field != FieldEnd; (void)++Field, ++i) {
40144b4321feab46299d3f5cfd404680884752a0fcfDouglas Gregor    if (*Field == FD)
4024efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      break;
4034efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
4044efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
405efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += CharUnits::fromQuantity(RL.getFieldOffset(i) / 8);
406efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4074efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
4084efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
409efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
4103068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluatePointer(E->getBase(), Result, Info))
411efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4133068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  APSInt Index;
4143068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  if (!EvaluateInteger(E->getIdx(), Index, Info))
415efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4163068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
417199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(E->getType());
418efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  Result.Offset += Index.getSExtValue() * ElementSize;
419efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
4203068d117951a8df54bae9db039b56201ab10962bAnders Carlsson}
4214efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
422efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool LValueExprEvaluator::VisitUnaryDeref(UnaryOperator *E) {
423efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluatePointer(E->getSubExpr(), Result, Info);
424e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman}
425e8761c8fe2ee6b628104a0885f49fd3c21c08a4fEli Friedman
4264efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman//===----------------------------------------------------------------------===//
427f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner// Pointer Evaluation
428f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
429f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
430c754aa62643e66ab967ca32ae8b0b3fc419bba25Anders Carlssonnamespace {
431770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass PointerExprEvaluator
432efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  : public StmtVisitor<PointerExprEvaluator, bool> {
43387eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  EvalInfo &Info;
434efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue &Result;
435efdb83e26f9a1fd2566afe54461216cd84814d42John McCall
436efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool Success(Expr *E) {
437efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Base = E;
438efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset = CharUnits::Zero();
439efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
440efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
4412bad1687fe6f00e10767a691a33b070b151902b6Anders Carlssonpublic:
4421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
443efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  PointerExprEvaluator(EvalInfo &info, LValue &Result)
444efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    : Info(info), Result(Result) {}
445f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
446efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitStmt(Stmt *S) {
447efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4482bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
4492bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
450efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
4512bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
452efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBinaryOperator(const BinaryOperator *E);
453efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCastExpr(CastExpr* E);
454efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
4552217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman      { return Visit(E->getSubExpr()); }
456efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitUnaryAddrOf(const UnaryOperator *E);
457efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitObjCStringLiteral(ObjCStringLiteral *E)
458efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
459efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitAddrLabelExpr(AddrLabelExpr *E)
460efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success(E); }
461efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCallExpr(CallExpr *E);
462efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitBlockExpr(BlockExpr *E) {
463b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump    if (!E->hasBlockDeclRefExprs())
464efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      return Success(E);
465efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
466b83d287bc7f47d36fb0751a481e2ef9308b37252Mike Stump  }
467efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitImplicitValueInitExpr(ImplicitValueInitExpr *E)
468efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
469efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitConditionalOperator(ConditionalOperator *E);
470efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitChooseExpr(ChooseExpr *E)
4716e8ed16ffef02b82995a90bdcf10ffff7d63839aSebastian Redl      { return Visit(E->getChosenSubExpr(Info.Ctx)); }
472efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  bool VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E)
473efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      { return Success((Expr*)0); }
474ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME: Missing: @protocol, @selector
4752bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson};
476f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner} // end anonymous namespace
4772bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
478efdb83e26f9a1fd2566afe54461216cd84814d42John McCallstatic bool EvaluatePointer(const Expr* E, LValue& Result, EvalInfo &Info) {
4797db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->hasPointerRepresentation());
480efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return PointerExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
481f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner}
482650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
483efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
484650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() != BinaryOperator::Add &&
485650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson      E->getOpcode() != BinaryOperator::Sub)
486efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
488650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *PExp = E->getLHS();
489650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  const Expr *IExp = E->getRHS();
490650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (IExp->getType()->isPointerType())
491f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner    std::swap(PExp, IExp);
4921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
493efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluatePointer(PExp, Result, Info))
494efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
4951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
496efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  llvm::APSInt Offset;
497efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (!EvaluateInteger(IExp, Offset, Info))
498efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return false;
499efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  int64_t AdditionalOffset
500efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    = Offset.isSigned() ? Offset.getSExtValue()
501efdb83e26f9a1fd2566afe54461216cd84814d42John McCall                        : static_cast<int64_t>(Offset.getZExtValue());
502650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
503e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  // Compute the new offset in the appropriate width.
504e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar
505e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar  QualType PointeeType =
506e0cdb4edd8f265d0fd22d178d03c597dd201cda2Daniel Dunbar    PExp->getType()->getAs<PointerType>()->getPointeeType();
507efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  CharUnits SizeOfPointee;
5081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5094d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  // Explicitly handle GNU void* and function pointer arithmetic extensions.
5104d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  if (PointeeType->isVoidType() || PointeeType->isFunctionType())
511efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = CharUnits::One();
5124d4c50dd8b4db191c38782414665fb7608315a36Anders Carlsson  else
513efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    SizeOfPointee = Info.Ctx.getTypeSizeInChars(PointeeType);
5144efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
515650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  if (E->getOpcode() == BinaryOperator::Add)
516efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset += AdditionalOffset * SizeOfPointee;
517650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  else
518efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Result.Offset -= AdditionalOffset * SizeOfPointee;
5194efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
520efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return true;
521650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson}
5224efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
523efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitUnaryAddrOf(const UnaryOperator *E) {
524efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return EvaluateLValue(E->getSubExpr(), Result, Info);
5254efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
5261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
527650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
528efdb83e26f9a1fd2566afe54461216cd84814d42John McCallbool PointerExprEvaluator::VisitCastExpr(CastExpr* E) {
52909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  Expr* SubExpr = E->getSubExpr();
530650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson
53109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  switch (E->getCastKind()) {
53209a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  default:
53309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    break;
53409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
53509a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_Unknown: {
53609a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    // FIXME: The handling for CK_Unknown is ugly/shouldn't be necessary!
53709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
53809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    // Check for pointer->pointer cast
53909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    if (SubExpr->getType()->isPointerType() ||
54009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isObjCObjectPointerType() ||
54109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isNullPtrType() ||
54209a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        SubExpr->getType()->isBlockPointerType())
54309a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      return Visit(SubExpr);
54409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
5452ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType()) {
546efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      APValue Value;
547efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      if (!EvaluateIntegerOrLValue(SubExpr, Value, Info))
54809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman        break;
54909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
550efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      if (Value.isInt()) {
551efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Value.getInt().extOrTrunc((unsigned)Info.Ctx.getTypeSize(E->getType()));
552efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Base = 0;
553efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Offset = CharUnits::fromQuantity(Value.getInt().getZExtValue());
554efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        return true;
555efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      } else {
556efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Base = Value.getLValueBase();
557efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        Result.Offset = Value.getLValueOffset();
558efdb83e26f9a1fd2566afe54461216cd84814d42John McCall        return true;
55909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman      }
56009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    }
56109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    break;
562650c92fdcc27a950a8a848ecab6a74e6f5e80788Anders Carlsson  }
5631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
56409a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_NoOp:
56509a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_BitCast:
566e39a3894513349908cdb3beba2614e53cb288e6cDouglas Gregor  case CastExpr::CK_LValueBitCast:
56709a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_AnyPointerToObjCPointerCast:
56809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_AnyPointerToBlockPointerCast:
56909a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman    return Visit(SubExpr);
57009a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman
57109a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::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  }
58809a8a0e6ec1252cad52666e9dbb21002b9c80f38Eli Friedman  case CastExpr::CK_ArrayToPointerDecay:
589efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  case CastExpr::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
101178d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        if (Visit(const_cast<Expr*>(Init))) {
101278d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          // Cache the evaluated value in the variable declaration.
1013c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman          VD->setEvaluatedValue(Result);
101478d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor          return true;
101578d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        }
101678d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor
1017c013118155077cf8bdaef743c5dbce760114eed3Eli Friedman        VD->setEvaluatedValue(APValue());
101878d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor        return false;
101978d1583d0b36b7d6d8d10234cdc19ab94adf765aDouglas Gregor      }
1020b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl    }
1021b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl  }
1022b2bc62bd35129f09d0746ef6a5ef4dc392b6d6e1Sebastian Redl
10234c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Otherwise, random variable references are not constants.
10240e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson  return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
10254c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
10264c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner
1027a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// EvaluateBuiltinClassifyType - Evaluate __builtin_classify_type the same way
1028a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner/// as GCC.
1029a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattnerstatic int EvaluateBuiltinClassifyType(const CallExpr *E) {
1030a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // The following enum mimics the values returned by GCC.
10317c80bd64032e610c0dbd74fc0ef6ea334447f2fdSebastian Redl  // FIXME: Does GCC differ between lvalue and rvalue references here?
1032a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  enum gcc_type_class {
1033a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    no_type_class = -1,
1034a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    void_type_class, integer_type_class, char_type_class,
1035a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    enumeral_type_class, boolean_type_class,
1036a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    pointer_type_class, reference_type_class, offset_type_class,
1037a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    real_type_class, complex_type_class,
1038a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    function_type_class, method_type_class,
1039a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    record_type_class, union_type_class,
1040a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    array_type_class, string_type_class,
1041a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    lang_type_class
1042a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  };
10431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // If no argument was supplied, default to "no_type_class". This isn't
1045a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  // ideal, however it is what gcc does.
1046a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (E->getNumArgs() == 0)
1047a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return no_type_class;
10481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1049a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  QualType ArgTy = E->getArg(0)->getType();
1050a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  if (ArgTy->isVoidType())
1051a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return void_type_class;
1052a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isEnumeralType())
1053a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return enumeral_type_class;
1054a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isBooleanType())
1055a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return boolean_type_class;
1056a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isCharType())
1057a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return string_type_class; // gcc doesn't appear to use char_type_class
1058a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isIntegerType())
1059a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return integer_type_class;
1060a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isPointerType())
1061a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return pointer_type_class;
1062a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isReferenceType())
1063a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return reference_type_class;
1064a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isRealType())
1065a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return real_type_class;
1066a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isComplexType())
1067a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return complex_type_class;
1068a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isFunctionType())
1069a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return function_type_class;
1070fb87b89fc9eb103e19fb8e4b925c23f0bd091b99Douglas Gregor  else if (ArgTy->isStructureOrClassType())
1071a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return record_type_class;
1072a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1073a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1074a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isArrayType())
1075a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return array_type_class;
1076a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else if (ArgTy->isUnionType())
1077a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    return union_type_class;
1078a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  else  // FIXME: offset_type_class, method_type_class, & lang_type_class?
1079a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner    assert(0 && "CallExpr::isBuiltinClassifyType(): unimplemented type");
1080a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner  return -1;
1081a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner}
1082a4d55d89c8076b402bb168e3edeef0c2cd2a78c3Chris Lattner
108342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Retrieves the "underlying object type" of the given expression,
108442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// as used by __builtin_object_size.
108542c8f87eb60958170c46767273bf93e6c96125bfJohn McCallQualType IntExprEvaluator::GetObjectType(const Expr *E) {
108642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
108742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl()))
108842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return VD->getType();
108942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  } else if (isa<CompoundLiteralExpr>(E)) {
109042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return E->getType();
109142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  }
109242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return QualType();
109442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
109542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
109642c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool IntExprEvaluator::TryEvaluateBuiltinObjectSize(CallExpr *E) {
109742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // TODO: Perhaps we should let LLVM lower this?
109842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  LValue Base;
109942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!EvaluatePointer(E->getArg(0), Base, Info))
110042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
110142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  // If we can prove the base is null, lower to zero now.
110342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *LVBase = Base.getLValueBase();
110442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!LVBase) return Success(0, E);
110542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
110642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  QualType T = GetObjectType(LVBase);
110742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (T.isNull() ||
110842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isIncompleteType() ||
110942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !T->isObjectType() ||
111042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isVariablyModifiedType() ||
111142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      T->isDependentType())
111242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    return false;
111342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Size = Info.Ctx.getTypeSizeInChars(T);
111542c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  CharUnits Offset = Base.getLValueOffset();
111642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
111742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (!Offset.isNegative() && Offset <= Size)
111842c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size -= Offset;
111942c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  else
112042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    Size = CharUnits::Zero();
112142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  return Success(Size.getQuantity(), E);
112242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall}
112342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall
1124c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedmanbool IntExprEvaluator::VisitCallExpr(CallExpr *E) {
11253c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
1126019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  default:
11270e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
112864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
112964eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  case Builtin::BI__builtin_object_size: {
113042c8f87eb60958170c46767273bf93e6c96125bfJohn McCall    if (TryEvaluateBuiltinObjectSize(E))
113142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return true;
113264eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1133b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // If evaluating the argument has side-effects we can't determine
1134b2aaf51ed73a4774322a39a0dd59d0fe7e3258c0Eric Christopher    // the size of the object and lower it to unknown now.
1135393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian    if (E->getArg(0)->HasSideEffects(Info.Ctx)) {
11363f27b384801de26ce7efaa395699b42719372f24Benjamin Kramer      if (E->getArg(1)->EvaluateAsInt(Info.Ctx).getZExtValue() <= 1)
1137cf184655319cf7a5b811067cff9d26a5741fd161Chris Lattner        return Success(-1ULL, E);
113864eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump      return Success(0, E);
113964eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    }
1140c4c9045dabfc0f0d37dea1b3eb2992654d5b2db1Mike Stump
114164eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump    return Error(E->getLocStart(), diag::note_invalid_subexpr_in_ice, E);
114264eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump  }
114364eda9e50b593f935c95bd1edc98c4bfda03f601Mike Stump
1144019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_classify_type:
1145131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(EvaluateBuiltinClassifyType(E), E);
11461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11474bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson  case Builtin::BI__builtin_constant_p:
1148019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // __builtin_constant_p always has one operand: it returns true if that
1149019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner    // operand can be folded, false otherwise.
1150131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(E->getArg(0)->isEvaluatable(Info.Ctx), E);
115121fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner
115221fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  case Builtin::BI__builtin_eh_return_data_regno: {
115321fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    int Operand = E->getArg(0)->EvaluateAsInt(Info.Ctx).getZExtValue();
115421fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    Operand = Info.Ctx.Target.getEHDataRegisterNumber(Operand);
115521fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner    return Success(Operand, E);
115621fb98ee003e992b0c4e204d98a19e0ef544cae3Chris Lattner  }
1157c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman
1158c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman  case Builtin::BI__builtin_expect:
1159c4a2638b5ef3e2d35d872614ceb655a7a22c58beEli Friedman    return Visit(E->getArg(0));
1160019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
11614c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner}
1162f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
1163b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
1164a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == BinaryOperator::Comma) {
1165027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    if (!Visit(E->getRHS()))
1166027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson      return false;
11674fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson
116833ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
116933ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    // conservatively mark it.
117033ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
117133ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      Info.EvalResult.HasSideEffects = true;
1172c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner
1173027f62ec1860f4ab0c91bd863b238938880b8102Anders Carlsson    return true;
1174a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1175a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1176a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->isLogicalOp()) {
1177a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // These need to be handled specially because the operands aren't
1178a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // necessarily integral
1179fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    bool lhsResult, rhsResult;
11801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1181fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson    if (HandleConversionToBool(E->getLHS(), lhsResult, Info)) {
118251fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // We were able to evaluate the LHS, see if we can get away with not
118351fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson      // evaluating the RHS: 0 && X -> 0, 1 || X -> 1
118433ef1456a19779fb15c51aea362bd9c7cb4087f3Eli Friedman      if (lhsResult == (E->getOpcode() == BinaryOperator::LOr))
11853f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar        return Success(lhsResult, E);
1186a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1187fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
11884bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        if (E->getOpcode() == BinaryOperator::LOr)
1189131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult || rhsResult, E);
11904bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        else
1191131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(lhsResult && rhsResult, E);
11924bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
11934bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson    } else {
1194fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson      if (HandleConversionToBool(E->getRHS(), rhsResult, Info)) {
11954bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // We can't evaluate the LHS; however, sometimes the result
11964bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        // is determined by the RHS: X && 0 -> 0, X || 1 -> 1.
11971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        if (rhsResult == (E->getOpcode() == BinaryOperator::LOr) ||
1198fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson            !rhsResult == (E->getOpcode() == BinaryOperator::LAnd)) {
1199131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          // Since we weren't able to evaluate the left hand side, it
1200fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          // must have had side effects.
1201fcb4d09531abbf2a5cf398c2f946fb3bc2875f64Anders Carlsson          Info.EvalResult.HasSideEffects = true;
1202131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar
1203131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar          return Success(rhsResult, E);
12044bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson        }
12054bbc0e05a714d4ee4918a92a4a7049dd6ef33ad0Anders Carlsson      }
1206a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    }
1207a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1208a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1209c8cc9ccc7b87a7ed1749b074f6b670bcec49abc1Chris Lattner  }
121054176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner
1211286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType LHSTy = E->getLHS()->getType();
1212286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  QualType RHSTy = E->getRHS()->getType();
12134087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12144087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  if (LHSTy->isAnyComplexType()) {
12154087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    assert(RHSTy->isAnyComplexType() && "Invalid comparison");
1216f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LHS, RHS;
12174087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12184087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getLHS(), LHS, Info))
12194087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12204087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12214087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (!EvaluateComplex(E->getRHS(), RHS, Info))
12224087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      return false;
12234087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12244087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    if (LHS.isComplexFloat()) {
12251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_r =
12264087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatReal().compare(RHS.getComplexFloatReal());
12271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      APFloat::cmpResult CR_i =
12284087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar        LHS.getComplexFloatImag().compare(RHS.getComplexFloatImag());
12294087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar
12304087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
1231131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((CR_r == APFloat::cmpEqual &&
1232131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        CR_i == APFloat::cmpEqual), E);
1233131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
1234131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
1235131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid complex comparison.");
12361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        return Success(((CR_r == APFloat::cmpGreaterThan ||
1237fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpLessThan ||
1238fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_r == APFloat::cmpUnordered) ||
12391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                        (CR_i == APFloat::cmpGreaterThan ||
1240fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpLessThan ||
1241fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                         CR_i == APFloat::cmpUnordered)), E);
1242131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12434087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    } else {
12444087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar      if (E->getOpcode() == BinaryOperator::EQ)
1245131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() == RHS.getComplexIntReal() &&
1246131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() == RHS.getComplexIntImag()), E);
1247131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      else {
1248131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        assert(E->getOpcode() == BinaryOperator::NE &&
1249131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar               "Invalid compex comparison.");
1250131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar        return Success((LHS.getComplexIntReal() != RHS.getComplexIntReal() ||
1251131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                        LHS.getComplexIntImag() != RHS.getComplexIntImag()), E);
1252131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      }
12534087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar    }
12544087e24f73d05d96ac2d259679751d054d3ddfbcDaniel Dunbar  }
12551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1256286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  if (LHSTy->isRealFloatingType() &&
1257286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      RHSTy->isRealFloatingType()) {
1258286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat RHS(0.0), LHS(0.0);
12591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1260286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getRHS(), RHS, Info))
1261286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1263286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    if (!EvaluateFloat(E->getLHS(), LHS, Info))
1264286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      return false;
12651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1266286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    APFloat::cmpResult CR = LHS.compare(RHS);
1267529569e68d10b0fd3750fd2124faf742249b846bAnders Carlsson
1268286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    switch (E->getOpcode()) {
1269286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    default:
1270286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson      assert(0 && "Invalid binary operator!");
1271286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LT:
1272131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan, E);
1273286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GT:
1274131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpGreaterThan, E);
1275286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::LE:
1276131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpLessThan || CR == APFloat::cmpEqual, E);
1277286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::GE:
12781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan || CR == APFloat::cmpEqual,
1279131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar                     E);
1280286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::EQ:
1281131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar      return Success(CR == APFloat::cmpEqual, E);
1282286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    case BinaryOperator::NE:
12831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return Success(CR == APFloat::cmpGreaterThan
1284fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpLessThan
1285fc39dc4c7886723310419b1869cf651ca55b7af4Mon P Wang                     || CR == APFloat::cmpUnordered, E);
1286286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson    }
1287286f85e791dda3634fee7f6c67f0ed92296c028fAnders Carlsson  }
12881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1289ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman  if (LHSTy->isPointerType() && RHSTy->isPointerType()) {
1290ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman    if (E->getOpcode() == BinaryOperator::Sub || E->isEqualityOp()) {
1291efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue LHSValue;
12923068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getLHS(), LHSValue, Info))
12933068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1294a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1295efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LValue RHSValue;
12963068d117951a8df54bae9db039b56201ab10962bAnders Carlsson      if (!EvaluatePointer(E->getRHS(), RHSValue, Info))
12973068d117951a8df54bae9db039b56201ab10962bAnders Carlsson        return false;
1298a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
12995bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // Reject any bases from the normal codepath; we special-case comparisons
13005bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      // to null.
13015bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      if (LHSValue.getLValueBase()) {
13025bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13035bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1304a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (RHSValue.getLValueBase() || !RHSValue.getLValueOffset().isZero())
13055bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13065bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13075bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(LHSValue, bres))
13085bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13095bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        return Success(bres ^ (E->getOpcode() == BinaryOperator::EQ), E);
13105bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      } else if (RHSValue.getLValueBase()) {
13115bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!E->isEqualityOp())
13125bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
1313a73058324197b7bdfd19307965954f626e26199dKen Dyck        if (LHSValue.getLValueBase() || !LHSValue.getLValueOffset().isZero())
13145bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13155bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        bool bres;
13165bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        if (!EvalPointerValueAsBool(RHSValue, bres))
13175bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman          return false;
13185bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman        return Success(bres ^ (E->getOpcode() == BinaryOperator::EQ), E);
13195bc86103767c2abcbfdd6518e0ccbbbb6aa59e0fEli Friedman      }
1320a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1321ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      if (E->getOpcode() == BinaryOperator::Sub) {
13224992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType Type = E->getLHS()->getType();
13234992bdde387c5f033bb450a716eaabc0fda52688Chris Lattner        QualType ElementType = Type->getAs<PointerType>()->getPointeeType();
13243068d117951a8df54bae9db039b56201ab10962bAnders Carlsson
1325a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits ElementSize = CharUnits::One();
1326ce1bca73aef9bbf6359ab8420278203dda81a054Eli Friedman        if (!ElementType->isVoidType() && !ElementType->isFunctionType())
1327a73058324197b7bdfd19307965954f626e26199dKen Dyck          ElementSize = Info.Ctx.getTypeSizeInChars(ElementType);
1328a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1329a73058324197b7bdfd19307965954f626e26199dKen Dyck        CharUnits Diff = LHSValue.getLValueOffset() -
1330a73058324197b7bdfd19307965954f626e26199dKen Dyck                             RHSValue.getLValueOffset();
1331a73058324197b7bdfd19307965954f626e26199dKen Dyck        return Success(Diff / ElementSize, E);
1332ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1333ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      bool Result;
1334ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      if (E->getOpcode() == BinaryOperator::EQ) {
1335ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() == RHSValue.getLValueOffset();
1336267c0ab1b9a15768f3f15abbfc40ce344751c78bEli Friedman      } else {
1337ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman        Result = LHSValue.getLValueOffset() != RHSValue.getLValueOffset();
1338ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      }
1339ad02d7debd03ff275ac8ea27891a4ecccdb78068Eli Friedman      return Success(Result, E);
13403068d117951a8df54bae9db039b56201ab10962bAnders Carlsson    }
13413068d117951a8df54bae9db039b56201ab10962bAnders Carlsson  }
13422ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!LHSTy->isIntegralOrEnumerationType() ||
13432ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor      !RHSTy->isIntegralOrEnumerationType()) {
1344a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // We can't continue from here for non-integral types, and they
1345a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // could potentially confuse the following operations.
1346a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false;
1347a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1348a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
1349a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  // The LHS of a constant expr is always evaluated and needed.
135030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  if (!Visit(E->getLHS()))
1351a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    return false; // error in subexpression.
1352d9f4bcda18bfbf79341edd9d381d4b6a3cffe655Eli Friedman
135342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APValue RHSVal;
135442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!EvaluateIntegerOrLValue(E->getRHS(), RHSVal, Info))
135530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return false;
135642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
135742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like (unsigned long)&a + 4.
135842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->isAdditiveOp() && Result.isLValue() && RHSVal.isInt()) {
1359a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = Result.getLValueOffset();
1360a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits AdditionalOffset = CharUnits::fromQuantity(
1361a73058324197b7bdfd19307965954f626e26199dKen Dyck                                     RHSVal.getInt().getZExtValue());
136242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    if (E->getOpcode() == BinaryOperator::Add)
1363a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset += AdditionalOffset;
136442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    else
1365a73058324197b7bdfd19307965954f626e26199dKen Dyck      Offset -= AdditionalOffset;
1366a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(Result.getLValueBase(), Offset);
136742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
136842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
136942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
137042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // Handle cases like 4 + (unsigned long)&a
137142edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (E->getOpcode() == BinaryOperator::Add &&
137242edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman        RHSVal.isLValue() && Result.isInt()) {
1373a73058324197b7bdfd19307965954f626e26199dKen Dyck    CharUnits Offset = RHSVal.getLValueOffset();
1374a73058324197b7bdfd19307965954f626e26199dKen Dyck    Offset += CharUnits::fromQuantity(Result.getInt().getZExtValue());
1375a73058324197b7bdfd19307965954f626e26199dKen Dyck    Result = APValue(RHSVal.getLValueBase(), Offset);
137642edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman    return true;
137742edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  }
137842edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
137942edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  // All the following cases expect both operands to be an integer
138042edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  if (!Result.isInt() || !RHSVal.isInt())
1381b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner    return false;
1382a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
138342edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman  APSInt& RHS = RHSVal.getInt();
138442edd0d32a729d2735a6fb152ba6bf349bf0a169Eli Friedman
1385a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  switch (E->getOpcode()) {
138632fea9d18cc3658a1b01df5ca6f2ac302625c61dChris Lattner  default:
13870e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
138830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Mul: return Success(Result.getInt() * RHS, E);
138930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Add: return Success(Result.getInt() + RHS, E);
139030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Sub: return Success(Result.getInt() - RHS, E);
139130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::And: return Success(Result.getInt() & RHS, E);
139230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Xor: return Success(Result.getInt() ^ RHS, E);
139330c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::Or:  return Success(Result.getInt() | RHS, E);
1394a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Div:
139554176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
13960e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
139730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() / RHS, E);
1398a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  case BinaryOperator::Rem:
139954176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    if (RHS == 0)
14000e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson      return Error(E->getOperatorLoc(), diag::note_expr_divide_by_zero, E);
140130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() % RHS, E);
14023f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shl: {
140354176fdb044312b4b77c3da6682d3575b3728d30Chris Lattner    // FIXME: Warn about out of range shift amounts!
14041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
140530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
140630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() << SA, E);
14073f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14083f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  case BinaryOperator::Shr: {
14091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    unsigned SA =
141030c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar      (unsigned) RHS.getLimitedValue(Result.getInt().getBitWidth()-1);
141130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(Result.getInt() >> SA, E);
14123f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar  }
14131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
141430c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LT: return Success(Result.getInt() < RHS, E);
141530c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GT: return Success(Result.getInt() > RHS, E);
141630c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::LE: return Success(Result.getInt() <= RHS, E);
141730c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::GE: return Success(Result.getInt() >= RHS, E);
141830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::EQ: return Success(Result.getInt() == RHS, E);
141930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar  case BinaryOperator::NE: return Success(Result.getInt() != RHS, E);
1420b11e77836dd0867955c5abf32baf1c3e6c7f81e1Eli Friedman  }
1421a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
1422a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
1423ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopesbool IntExprEvaluator::VisitConditionalOperator(const ConditionalOperator *E) {
1424a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  bool Cond;
1425a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  if (!HandleConversionToBool(E->getCond(), Cond, Info))
1426ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes    return false;
1427ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
1428a25bd55e63b9ab26ca69390533356aaae7f1a40eNuno Lopes  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
1429ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes}
1430ca7c2eaca35edac066ab7cec15f7ecebbb3731d8Nuno Lopes
14318b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfType(QualType T) {
14325d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14335d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14345d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14355d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14365d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = T->getAs<ReferenceType>())
14375d484e8cf710207010720589d89602233de61d01Sebastian Redl    T = Ref->getPointeeType();
14385d484e8cf710207010720589d89602233de61d01Sebastian Redl
1439e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the alignment.
1440e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  unsigned CharSize = Info.Ctx.Target.getCharWidth();
144118857644059c45da6776f1a288eec7b4cf3a844aDouglas Gregor
14422be586108bb401019647791feca19ea03fd477ceEli Friedman  // __alignof is defined to return the preferred alignment.
14438b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck  return CharUnits::fromQuantity(
14448b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      Info.Ctx.getPreferredTypeAlign(T.getTypePtr()) / CharSize);
1445e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1446e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14478b752f10c394b140f9ef89e049cbad1a7676fc25Ken DyckCharUnits IntExprEvaluator::GetAlignOfExpr(const Expr *E) {
1448af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  E = E->IgnoreParens();
1449af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1450af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  // alignof decl is always accepted, even if it doesn't make sense: we default
14511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // to 1 in those cases.
1452af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
14538b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(DRE->getDecl(),
14548b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1455a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1456af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner  if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
14578b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck    return Info.Ctx.getDeclAlign(ME->getMemberDecl(),
14588b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck                                 /*RefAsPointee*/true);
1459af707ab8fbb9451e8febb8d766f6c043628125c4Chris Lattner
1460e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  return GetAlignOfType(E->getType());
1461e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner}
1462e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
1463e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner
14640518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// VisitSizeAlignOfExpr - Evaluate a sizeof or alignof with a result as the
14650518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl/// expression's type.
14660518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redlbool IntExprEvaluator::VisitSizeOfAlignOfExpr(const SizeOfAlignOfExpr *E) {
1467e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Handle alignof separately.
1468e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!E->isSizeOf()) {
1469e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    if (E->isArgumentType())
14708b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfType(E->getArgumentType()).getQuantity(), E);
1471e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner    else
14728b752f10c394b140f9ef89e049cbad1a7676fc25Ken Dyck      return Success(GetAlignOfExpr(E->getArgumentExpr()).getQuantity(), E);
1473e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  }
1474a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
14750518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl  QualType SrcTy = E->getTypeOfArgument();
14765d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.sizeof]p2: "When applied to a reference or a reference type,
14775d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   the result is the size of the referenced type."
14785d484e8cf710207010720589d89602233de61d01Sebastian Redl  // C++ [expr.alignof]p3: "When alignof is applied to a reference type, the
14795d484e8cf710207010720589d89602233de61d01Sebastian Redl  //   result shall be the alignment of the referenced type."
14805d484e8cf710207010720589d89602233de61d01Sebastian Redl  if (const ReferenceType *Ref = SrcTy->getAs<ReferenceType>())
14815d484e8cf710207010720589d89602233de61d01Sebastian Redl    SrcTy = Ref->getPointeeType();
14820518999d3adcc289997bd974dce90cc97f5c1c44Sebastian Redl
1483131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // sizeof(void), __alignof__(void), sizeof(function) = 1 as a gcc
1484131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  // extension.
1485131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar  if (SrcTy->isVoidType() || SrcTy->isFunctionType())
1486131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(1, E);
1487a1f47c447a919c6a05c63801cb6a52c4c288e2ccEli Friedman
1488fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner  // sizeof(vla) is not a constantexpr: C99 6.5.3.4p2.
1489e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  if (!SrcTy->isConstantSizeType())
1490fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner    return false;
1491f2da9dfef96dc11b7b5effb1d02cb427b2d71599Eli Friedman
1492e9feb475d72ba50dc29cec62a8c47cae721065ebChris Lattner  // Get information about the size.
1493199c3d6cd16aebbb9c7f0d42af9d922c9628bf70Ken Dyck  return Success(Info.Ctx.getTypeSizeInChars(SrcTy).getQuantity(), E);
1494fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner}
1495fcee0019b76f9f368f2b3d6d4048a98232593f29Chris Lattner
14968ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregorbool IntExprEvaluator::VisitOffsetOfExpr(const OffsetOfExpr *E) {
14978ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  CharUnits Result;
14988ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  unsigned n = E->getNumComponents();
14998ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  OffsetOfExpr* OOE = const_cast<OffsetOfExpr*>(E);
15008ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  if (n == 0)
15018ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    return false;
15028ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  QualType CurrentType = E->getTypeSourceInfo()->getType();
15038ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  for (unsigned i = 0; i != n; ++i) {
15048ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    OffsetOfExpr::OffsetOfNode ON = OOE->getComponent(i);
15058ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    switch (ON.getKind()) {
15068ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Array: {
15078ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Expr *Idx = OOE->getIndexExpr(ON.getArrayExprIndex());
15088ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      APSInt IdxResult;
15098ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!EvaluateInteger(Idx, IdxResult, Info))
15108ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15118ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ArrayType *AT = Info.Ctx.getAsArrayType(CurrentType);
15128ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!AT)
15138ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15148ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = AT->getElementType();
15158ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CharUnits ElementSize = Info.Ctx.getTypeSizeInChars(CurrentType);
15168ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      Result += IdxResult.getSExtValue() * ElementSize;
15178ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        break;
15188ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15198ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15208ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Field: {
15218ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      FieldDecl *MemberDecl = ON.getField();
15228ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
15238ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      if (!RT)
15248ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
15258ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      RecordDecl *RD = RT->getDecl();
15268ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
15278ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      unsigned i = 0;
15288ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      // FIXME: It would be nice if we didn't have to loop here!
15298ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      for (RecordDecl::field_iterator Field = RD->field_begin(),
15308ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor                                      FieldEnd = RD->field_end();
15318ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor           Field != FieldEnd; (void)++Field, ++i) {
15328ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        if (*Field == MemberDecl)
15338ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor          break;
15348ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      }
1535cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      assert(i < RL.getFieldCount() && "offsetof field in wrong type");
1536cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1537cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                           RL.getFieldOffset(i) / Info.Ctx.getCharWidth());
15388ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      CurrentType = MemberDecl->getType().getNonReferenceType();
15398ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      break;
15408ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15418ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
15428ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Identifier:
15438ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor      llvm_unreachable("dependent __builtin_offsetof");
1544cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      return false;
1545cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1546cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    case OffsetOfExpr::OffsetOfNode::Base: {
1547cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CXXBaseSpecifier *BaseSpec = ON.getBase();
1548cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (BaseSpec->isVirtual())
1549cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1550cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1551cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the layout of the class whose base we are looking into.
1552cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *RT = CurrentType->getAs<RecordType>();
1553cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!RT)
1554cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1555cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      RecordDecl *RD = RT->getDecl();
1556cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const ASTRecordLayout &RL = Info.Ctx.getASTRecordLayout(RD);
1557cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1558cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Find the base class itself.
1559cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      CurrentType = BaseSpec->getType();
1560cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      const RecordType *BaseRT = CurrentType->getAs<RecordType>();
1561cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      if (!BaseRT)
1562cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor        return false;
1563cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor
1564cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      // Add the offset to the base.
1565cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      Result += CharUnits::fromQuantity(
1566cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                RL.getBaseClassOffset(cast<CXXRecordDecl>(BaseRT->getDecl()))
1567cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor                                        / Info.Ctx.getCharWidth());
1568cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor      break;
1569cc8a5d5f90bbbbcb46f342117b851b7e07ec34f1Douglas Gregor    }
15708ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    }
15718ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  }
15728ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  return Success(Result.getQuantity(), E);
15738ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor}
15748ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1575b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattnerbool IntExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
15764c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // Special case unary operators that do not need their subexpression
15774c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  // evaluated.  offsetof/sizeof/alignof are all special.
157835183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman  if (E->isOffsetOfOp()) {
157935183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    // The AST for offsetof is defined in such a way that we can just
158035183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    // directly Evaluate it as an l-value.
1581efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
158235183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    if (!EvaluateLValue(E->getSubExpr(), LV, Info))
15838ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
158435183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman    if (LV.getLValueBase())
15858ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor        return false;
1586a73058324197b7bdfd19307965954f626e26199dKen Dyck    return Success(LV.getLValueOffset().getQuantity(), E);
158735183aca180a2b9b2c637cd625a40a7e147d6a32Eli Friedman  }
15888ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
1589a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  if (E->getOpcode() == UnaryOperator::LNot) {
1590a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    // LNot's operand isn't necessarily an integer, so we handle it specially.
1591a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    bool bres;
1592a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman    if (!HandleConversionToBool(E->getSubExpr(), bres, Info))
1593a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman      return false;
1594131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(!bres, E);
1595a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman  }
1596a6afa768aa7bd3102a2807aa720917e4a1771e4eEli Friedman
15974fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar  // Only handle integral operations...
15982ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getSubExpr()->getType()->isIntegralOrEnumerationType())
15994fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar    return false;
16004fff4819a913c65ae23cfd389bc47c61919e4e1fDaniel Dunbar
160187eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  // Get the operand value into 'Result'.
160287eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner  if (!Visit(E->getSubExpr()))
160375a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    return false;
1604a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson
160575a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  switch (E->getOpcode()) {
16064c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner  default:
160775a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // Address, indirect, pre/post inc/dec, etc are not valid constant exprs.
160875a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner    // See C99 6.6p3.
16090e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getOperatorLoc(), diag::note_invalid_subexpr_in_ice, E);
161075a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Extension:
16114c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // FIXME: Should extension allow i-c-e extension expressions in its scope?
16124c4867e140327fa3b56306fa03c64c8e6a7c95efChris Lattner    // If so, we could clear the diagnostic ID.
16133f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
161475a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Plus:
16151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The result is always just the subexpr.
16163f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    return true;
161775a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Minus:
161830c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
161930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(-Result.getInt(), E);
162075a4881047deeb3a300ff9293dc6ba8570048bb5Chris Lattner  case UnaryOperator::Not:
162130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    if (!Result.isInt()) return false;
162230c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar    return Success(~Result.getInt(), E);
162306a3675627e3b3c47b49c689c8e404a33144194aAnders Carlsson  }
1624a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
16251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1626732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// HandleCast - This is used to evaluate implicit or explicit casts where the
1627732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner/// result type is integer.
1628a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbarbool IntExprEvaluator::VisitCastExpr(CastExpr *E) {
162982206e267ce6cc709797127616f64672d255b310Anders Carlsson  Expr *SubExpr = E->getSubExpr();
163082206e267ce6cc709797127616f64672d255b310Anders Carlsson  QualType DestType = E->getType();
1631b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  QualType SrcType = SubExpr->getType();
163282206e267ce6cc709797127616f64672d255b310Anders Carlsson
16334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (DestType->isBooleanType()) {
16344efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    bool BoolResult;
16354efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!HandleConversionToBool(SubExpr, BoolResult, Info))
16364efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1637131eb438d8c216b2e2a4f8fa8158ea88b787dc14Daniel Dunbar    return Success(BoolResult, E);
16384efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
16394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1640a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson  // Handle simple integer->integer casts.
16412ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SrcType->isIntegralOrEnumerationType()) {
1642732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner    if (!Visit(SubExpr))
1643b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
1644a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
1645be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!Result.isInt()) {
1646be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      // Only allow casts of lvalues if they are lossless.
1647be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return Info.Ctx.getTypeSize(DestType) == Info.Ctx.getTypeSize(SrcType);
1648be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    }
164930c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar
1650dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType,
165130c37f4d2ee5811e85f692c22fb67d74ddc88079Daniel Dunbar                                      Result.getInt(), Info.Ctx), E);
1652732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  }
16531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1654732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner  // FIXME: Clean this up!
1655b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (SrcType->isPointerType()) {
1656efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
165787eae5ecf94e38baa20d9a327b8f73f8bdc72436Chris Lattner    if (!EvaluatePointer(SubExpr, LV, Info))
1658b542afe02d317411d53b3541946f9f2a8f509a11Chris Lattner      return false;
16594efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1660dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    if (LV.getLValueBase()) {
1661dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      // Only allow based lvalue casts if they are lossless.
1662dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(SrcType))
1663dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar        return false;
1664dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar
1665efdb83e26f9a1fd2566afe54461216cd84814d42John McCall      LV.moveInto(Result);
1666dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar      return true;
1667dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    }
16684efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1669a73058324197b7bdfd19307965954f626e26199dKen Dyck    APSInt AsInt = Info.Ctx.MakeIntValue(LV.getLValueOffset().getQuantity(),
1670a73058324197b7bdfd19307965954f626e26199dKen Dyck                                         SrcType);
1671dd2116462ae311043986ae8b7fba27e68c1b2e66Daniel Dunbar    return Success(HandleIntToIntCast(DestType, SrcType, AsInt, Info.Ctx), E);
16722bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson  }
16734efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1674be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  if (SrcType->isArrayType() || SrcType->isFunctionType()) {
1675be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // This handles double-conversion cases, where there's both
1676be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    // an l-value promotion and an implicit conversion to int.
1677efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
1678be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (!EvaluateLValue(SubExpr, LV, Info))
1679be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1680be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1681be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    if (Info.Ctx.getTypeSize(DestType) != Info.Ctx.getTypeSize(Info.Ctx.VoidPtrTy))
1682be26570e3faa009bdcefedfaf04473e518940520Eli Friedman      return false;
1683be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
1684efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result);
1685be26570e3faa009bdcefedfaf04473e518940520Eli Friedman    return true;
1686be26570e3faa009bdcefedfaf04473e518940520Eli Friedman  }
1687be26570e3faa009bdcefedfaf04473e518940520Eli Friedman
16881725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  if (SrcType->isAnyComplexType()) {
1689f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue C;
16901725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (!EvaluateComplex(SubExpr, C, Info))
16911725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return false;
16921725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    if (C.isComplexFloat())
16931725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleFloatToIntCast(DestType, SrcType,
16941725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                          C.getComplexFloatReal(), Info.Ctx),
16951725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                     E);
16961725f683432715e5afe34d476024bd6f16eac3fcEli Friedman    else
16971725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      return Success(HandleIntToIntCast(DestType, SrcType,
16981725f683432715e5afe34d476024bd6f16eac3fcEli Friedman                                        C.getComplexIntReal(), Info.Ctx), E);
16991725f683432715e5afe34d476024bd6f16eac3fcEli Friedman  }
17002217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle vectors
17012217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1702b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  if (!SrcType->isRealFloatingType())
17030e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1704732b2236ae4a4f11e7642677cebbd169c07ea877Chris Lattner
1705d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat F(0.0);
1706d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(SubExpr, F, Info))
17070e8acbbba71ec6acd5dceb4fcbce63e463e1b755Anders Carlsson    return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
17081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1709b92dac8bc2f6f73919825f9af693a8a7e89ae1d4Daniel Dunbar  return Success(HandleFloatToIntCast(DestType, SrcType, F, Info.Ctx), E);
1710a25ae3d68d84d2b89907f998df6a396549589da5Anders Carlsson}
17112bad1687fe6f00e10767a691a33b070b151902b6Anders Carlsson
1712722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedmanbool IntExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1713722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isAnyComplexType()) {
1714f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1715722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1716722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1717722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntReal(), E);
1718722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1719722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1720722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  return Visit(E->getSubExpr());
1721722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman}
1722722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1723664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedmanbool IntExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1724722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  if (E->getSubExpr()->getType()->isComplexIntegerType()) {
1725f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    ComplexValue LV;
1726722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    if (!EvaluateComplex(E->getSubExpr(), LV, Info) || !LV.isComplexInt())
1727722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman      return Error(E->getExprLoc(), diag::note_invalid_subexpr_in_ice, E);
1728722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman    return Success(LV.getComplexIntImag(), E);
1729722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman  }
1730722c717cd833e410ca6e7976d78baea16995e0c4Eli Friedman
1731664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  if (!E->getSubExpr()->isEvaluatable(Info.Ctx))
1732664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman    Info.EvalResult.HasSideEffects = true;
1733664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman  return Success(0, E);
1734664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman}
1735664a104ba0b8f47b8908ec6af694d9646adba1fcEli Friedman
1736f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
1737d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman// Float Evaluation
1738d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1739d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1740d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmannamespace {
1741770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass FloatExprEvaluator
1742d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  : public StmtVisitor<FloatExprEvaluator, bool> {
1743d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  EvalInfo &Info;
1744d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  APFloat &Result;
1745d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanpublic:
1746d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  FloatExprEvaluator(EvalInfo &info, APFloat &result)
1747d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    : Info(info), Result(result) {}
1748d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1749d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitStmt(Stmt *S) {
1750d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1751d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1752d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1753d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
1754019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  bool VisitCallExpr(const CallExpr *E);
1755d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
17565db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  bool VisitUnaryOperator(const UnaryOperator *E);
1757d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitBinaryOperator(const BinaryOperator *E);
1758d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  bool VisitFloatingLiteral(const FloatingLiteral *E);
17594efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  bool VisitCastExpr(CastExpr *E);
1760ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  bool VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
176167f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool VisitConditionalOperator(ConditionalOperator *E);
17622217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman
1763ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitChooseExpr(const ChooseExpr *E)
1764ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
1765ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  bool VisitUnaryExtension(const UnaryOperator *E)
1766ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
1767abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryReal(const UnaryOperator *E);
1768abd3a857ace59100305790545d1baae5877b8945John McCall  bool VisitUnaryImag(const UnaryOperator *E);
1769ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman
1770abd3a857ace59100305790545d1baae5877b8945John McCall  // FIXME: Missing: array subscript of vector, member of vector,
1771abd3a857ace59100305790545d1baae5877b8945John McCall  //                 ImplicitValueInitExpr
1772d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman};
1773d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman} // end anonymous namespace
1774d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1775d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanstatic bool EvaluateFloat(const Expr* E, APFloat& Result, EvalInfo &Info) {
17767db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isRealFloatingType());
1777d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return FloatExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
1778d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1779d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1780db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCallstatic bool TryEvaluateBuiltinNaN(ASTContext &Context,
1781db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  QualType ResultTy,
1782db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  const Expr *Arg,
1783db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  bool SNaN,
1784db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                  llvm::APFloat &Result) {
1785db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const StringLiteral *S = dyn_cast<StringLiteral>(Arg->IgnoreParenCasts());
1786db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (!S) return false;
1787db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1788db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  const llvm::fltSemantics &Sem = Context.getFloatTypeSemantics(ResultTy);
1789db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1790db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  llvm::APInt fill;
1791db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1792db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  // Treat empty strings as if they were zero.
1793db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (S->getString().empty())
1794db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    fill = llvm::APInt(32, 0);
1795db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else if (S->getString().getAsInteger(0, fill))
1796db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return false;
1797db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1798db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  if (SNaN)
1799db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getSNaN(Sem, false, &fill);
1800db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  else
1801db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    Result = llvm::APFloat::getQNaN(Sem, false, &fill);
1802db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  return true;
1803db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall}
1804db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
1805019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattnerbool FloatExprEvaluator::VisitCallExpr(const CallExpr *E) {
18063c385e5f8d9008fff18597ca302be19fa86e51f6Douglas Gregor  switch (E->isBuiltinCall(Info.Ctx)) {
180734a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner  default: return false;
1808019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_val:
1809019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_valf:
1810019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_huge_vall:
1811019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inf:
1812019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  case Builtin::BI__builtin_inff:
18137cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  case Builtin::BI__builtin_infl: {
18147cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar    const llvm::fltSemantics &Sem =
18157cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar      Info.Ctx.getFloatTypeSemantics(E->getType());
181634a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    Result = llvm::APFloat::getInf(Sem);
181734a74ab81600a40c6324fd76adb724b803dfaf91Chris Lattner    return true;
18187cbed03c00e246682e5292785d01e1c120ce54bdDaniel Dunbar  }
18191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1820db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nans:
1821db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansf:
1822db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall  case Builtin::BI__builtin_nansl:
1823db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1824db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 true, Result);
1825db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall
18269e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nan:
18279e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanf:
18289e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner  case Builtin::BI__builtin_nanl:
18294572baba9d18c275968ac113fd73b0e3c77cccb8Mike Stump    // If this is __builtin_nan() turn this into a nan, otherwise we
18309e62171a25e3a08fb5c49fb370f83faf5ae786f5Chris Lattner    // can't constant fold it.
1831db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall    return TryEvaluateBuiltinNaN(Info.Ctx, E->getType(), E->getArg(0),
1832db7b72a82a6834680ccf1eeb51dc57e6d935c655John McCall                                 false, Result);
18335db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18345db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabs:
18355db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsf:
18365db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_fabsl:
18375db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info))
18385db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18405db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (Result.isNegative())
18415db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      Result.changeSign();
18425db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18435db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysign:
18451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case Builtin::BI__builtin_copysignf:
18465db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case Builtin::BI__builtin_copysignl: {
18475db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    APFloat RHS(0.);
18485db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    if (!EvaluateFloat(E->getArg(0), Result, Info) ||
18495db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar        !EvaluateFloat(E->getArg(1), RHS, Info))
18505db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar      return false;
18515db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.copySign(RHS);
18525db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18535db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
1854019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner  }
1855019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner}
1856019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1857abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryReal(const UnaryOperator *E) {
1858abd3a857ace59100305790545d1baae5877b8945John McCall  ComplexValue CV;
1859abd3a857ace59100305790545d1baae5877b8945John McCall  if (!EvaluateComplex(E->getSubExpr(), CV, Info))
1860abd3a857ace59100305790545d1baae5877b8945John McCall    return false;
1861abd3a857ace59100305790545d1baae5877b8945John McCall  Result = CV.FloatReal;
1862abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1863abd3a857ace59100305790545d1baae5877b8945John McCall}
1864abd3a857ace59100305790545d1baae5877b8945John McCall
1865abd3a857ace59100305790545d1baae5877b8945John McCallbool FloatExprEvaluator::VisitUnaryImag(const UnaryOperator *E) {
1866abd3a857ace59100305790545d1baae5877b8945John McCall  ComplexValue CV;
1867abd3a857ace59100305790545d1baae5877b8945John McCall  if (!EvaluateComplex(E->getSubExpr(), CV, Info))
1868abd3a857ace59100305790545d1baae5877b8945John McCall    return false;
1869abd3a857ace59100305790545d1baae5877b8945John McCall  Result = CV.FloatImag;
1870abd3a857ace59100305790545d1baae5877b8945John McCall  return true;
1871abd3a857ace59100305790545d1baae5877b8945John McCall}
1872abd3a857ace59100305790545d1baae5877b8945John McCall
18735db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbarbool FloatExprEvaluator::VisitUnaryOperator(const UnaryOperator *E) {
1874a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes  if (E->getOpcode() == UnaryOperator::Deref)
1875a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes    return false;
1876a468d34bed16861f25aff6c8354f4e75d3358c1aNuno Lopes
18775db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  if (!EvaluateFloat(E->getSubExpr(), Result, Info))
18785db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return false;
18795db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar
18805db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  switch (E->getOpcode()) {
18815db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  default: return false;
18821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  case UnaryOperator::Plus:
18835db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18845db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  case UnaryOperator::Minus:
18855db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    Result.changeSign();
18865db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar    return true;
18875db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  }
18885db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar}
1889019f4e858e78587f2241ff1a76c747d7bcd7578cChris Lattner
1890d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
18917f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  if (E->getOpcode() == BinaryOperator::Comma) {
18927f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!EvaluateFloat(E->getRHS(), Result, Info))
18937f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      return false;
18947f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
18957f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // If we can't evaluate the LHS, it might have side effects;
18967f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    // conservatively mark it.
18977f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    if (!E->getLHS()->isEvaluatable(Info.Ctx))
18987f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman      Info.EvalResult.HasSideEffects = true;
18997f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
19007f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman    return true;
19017f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman  }
19027f92f0362ef2cf218bc19bb83e1a97dd254b5527Eli Friedman
1903d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // FIXME: Diagnostics?  I really don't understand how the warnings
1904d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  // and errors are supposed to work.
19055db4b3f3ed9f769d5b02c1d1ccc52bfd71fb9afbDaniel Dunbar  APFloat RHS(0.0);
1906d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getLHS(), Result, Info))
1907d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1908d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  if (!EvaluateFloat(E->getRHS(), RHS, Info))
1909d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return false;
1910d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1911d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  switch (E->getOpcode()) {
1912d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  default: return false;
1913d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Mul:
1914d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.multiply(RHS, APFloat::rmNearestTiesToEven);
1915d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1916d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Add:
1917d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.add(RHS, APFloat::rmNearestTiesToEven);
1918d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1919d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Sub:
1920d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.subtract(RHS, APFloat::rmNearestTiesToEven);
1921d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1922d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  case BinaryOperator::Div:
1923d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    Result.divide(RHS, APFloat::rmNearestTiesToEven);
1924d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman    return true;
1925d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  }
1926d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1927d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
1928d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedmanbool FloatExprEvaluator::VisitFloatingLiteral(const FloatingLiteral *E) {
1929d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  Result = E->getValue();
1930d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman  return true;
1931d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman}
1932d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman
19334efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedmanbool FloatExprEvaluator::VisitCastExpr(CastExpr *E) {
19344efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Expr* SubExpr = E->getSubExpr();
19351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19362ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (SubExpr->getType()->isIntegralOrEnumerationType()) {
19374efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    APSInt IntResult;
19383f7d995390009fede92b333a040da80e1ce90997Daniel Dunbar    if (!EvaluateInteger(SubExpr, IntResult, Info))
19394efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
19401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Result = HandleIntToFloatCast(E->getType(), SubExpr->getType(),
1941a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                  IntResult, Info.Ctx);
19424efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19434efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19444efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  if (SubExpr->getType()->isRealFloatingType()) {
19454efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    if (!Visit(SubExpr))
19464efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman      return false;
1947a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    Result = HandleFloatToFloatCast(E->getType(), SubExpr->getType(),
1948a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar                                    Result, Info.Ctx);
19494efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman    return true;
19504efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  }
19512217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  // FIXME: Handle complex types
19524efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
19534efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return false;
19544efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19554efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
1956ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregorbool FloatExprEvaluator::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19574efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  Result = APFloat::getZero(Info.Ctx.getFloatTypeSemantics(E->getType()));
19584efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman  return true;
19594efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman}
19604efaa276bc0ce8f7baf6138ead11915f3e3e58d9Eli Friedman
196167f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedmanbool FloatExprEvaluator::VisitConditionalOperator(ConditionalOperator *E) {
196267f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  bool Cond;
196367f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  if (!HandleConversionToBool(E->getCond(), Cond, Info))
196467f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman    return false;
196567f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
196667f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman  return Visit(Cond ? E->getTrueExpr() : E->getFalseExpr());
196767f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman}
196867f85fcdab64f36233a7211f963ed1cccffcbfb8Eli Friedman
1969d8bfe7f25a695ca947effbccdf9ecbe3e018e221Eli Friedman//===----------------------------------------------------------------------===//
1970a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar// Complex Evaluation (for float and integer)
19719ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
19729ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
19739ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonnamespace {
1974770b4a8834670e9427d3ce5a1a8472eb86f45fd2Benjamin Kramerclass ComplexExprEvaluator
1975f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  : public StmtVisitor<ComplexExprEvaluator, bool> {
19769ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  EvalInfo &Info;
1977f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue &Result;
19781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19799ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlssonpublic:
1980f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexExprEvaluator(EvalInfo &info, ComplexValue &Result)
1981f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    : Info(info), Result(Result) {}
19821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19839ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
19849ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //                            Visitor Methods
19859ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  //===--------------------------------------------------------------------===//
19869ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1987f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitStmt(Stmt *S) {
1988f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
19899ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
19901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1991f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitParenExpr(ParenExpr *E) { return Visit(E->getSubExpr()); }
19929ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
1993f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitImaginaryLiteral(ImaginaryLiteral *E) {
1994a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    Expr* SubExpr = E->getSubExpr();
1995a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
1996a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (SubExpr->getType()->isRealFloatingType()) {
1997f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.makeComplexFloat();
1998f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APFloat &Imag = Result.FloatImag;
1999f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateFloat(SubExpr, Imag, Info))
2000f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2002f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.FloatReal = APFloat(Imag.getSemantics());
2003f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2004a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
20051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      assert(SubExpr->getType()->isIntegerType() &&
2006a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar             "Unexpected imaginary literal.");
2007a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
2008f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.makeComplexInt();
2009f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APSInt &Imag = Result.IntImag;
2010f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateInteger(SubExpr, Imag, Info))
2011f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2013f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      Result.IntReal = APSInt(Imag.getBitWidth(), !Imag.isSigned());
2014f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      return true;
2015a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
20169ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson  }
20179ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2018f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitCastExpr(CastExpr *E) {
2019ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    Expr* SubExpr = E->getSubExpr();
2020183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall    QualType EltType = E->getType()->getAs<ComplexType>()->getElementType();
2021a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    QualType SubType = SubExpr->getType();
2022ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2023a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    if (SubType->isRealFloatingType()) {
2024f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APFloat &Real = Result.FloatReal;
2025f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateFloat(SubExpr, Real, Info))
2026f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
20271725f683432715e5afe34d476024bd6f16eac3fcEli Friedman
20281725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2029f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2030f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Real = HandleFloatToFloatCast(EltType, SubType, Real, Info.Ctx);
2031f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatImag = APFloat(Real.getSemantics());
2032f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20331725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2034f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2035f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntReal = HandleFloatToIntCast(EltType, SubType, Real, Info.Ctx);
2036f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntImag = APSInt(Result.IntReal.getBitWidth(),
2037f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                !Result.IntReal.isSigned());
2038f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20391725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2040a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar    } else if (SubType->isIntegerType()) {
2041f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      APSInt &Real = Result.IntReal;
2042f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!EvaluateInteger(SubExpr, Real, Info))
2043f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
2044a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
20451725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      if (EltType->isRealFloatingType()) {
2046f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexFloat();
2047f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatReal
2048f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          = HandleIntToFloatCast(EltType, SubType, Real, Info.Ctx);
2049f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.FloatImag = APFloat(Result.FloatReal.getSemantics());
2050f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20511725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      } else {
2052f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.makeComplexInt();
2053f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Real = HandleIntToIntCast(EltType, SubType, Real, Info.Ctx);
2054f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        Result.IntImag = APSInt(Real.getBitWidth(), !Real.isSigned());
2055f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return true;
20561725f683432715e5afe34d476024bd6f16eac3fcEli Friedman      }
2057183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall    } else if (const ComplexType *CT = SubType->getAs<ComplexType>()) {
2058f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (!Visit(SubExpr))
2059f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        return false;
2060a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2061a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      QualType SrcType = CT->getElementType();
2062a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar
2063f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      if (Result.isComplexFloat()) {
2064a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
2065f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexFloat();
2066f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatReal = HandleFloatToFloatCast(EltType, SrcType,
2067f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Result.FloatReal,
2068f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Info.Ctx);
2069f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatImag = HandleFloatToFloatCast(EltType, SrcType,
2070f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Result.FloatImag,
2071f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                    Info.Ctx);
2072f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2073a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
2074f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexInt();
2075f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntReal = HandleFloatToIntCast(EltType, SrcType,
2076f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Result.FloatReal,
2077f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Info.Ctx);
2078f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntImag = HandleFloatToIntCast(EltType, SrcType,
2079f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Result.FloatImag,
2080f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                Info.Ctx);
2081f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2082a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
2083a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      } else {
2084f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall        assert(Result.isComplexInt() && "Invalid evaluate result.");
2085a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        if (EltType->isRealFloatingType()) {
2086f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexFloat();
2087f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatReal = HandleIntToFloatCast(EltType, SrcType,
2088f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Result.IntReal,
2089f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Info.Ctx);
2090f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.FloatImag = HandleIntToFloatCast(EltType, SrcType,
2091f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Result.IntImag,
2092f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                                  Info.Ctx);
2093f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2094a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        } else {
2095f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.makeComplexInt();
2096f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntReal = HandleIntToIntCast(EltType, SrcType,
2097f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Result.IntReal,
2098f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Info.Ctx);
2099f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          Result.IntImag = HandleIntToIntCast(EltType, SrcType,
2100f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Result.IntImag,
2101f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                                              Info.Ctx);
2102f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall          return true;
2103a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar        }
2104a2cfd34952204c9a160fe1a5da5ba2f231df891dDaniel Dunbar      }
2105ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    }
2106ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2107ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson    // FIXME: Handle more casts.
2108f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2109ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
21101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2111f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitBinaryOperator(const BinaryOperator *E);
2112f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitChooseExpr(const ChooseExpr *E)
2113ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getChosenSubExpr(Info.Ctx)); }
2114f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  bool VisitUnaryExtension(const UnaryOperator *E)
2115ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman    { return Visit(E->getSubExpr()); }
2116ba98d6bb414861965a1f22628494ea046785ecd4Eli Friedman  // FIXME Missing: unary +/-/~, binary div, ImplicitValueInitExpr,
21172217c87bdc5ab357046a5453bdb06f469c41024eEli Friedman  //                conditional ?:, comma
21189ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson};
21199ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson} // end anonymous namespace
21209ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2121f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallstatic bool EvaluateComplex(const Expr *E, ComplexValue &Result,
2122f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall                            EvalInfo &Info) {
21237db7acbbb84b82d0522266a50ebabc3a52a9e5d1John McCall  assert(E->getType()->isAnyComplexType());
2124f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return ComplexExprEvaluator(Info, Result).Visit(const_cast<Expr*>(E));
21259ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson}
21269ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson
2127f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCallbool ComplexExprEvaluator::VisitBinaryOperator(const BinaryOperator *E) {
2128f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  if (!Visit(E->getLHS()))
2129f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
21301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2131f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  ComplexValue RHS;
2132a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar  if (!EvaluateComplex(E->getRHS(), RHS, Info))
2133f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall    return false;
2134a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar
21353f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  assert(Result.isComplexFloat() == RHS.isComplexFloat() &&
21363f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         "Invalid operands to binary operator.");
2137ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  switch (E->getOpcode()) {
2138f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  default: return false;
2139ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Add:
2140a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2141a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().add(RHS.getComplexFloatReal(),
2142a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2143a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().add(RHS.getComplexFloatImag(),
2144a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                       APFloat::rmNearestTiesToEven);
2145a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2146a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() += RHS.getComplexIntReal();
2147a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() += RHS.getComplexIntImag();
2148a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21493f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
2150ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  case BinaryOperator::Sub:
2151a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    if (Result.isComplexFloat()) {
2152a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatReal().subtract(RHS.getComplexFloatReal(),
2153a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2154a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexFloatImag().subtract(RHS.getComplexFloatImag(),
2155a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar                                            APFloat::rmNearestTiesToEven);
2156a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    } else {
2157a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntReal() -= RHS.getComplexIntReal();
2158a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar      Result.getComplexIntImag() -= RHS.getComplexIntImag();
2159a5fd07bbc5e4bae542c06643da3fbfe4967a9379Daniel Dunbar    }
21603f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
21613f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar  case BinaryOperator::Mul:
21623f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    if (Result.isComplexFloat()) {
2163f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
21643f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_r = LHS.getComplexFloatReal();
21653f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &LHS_i = LHS.getComplexFloatImag();
21663f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_r = RHS.getComplexFloatReal();
21673f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat &RHS_i = RHS.getComplexFloatImag();
21681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
21693f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      APFloat Tmp = LHS_r;
21703f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
21713f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal() = Tmp;
21723f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
21733f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
21743f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatReal().subtract(Tmp, APFloat::rmNearestTiesToEven);
21753f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar
21763f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_r;
21773f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_i, APFloat::rmNearestTiesToEven);
21783f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag() = Tmp;
21793f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp = LHS_i;
21803f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Tmp.multiply(RHS_r, APFloat::rmNearestTiesToEven);
21813f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar      Result.getComplexFloatImag().add(Tmp, APFloat::rmNearestTiesToEven);
21823f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    } else {
2183f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall      ComplexValue LHS = Result;
21841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntReal() =
21853f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntReal() -
21863f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntImag());
21871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      Result.getComplexIntImag() =
21883f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar        (LHS.getComplexIntReal() * RHS.getComplexIntImag() +
21893f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar         LHS.getComplexIntImag() * RHS.getComplexIntReal());
21903f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    }
21913f2798757c9ee353e207e18115e2e966432a4beeDaniel Dunbar    break;
2192ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson  }
2193ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
2194f4cf1a18d09d57b757b3cb47eab36c1457091ef7John McCall  return true;
2195ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson}
2196ccc3fce5697e33f005990f9795e1c7cb8b4559ecAnders Carlsson
21979ad16aebc0e840a5e7d425da72eb6cbe25e4b58cAnders Carlsson//===----------------------------------------------------------------------===//
21986ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner// Top level Expr::Evaluate method.
2199f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner//===----------------------------------------------------------------------===//
2200f5eeb055ecbadbc25c83df0867cdada2c2559dcfChris Lattner
220142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// Evaluate - Return true if this is a constant which we can fold using
220242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// any crazy technique (that has nothing to do with language standards) that
220342c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// we want to.  If this function returns true, it returns the folded constant
220442c8f87eb60958170c46767273bf93e6c96125bfJohn McCall/// in Result.
220542c8f87eb60958170c46767273bf93e6c96125bfJohn McCallbool Expr::Evaluate(EvalResult &Result, ASTContext &Ctx) const {
220642c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  const Expr *E = this;
220742c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  EvalInfo Info(Ctx, Result);
2208efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  if (E->getType()->isVectorType()) {
2209efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateVector(E, Info.EvalResult.Val, Info))
221059b5da6d853b4368b984700315adf7b37de05764Nate Begeman      return false;
2211efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isIntegerType()) {
2212efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!IntExprEvaluator(Info, Info.EvalResult.Val).Visit(const_cast<Expr*>(E)))
22136dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22140f2b692bb10be35fdc60d0a72a847bdd73124670John McCall    if (Result.Val.isLValue() && !IsGlobalLValue(Result.Val.getLValueBase()))
22150f2b692bb10be35fdc60d0a72a847bdd73124670John McCall      return false;
2216efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->hasPointerRepresentation()) {
2217efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LValue LV;
2218efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluatePointer(E, LV, Info))
22196dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
2220e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    if (!IsGlobalLValue(LV.Base))
222142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      return false;
2222efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Info.EvalResult.Val);
2223efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isRealFloatingType()) {
2224efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    llvm::APFloat F(0.0);
2225efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateFloat(E, F, Info))
22266dde0d5dc09f45f4d9508c964703e36fef1a0198Anders Carlsson      return false;
22271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2228efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    Info.EvalResult.Val = APValue(F);
2229efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  } else if (E->getType()->isAnyComplexType()) {
2230efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    ComplexValue C;
2231efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    if (!EvaluateComplex(E, C, Info))
2232660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump      return false;
2233efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    C.moveInto(Info.EvalResult.Val);
2234660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  } else
2235660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump    return false;
2236660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2237660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump  return true;
2238660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump}
2239660e6f79a138a30a437c02142f23e7ef4eb21b2eMike Stump
2240cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCallbool Expr::EvaluateAsBooleanCondition(bool &Result, ASTContext &Ctx) const {
2241cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalResult Scratch;
2242cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  EvalInfo Info(Ctx, Scratch);
2243cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
2244cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall  return HandleConversionToBool(this, Result, Info);
2245cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall}
2246cd7a445c6b46c5585580dfb652300c8483c0cb6bJohn McCall
22471b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlssonbool Expr::EvaluateAsLValue(EvalResult &Result, ASTContext &Ctx) const {
22481b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson  EvalInfo Info(Ctx, Result);
22491b78276a75a5a0f496a82429c1ff9604d622a76dAnders Carlsson
2250efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  LValue LV;
225142c8f87eb60958170c46767273bf93e6c96125bfJohn McCall  if (EvaluateLValue(this, LV, Info) &&
225242c8f87eb60958170c46767273bf93e6c96125bfJohn McCall      !Result.HasSideEffects &&
2253e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara      IsGlobalLValue(LV.Base)) {
2254e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    LV.moveInto(Result.Val);
2255e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara    return true;
2256e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  }
2257e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  return false;
2258e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara}
2259e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2260e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnarabool Expr::EvaluateAsAnyLValue(EvalResult &Result, ASTContext &Ctx) const {
2261e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  EvalInfo Info(Ctx, Result);
2262e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2263e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  LValue LV;
2264e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara  if (EvaluateLValue(this, LV, Info)) {
2265efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    LV.moveInto(Result.Val);
2266efdb83e26f9a1fd2566afe54461216cd84814d42John McCall    return true;
2267efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  }
2268efdb83e26f9a1fd2566afe54461216cd84814d42John McCall  return false;
2269b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman}
2270b2f295c8050fb8c141bf2cf38eed0a56e99d0092Eli Friedman
22716ee7aa154e8bbb21a21254293410b944f78b0bfeChris Lattner/// isEvaluatable - Call Evaluate to see if this expression can be constant
227245b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner/// folded, but discard the result.
227345b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattnerbool Expr::isEvaluatable(ASTContext &Ctx) const {
22744fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  EvalResult Result;
22754fdfb0965b396f2778091f7e6c051d17ff9791baAnders Carlsson  return Evaluate(Result, Ctx) && !Result.HasSideEffects;
227645b6b9d080ac56917337d73d8f1cd6374b27b05dChris Lattner}
227751fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
2278393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanianbool Expr::HasSideEffects(ASTContext &Ctx) const {
2279393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  Expr::EvalResult Result;
2280393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  EvalInfo Info(Ctx, Result);
2281393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian  return HasSideEffect(Info).Visit(const_cast<Expr*>(this));
2282393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian}
2283393c247fe025ccb5f914e37e948192ea86faef8cFariborz Jahanian
228451fe996231b1d7199f76e4005ff4c943d5deeecdAnders CarlssonAPSInt Expr::EvaluateAsInt(ASTContext &Ctx) const {
22851c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  EvalResult EvalResult;
22861c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  bool Result = Evaluate(EvalResult, Ctx);
2287f185319f25efd6094870f287030270fad26085baDaniel Dunbar  Result = Result;
228851fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson  assert(Result && "Could not evaluate expression");
22891c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  assert(EvalResult.Val.isInt() && "Expression did not evaluate to integer");
229051fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson
22911c0cfd4599e816cfd7a8f348286bf0ad79652ffcAnders Carlsson  return EvalResult.Val.getInt();
229251fe996231b1d7199f76e4005ff4c943d5deeecdAnders Carlsson}
2293d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2294e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara bool Expr::EvalResult::isGlobalLValue() const {
2295e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   assert(Val.isLValue());
2296e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara   return IsGlobalLValue(Val.getLValueBase());
2297e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara }
2298e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2299e17a6436b429e4b18a5e157061fb13bbc677b3b0Abramo Bagnara
2300d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// isIntegerConstantExpr - this recursive routine will test if an expression is
2301d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// an integer constant expression.
2302d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2303d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Pass up a reason why! Invalid operation in i-c-e, division by zero,
2304d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// comma, etc
2305d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall///
2306d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// FIXME: Handle offsetof.  Two things to do:  Handle GCC's __builtin_offsetof
2307d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// to support gcc 4.0+  and handle the idiom GCC recognizes with a null pointer
2308d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall/// cast+dereference.
2309d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2310d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// CheckICE - This function does the fundamental ICE checking: the returned
2311d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// ICEDiag contains a Val of 0, 1, or 2, and a possibly null SourceLocation.
2312d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Note that to reduce code duplication, this helper does no evaluation
2313d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// itself; the caller checks whether the expression is evaluatable, and
2314d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// in the rare cases where CheckICE actually cares about the evaluated
2315d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// value, it calls into Evalute.
2316d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//
2317d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// Meanings of Val:
2318d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 0: This expression is an ICE if it can be evaluated by Evaluate.
2319d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 1: This expression is not an ICE, but if it isn't evaluated, it's
2320d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    a legal subexpression for an ICE. This return value is used to handle
2321d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall//    the comma operator in C99 mode.
2322d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall// 2: This expression is not an ICE, and is not a legal subexpression for one.
2323d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
23243c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmannamespace {
23253c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
2326d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstruct ICEDiag {
2327d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  unsigned Val;
2328d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  SourceLocation Loc;
2329d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2330d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  public:
2331d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag(unsigned v, SourceLocation l) : Val(v), Loc(l) {}
2332d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag() : Val(0) {}
2333d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall};
2334d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
23353c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman}
23363c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohman
23373c46e8db99196179b30e7ac5c20c4efd5f3926d7Dan Gohmanstatic ICEDiag NoDiag() { return ICEDiag(); }
2338d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2339d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckEvalInICE(const Expr* E, ASTContext &Ctx) {
2340d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Expr::EvalResult EVResult;
2341d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2342d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      !EVResult.Val.isInt()) {
2343d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2344d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2345d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return NoDiag();
2346d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2347d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2348d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallstatic ICEDiag CheckICE(const Expr* E, ASTContext &Ctx) {
2349d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!E->isValueDependent() && "Should not see value dependent exprs!");
23502ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor  if (!E->getType()->isIntegralOrEnumerationType()) {
2351d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2352d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2353d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2354d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  switch (E->getStmtClass()) {
2355d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define STMT(Node, Base) case Expr::Node##Class:
2356d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#define EXPR(Node, Base)
2357d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall#include "clang/AST/StmtNodes.inc"
2358d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::PredefinedExprClass:
2359d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::FloatingLiteralClass:
2360d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImaginaryLiteralClass:
2361d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StringLiteralClass:
2362d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ArraySubscriptExprClass:
2363d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::MemberExprClass:
2364d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundAssignOperatorClass:
2365d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CompoundLiteralExprClass:
2366d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ExtVectorElementExprClass:
2367d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::InitListExprClass:
2368d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DesignatedInitExprClass:
2369d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitValueInitExprClass:
2370d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenListExprClass:
2371d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::VAArgExprClass:
2372d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::AddrLabelExprClass:
2373d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::StmtExprClass:
2374d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXMemberCallExprClass:
2375d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDynamicCastExprClass:
2376d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTypeidExprClass:
2377d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNullPtrLiteralExprClass:
2378d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThisExprClass:
2379d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXThrowExprClass:
2380d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXNewExprClass:
2381d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDeleteExprClass:
2382d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXPseudoDestructorExprClass:
2383d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedLookupExprClass:
2384d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DependentScopeDeclRefExprClass:
2385d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstructExprClass:
2386d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindTemporaryExprClass:
2387d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBindReferenceExprClass:
2388d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXExprWithTemporariesClass:
2389d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXTemporaryObjectExprClass:
2390d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXUnresolvedConstructExprClass:
2391d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDependentScopeMemberExprClass:
2392d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnresolvedMemberExprClass:
2393d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCStringLiteralClass:
2394d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCEncodeExprClass:
2395d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCMessageExprClass:
2396d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSelectorExprClass:
2397d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCProtocolExprClass:
2398d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIvarRefExprClass:
2399d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCPropertyRefExprClass:
2400d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCImplicitSetterGetterRefExprClass:
2401d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCSuperExprClass:
2402d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ObjCIsaExprClass:
2403d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ShuffleVectorExprClass:
2404d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockExprClass:
2405d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BlockDeclRefExprClass:
2406d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::NoStmtClass:
2407d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2408d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2409d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::GNUNullExprClass:
2410d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // GCC considers the GNU __null value to be an integral constant expression.
2411d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2412d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2413d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ParenExprClass:
2414d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ParenExpr>(E)->getSubExpr(), Ctx);
2415d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::IntegerLiteralClass:
2416d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CharacterLiteralClass:
2417d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXBoolLiteralExprClass:
2418ed8abf18329df67b0abcbb3a10458bd8c1d2a595Douglas Gregor  case Expr::CXXScalarValueInitExprClass:
2419d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::TypesCompatibleExprClass:
2420d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryTypeTraitExprClass:
2421d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2422d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CallExprClass:
2423d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXOperatorCallExprClass: {
2424d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const CallExpr *CE = cast<CallExpr>(E);
2425d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CE->isBuiltinCall(Ctx))
2426d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2427d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2428d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2429d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::DeclRefExprClass:
2430d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<EnumConstantDecl>(cast<DeclRefExpr>(E)->getDecl()))
2431d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2432d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Ctx.getLangOptions().CPlusPlus &&
2433d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        E->getType().getCVRQualifiers() == Qualifiers::Const) {
2434d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
2435d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2436d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Parameter variables are never constants.  Without this check,
2437d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // getAnyInitializer() can find a default argument, which leads
2438d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // to chaos.
2439d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (isa<ParmVarDecl>(D))
2440d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2441d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2442d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // C++ 7.1.5.1p2
2443d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   A variable of non-volatile const-qualified integral or enumeration
2444d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      //   type initialized by an ICE can be used in ICEs.
2445d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (const VarDecl *Dcl = dyn_cast<VarDecl>(D)) {
2446d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Qualifiers Quals = Ctx.getCanonicalType(Dcl->getType()).getQualifiers();
2447d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Quals.hasVolatile() || !Quals.hasConst())
2448d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2449d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2450d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Look for a declaration of this variable that has an initializer.
2451d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const VarDecl *ID = 0;
2452d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        const Expr *Init = Dcl->getAnyInitializer(ID);
2453d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Init) {
2454d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (ID->isInitKnownICE()) {
2455d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // We have already checked whether this subexpression is an
2456d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            // integral constant expression.
2457d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (ID->isInitICE())
2458d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return NoDiag();
2459d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            else
2460d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2461d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2462d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2463d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // It's an ICE whether or not the definition we found is
2464d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // out-of-line.  See DR 721 and the discussion in Clang PR
2465d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // 6206 for details.
2466d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2467d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (Dcl->isCheckingICE()) {
2468d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(2, cast<DeclRefExpr>(E)->getLocation());
2469d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2470d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2471d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setCheckingICE();
2472d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          ICEDiag Result = CheckICE(Init, Ctx);
2473d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // Cache the result of the ICE test.
2474d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Dcl->setInitKnownICE(Result.Val == 0);
2475d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return Result;
2476d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2477d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2478d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2479d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2480d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::UnaryOperatorClass: {
2481d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const UnaryOperator *Exp = cast<UnaryOperator>(E);
2482d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
2483d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PostInc:
2484d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PostDec:
2485d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PreInc:
2486d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::PreDec:
2487d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::AddrOf:
2488d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Deref:
2489d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2490d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Extension:
2491d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::LNot:
2492d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Plus:
2493d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Minus:
2494d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Not:
2495d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Real:
2496d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::Imag:
2497d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(Exp->getSubExpr(), Ctx);
2498d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case UnaryOperator::OffsetOf:
2499d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      break;
2500d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2501d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2502d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // OffsetOf falls through here.
2503d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2504d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::OffsetOfExprClass: {
2505d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Note that per C99, offsetof must be an ICE. And AFAIK, using
2506d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // Evaluate matches the proposed gcc behavior for cases like
2507d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // "offsetof(struct s{int x[4];}, x[!.0])".  This doesn't affect
2508d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // compliance: we should warn earlier for offsetof expressions with
2509d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // array subscripts that aren't ICEs, and if the array subscripts
2510d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      // are ICEs, the value of the offsetof must be an integer constant.
2511d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckEvalInICE(E, Ctx);
2512d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2513d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::SizeOfAlignOfExprClass: {
2514d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const SizeOfAlignOfExpr *Exp = cast<SizeOfAlignOfExpr>(E);
2515d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->isSizeOf() && Exp->getTypeOfArgument()->isVariableArrayType())
2516d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2517d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return NoDiag();
2518d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2519d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::BinaryOperatorClass: {
2520d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const BinaryOperator *Exp = cast<BinaryOperator>(E);
2521d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    switch (Exp->getOpcode()) {
2522d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::PtrMemD:
2523d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::PtrMemI:
2524d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Assign:
2525d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::MulAssign:
2526d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::DivAssign:
2527d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::RemAssign:
2528d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::AddAssign:
2529d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::SubAssign:
2530d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::ShlAssign:
2531d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::ShrAssign:
2532d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::AndAssign:
2533d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::XorAssign:
2534d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::OrAssign:
2535d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return ICEDiag(2, E->getLocStart());
2536d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2537d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Mul:
2538d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Div:
2539d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Rem:
2540d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Add:
2541d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Sub:
2542d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Shl:
2543d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Shr:
2544d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LT:
2545d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::GT:
2546d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LE:
2547d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::GE:
2548d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::EQ:
2549d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::NE:
2550d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::And:
2551d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Xor:
2552d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Or:
2553d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::Comma: {
2554d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2555d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2556d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (Exp->getOpcode() == BinaryOperator::Div ||
2557d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          Exp->getOpcode() == BinaryOperator::Rem) {
2558d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Evaluate gives an error for undefined Div/Rem, so make sure
2559d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // we don't evaluate one.
2560d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (LHSResult.Val != 2 && RHSResult.Val != 2) {
2561d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          llvm::APSInt REval = Exp->getRHS()->EvaluateAsInt(Ctx);
2562d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval == 0)
2563d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2564d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (REval.isSigned() && REval.isAllOnesValue()) {
2565d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            llvm::APSInt LEval = Exp->getLHS()->EvaluateAsInt(Ctx);
2566d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            if (LEval.isMinSignedValue())
2567d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall              return ICEDiag(1, E->getLocStart());
2568d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          }
2569d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2570d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2571d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (Exp->getOpcode() == BinaryOperator::Comma) {
2572d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (Ctx.getLangOptions().C99) {
2573d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // C99 6.6p3 introduces a strange edge case: comma can be in an ICE
2574d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // if it isn't evaluated.
2575d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          if (LHSResult.Val == 0 && RHSResult.Val == 0)
2576d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            return ICEDiag(1, E->getLocStart());
2577d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        } else {
2578d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          // In both C89 and C++, commas in ICEs are illegal.
2579d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2580d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2581d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2582d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2583d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2584d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2585d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2586d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LAnd:
2587d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    case BinaryOperator::LOr: {
2588d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag LHSResult = CheckICE(Exp->getLHS(), Ctx);
2589d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      ICEDiag RHSResult = CheckICE(Exp->getRHS(), Ctx);
2590d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val == 0 && RHSResult.Val == 1) {
2591d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // Rare case where the RHS has a comma "side-effect"; we need
2592d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // to actually check the condition to see whether the side
2593d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        // with the comma is evaluated.
2594d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if ((Exp->getOpcode() == BinaryOperator::LAnd) !=
2595d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            (Exp->getLHS()->EvaluateAsInt(Ctx) == 0))
2596d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return RHSResult;
2597d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2598d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2599d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2600d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (LHSResult.Val >= RHSResult.Val)
2601d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return LHSResult;
2602d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return RHSResult;
2603d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2604d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2605d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2606d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ImplicitCastExprClass:
2607d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CStyleCastExprClass:
2608d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXFunctionalCastExprClass:
2609d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXStaticCastExprClass:
2610d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXReinterpretCastExprClass:
2611d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXConstCastExprClass: {
2612d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const Expr *SubExpr = cast<CastExpr>(E)->getSubExpr();
26132ade35e2cfd554e49d35a52047cea98a82787af9Douglas Gregor    if (SubExpr->getType()->isIntegralOrEnumerationType())
2614d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CheckICE(SubExpr, Ctx);
2615d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (isa<FloatingLiteral>(SubExpr->IgnoreParens()))
2616d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2617d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return ICEDiag(2, E->getLocStart());
2618d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2619d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ConditionalOperatorClass: {
2620d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    const ConditionalOperator *Exp = cast<ConditionalOperator>(E);
2621d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // If the condition (ignoring parens) is a __builtin_constant_p call,
2622d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // then only the true side is actually considered in an integer constant
2623d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // expression, and it is fully evaluated.  This is an important GNU
2624d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // extension.  See GCC PR38377 for discussion.
2625d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (const CallExpr *CallCE
2626d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        = dyn_cast<CallExpr>(Exp->getCond()->IgnoreParenCasts()))
2627d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      if (CallCE->isBuiltinCall(Ctx) == Builtin::BI__builtin_constant_p) {
2628d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        Expr::EvalResult EVResult;
2629d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        if (!E->Evaluate(EVResult, Ctx) || EVResult.HasSideEffects ||
2630d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall            !EVResult.Val.isInt()) {
2631d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall          return ICEDiag(2, E->getLocStart());
2632d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        }
2633d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall        return NoDiag();
2634d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      }
2635d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag CondResult = CheckICE(Exp->getCond(), Ctx);
2636d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag TrueResult = CheckICE(Exp->getTrueExpr(), Ctx);
2637d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    ICEDiag FalseResult = CheckICE(Exp->getFalseExpr(), Ctx);
2638d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 2)
2639d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2640d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 2)
2641d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return TrueResult;
2642d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (FalseResult.Val == 2)
2643d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2644d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (CondResult.Val == 1)
2645d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return CondResult;
2646d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (TrueResult.Val == 0 && FalseResult.Val == 0)
2647d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return NoDiag();
2648d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Rare case where the diagnostics depend on which side is evaluated
2649d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // Note that if we get here, CondResult is 0, and at least one of
2650d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    // TrueResult and FalseResult is non-zero.
2651d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Exp->getCond()->EvaluateAsInt(Ctx) == 0) {
2652d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall      return FalseResult;
2653d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    }
2654d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return TrueResult;
2655d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2656d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::CXXDefaultArgExprClass:
2657d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<CXXDefaultArgExpr>(E)->getExpr(), Ctx);
2658d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  case Expr::ChooseExprClass: {
2659d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return CheckICE(cast<ChooseExpr>(E)->getChosenSubExpr(Ctx), Ctx);
2660d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2661d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2662d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2663d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  // Silence a GCC warning
2664d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return ICEDiag(2, E->getLocStart());
2665d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2666d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall
2667d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCallbool Expr::isIntegerConstantExpr(llvm::APSInt &Result, ASTContext &Ctx,
2668d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall                                 SourceLocation *Loc, bool isEvaluated) const {
2669d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  ICEDiag d = CheckICE(this, Ctx);
2670d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (d.Val != 0) {
2671d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    if (Loc) *Loc = d.Loc;
2672d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    return false;
2673d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  }
2674d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  EvalResult EvalResult;
2675d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  if (!Evaluate(EvalResult, Ctx))
2676d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall    llvm_unreachable("ICE cannot be evaluated!");
2677d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(!EvalResult.HasSideEffects && "ICE with side effects!");
2678d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  assert(EvalResult.Val.isInt() && "ICE that isn't integer!");
2679d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  Result = EvalResult.Val.getInt();
2680d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall  return true;
2681d905f5ad540c415d1a21b4f8b7bd715bfb7bb920John McCall}
2682