X-Git-Url: https://git.distorted.org.uk/~mdw/catacomb/blobdiff_plain/01f3ef938fe2a53bb09e59b7baa81c06b3401002..fbfcb6c0ffbad26c7eae30a7590e9d7897331a31:/pub/bbs-gen.c

diff --git a/pub/bbs-gen.c b/pub/bbs-gen.c
index d2b4b6e5..3239674f 100644
--- a/pub/bbs-gen.c
+++ b/pub/bbs-gen.c
@@ -60,7 +60,6 @@
 int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
 	    pgen_proc *event, void *ectx)
 {
-  rabin rb;
   pfilt jp;
   pgen_jumpctx j;
   pgen_gcdstepctx g;
@@ -73,7 +72,7 @@ int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
     goto fail_x;
   j.j = &jp;
   bp->p = pgen("p", MP_NEWSEC, x, event, ectx, n, pgen_jump, &j,
-	       rabin_iters(nb), pgen_test, &rb);
+	       PGEN_BAILLIEPSWNTESTS, pgen_bailliepswtest, 0);
   pfilt_destroy(&jp);
   if (!bp->p) goto fail_p;
 
@@ -89,7 +88,7 @@ int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
   g.g = MP_NEW;
   g.max = MP_ONE;
   bp->q = pgen("q", MP_NEWSEC, x, event, ectx, n, pgen_gcdstep, &g,
-	       rabin_iters(nb), pgen_test, &rb);
+	       PGEN_BAILLIEPSWNTESTS, pgen_bailliepswtest, 0);
   pfilt_destroy(&g.jp);
   mp_drop(g.r);
   mp_drop(g.g);
@@ -98,11 +97,14 @@ int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
   /* --- Compute @n@ --- */
 
   bp->n = mp_mul(MP_NEW, bp->p, bp->q);
+  if (mp_bits(bp->n) != nbits) goto fail_n;
   mp_drop(x);
   return (PGEN_DONE);
 
   /* --- Tidy up if things went wrong --- */
 
+fail_n:
+  mp_drop(bp->n);
 fail_q:
   mp_drop(bp->p);
 fail_p: