Add divide-by-zero check to execute

pc2 · Jul 6, 2024 · be630e4 · be630e4
1 parent 411956c
commit be630e4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/instantiation/execute.rs b/src/instantiation/execute.rs
@@ -253,6 +253,17 @@ impl<'fl, 'l> InstantiationContext<'fl, 'l> {
             &WireSource::BinaryOp{op, left, right} => {
                 let left_val = self.generation_state.get_generation_value(left)?;
                 let right_val = self.generation_state.get_generation_value(right)?;
+
+                match op {
+                    BinaryOperator::Divide | BinaryOperator::Modulo => {
+                        use num::Zero;
+                        if right_val.value.unwrap_integer().is_zero() {
+                            return Err((wire_inst.span, format!("Divide or Modulo by zero: {} / 0", left_val.unwrap_integer())));
+                        }
+                    }
+                    _ => {}
+                }
+
                 compute_binary_op(left_val, op, right_val)
             }
             WireSource::Constant(value) => TypedValue::from_value(value.clone())