Division by zero error

SingularityT3 · Apr 13, 2022 · 257aae6 · 257aae6
1 parent fac0245
commit 257aae6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/nodes/eval/arithmetic.cpp b/src/nodes/eval/arithmetic.cpp
@@ -101,12 +101,21 @@ std::unique_ptr<NodeResult> ArithmeticOperationNode::evaluate(PSC::Context &ctx)
             resNum = leftNum * rightNum;
             break;
         case TokenType::SLASH:
+            if ((rightNum.real && ((PSC::Real&) rightNum).value == 0)
+                || (!rightNum.real && ((PSC::Integer&) rightNum).value == 0)
+            ) throw PSC::RuntimeError(token, ctx, "Division by 0");
             resNum = leftNum / rightNum;
             break;
         case TokenType::MOD:
+            if ((rightNum.real && ((PSC::Real&) rightNum).value == 0)
+                || (!rightNum.real && ((PSC::Integer&) rightNum).value == 0)
+            ) throw PSC::RuntimeError(token, ctx, "Modulus by 0");
             resNum = leftNum % rightNum;
             break;
         case TokenType::DIV:
+        if ((rightNum.real && ((PSC::Real&) rightNum).value == 0)
+                || (!rightNum.real && ((PSC::Integer&) rightNum).value == 0)
+            ) throw PSC::RuntimeError(token, ctx, "Division by 0");
             resNum = leftNum | rightNum;
             break;
         default: