diff --git a/include/llvm/Constant.h b/include/llvm/Constant.h index 15722b62db0..24fb99036ec 100644 --- a/include/llvm/Constant.h +++ b/include/llvm/Constant.h @@ -54,6 +54,10 @@ public: virtual bool isNullValue() const = 0; virtual void print(std::ostream &O) const; + + /// canTrap - Return true if evaluation of this constant could trap. This is + /// true for things like constant expressions that could divide by zero. + bool canTrap() const; // Specialize get/setOperand for Constant's as their operands are always // constants as well. diff --git a/lib/VMCore/Constants.cpp b/lib/VMCore/Constants.cpp index d187125233c..cf8f79e96d2 100644 --- a/lib/VMCore/Constants.cpp +++ b/lib/VMCore/Constants.cpp @@ -58,6 +58,33 @@ void Constant::destroyConstantImpl() { delete this; } +/// canTrap - Return true if evaluation of this constant could trap. This is +/// true for things like constant expressions that could divide by zero. +bool Constant::canTrap() const { + assert(getType()->isFirstClassType() && "Cannot evaluate aggregate vals!"); + // The only thing that could possibly trap are constant exprs. + const ConstantExpr *CE = dyn_cast(this); + if (!CE) return false; + + // ConstantExpr traps if any operands can trap. + for (unsigned i = 0, e = getNumOperands(); i != e; ++i) + if (getOperand(i)->canTrap()) + return true; + + // Otherwise, only specific operations can trap. + switch (CE->getOpcode()) { + default: + return false; + case Instruction::Div: + case Instruction::Rem: + // Div and rem can trap if the RHS is not known to be non-zero. + if (!isa(getOperand(1)) || getOperand(1)->isNullValue()) + return true; + return false; + } +} + + // Static constructor to create a '0' constant of arbitrary type... Constant *Constant::getNullValue(const Type *Ty) { switch (Ty->getTypeID()) {