Throw if MVM_is_prime called with too many rounds

Valid range is 0..PRIME_SIZE (defined in libtommath).
MoarVM · Jul 2, 2019 · 0e7599e · 0e7599e
1 parent d6f9e02
commit 0e7599e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/math/bigintops.c b/src/math/bigintops.c
@@ -1203,8 +1203,10 @@ MVMint64 MVM_bigint_is_prime(MVMThreadContext *tc, MVMObject *a, MVMint64 b) {
             return 0;
         }
         else {
-            int result;
-            mp_prime_is_prime(ia, b, &result);
+            int result, err;
+            err = mp_prime_is_prime(ia, b, &result);
+            if (err != MP_OKAY)
+                MVM_exception_throw_adhoc(tc, "Invalid number of rounds (%ld), valid range is 0..%d\n", b, PRIME_SIZE);
             return result;
         }
     } else {