vdf fix (#142)

* Division by 0 catching. * Use mpz_sgn. Co-authored-by: Florin Chirica <[email protected]>
Chia-Network · Jun 27, 2023 · c75ec14 · c75ec14
1 parent 3097201
commit c75ec14
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/bqfc.c b/src/bqfc.c
@@ -68,6 +68,10 @@ int bqfc_decompr(mpz_t out_a, mpz_t out_b, const mpz_t D, const struct qfb_c *c)
         mpz_set(t, c->t);
     }
 
+    if (mpz_sgn(c->a) == 0) {
+        ret = -1;
+        goto out;
+    }
     mpz_gcdext(tmp, t_inv, NULL, t, c->a);
     if (mpz_cmp_ui(tmp, 1)) {
         ret = -1;

diff --git a/src/proof_common.h b/src/proof_common.h
@@ -65,6 +65,9 @@ form DeserializeForm(const integer &D, const uint8_t *bytes, size_t size)
 
 integer FastPow(uint64_t a, uint64_t b, integer& c) {
     integer res, a1 = integer(a);
+    if (mpz_sgn(c.impl) == 0) {
+        throw std::runtime_error("Division by 0");
+    }
     mpz_powm_ui(res.impl, a1.impl, b, c.impl);
     return res;
 }