Use `div_flo` (copy of `mrb_num_div_flo`) for float division.

1 files changed, 17 insertions, 3 deletions

diff --git a/mrbgems/mruby-complex/src/complex.c b/mrbgems/mruby-complex/src/complex.c
index c0af1904b..f0269f165 100644
--- a/mrbgems/mruby-complex/src/complex.c
+++ b/mrbgems/mruby-complex/src/complex.c
@@ -193,6 +193,20 @@ complex_mul(mrb_state *mrb, mrb_value x)
   }
 }
 
+mrb_float
+div_flo(mrb_float x, mrb_float y)
+{
+  if (y != 0.0) {
+    return x / y;
+  }
+  else if (x == 0.0) {
+    return NAN;
+  }
+  else {
+    return x * (signbit(y) ? -1.0 : 1.0) * INFINITY;
+  }
+}
+
 /* Arithmetic on (significand, exponent) pairs avoids premature overflow in
    complex division */
 struct float_pair {
@@ -229,7 +243,7 @@ static void
 div_pair(struct float_pair *q, struct float_pair const *a,
          struct float_pair const *b)
 {
-  q->s = a->s / b->s;
+  q->s = div_flo(a->s, b->s);
   q->x = a->x - b->x;
 }
 
@@ -242,7 +256,7 @@ complex_div(mrb_state *mrb, mrb_value self)
   a = complex_ptr(mrb, self);
   if (mrb_type(rhs) != MRB_TT_COMPLEX) {
     mrb_float f = mrb_to_flo(mrb, rhs);
-    return complex_new(mrb, a->real / f, a->imaginary / f);
+    return complex_new(mrb, div_flo(a->real, f), div_flo(a->imaginary, f));
   }
 
   struct float_pair ar, ai, br, bi;
@@ -309,7 +323,7 @@ int_div(mrb_state *mrb, mrb_value x)
     x = complex_new(mrb, (mrb_float)a, 0);
     return mrb_funcall_id(mrb, x, MRB_OPSYM(div), 1, y);
   default:
-    return mrb_float_value(mrb, mrb_num_div_flo(mrb, (mrb_float)a, mrb_to_flo(mrb, y)));
+    return mrb_float_value(mrb, div_flo(mrb, (mrb_float)a, mrb_to_flo(mrb, y)));
   }
 }