From 8d27a027051e93e6f63aa9b969fed1dbfcf2aa73 Mon Sep 17 00:00:00 2001
From: Andrew Kelley <superjoe30@gmail.com>
Date: Sun, 8 Jan 2017 22:35:31 -0500
Subject: pass division by zero test

---
 src/ir.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/ir.cpp')

diff --git a/src/ir.cpp b/src/ir.cpp
index 4923591c21..b4475499d3 100644
--- a/src/ir.cpp
+++ b/src/ir.cpp
@@ -6915,6 +6915,25 @@ static int ir_eval_bignum(ConstExprValue *op1_val, ConstExprValue *op2_val,
         ConstExprValue *out_val, bool (*bignum_fn)(BigNum *, BigNum *, BigNum *),
         TypeTableEntry *type, bool wrapping_op)
 {
+    bool is_int = false;
+    bool is_float = false;
+    if (bignum_fn == bignum_div || bignum_fn == bignum_mod) {
+        if (type->id == TypeTableEntryIdInt ||
+            type->id == TypeTableEntryIdNumLitInt)
+        {
+            is_int = true;
+        } else if (type->id == TypeTableEntryIdFloat ||
+                   type->id == TypeTableEntryIdNumLitFloat)
+        {
+            is_float = true;
+        }
+        if ((is_int && op2_val->data.x_bignum.data.x_uint == 0) ||
+            (is_float && op2_val->data.x_bignum.data.x_float == 0.0))
+        {
+            return ErrorDivByZero;
+        }
+    }
+
     bool overflow = bignum_fn(&out_val->data.x_bignum, &op1_val->data.x_bignum, &op2_val->data.x_bignum);
     if (overflow) {
         return ErrorOverflow;
-- 
cgit v1.2.3