From e5635f4ab3d0fd00dd3951865fea82003205dae1 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Thu, 1 Oct 2015 22:48:48 +0100
Subject: py: Catch all cases of integer (big and small) division by zero.

---
 py/runtime.c | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'py/runtime.c')

diff --git a/py/runtime.c b/py/runtime.c
index 69ac7549b..98cde83e5 100644
--- a/py/runtime.c
+++ b/py/runtime.c
@@ -389,6 +389,9 @@ mp_obj_t mp_binary_op(mp_uint_t op, mp_obj_t lhs, mp_obj_t rhs) {
 
                 case MP_BINARY_OP_MODULO:
                 case MP_BINARY_OP_INPLACE_MODULO: {
+                    if (rhs_val == 0) {
+                        goto zero_division;
+                    }
                     lhs_val = mp_small_int_modulo(lhs_val, rhs_val);
                     break;
                 }
-- 
cgit v1.2.3