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

diff --git a/pub/bbs-gen.c b/pub/bbs-gen.c
index d2b4b6e5..fcba8288 100644
--- a/pub/bbs-gen.c
+++ b/pub/bbs-gen.c
@@ -65,7 +65,7 @@ int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
   pgen_jumpctx j;
   pgen_gcdstepctx g;
   unsigned nb = nbits/2;
-  mp *x = MP_NEWSEC;
+  mp *x = MP_NEWSEC, *t = MP_NEW;
 
   /* --- Generate @p@ --- */
 
@@ -88,21 +88,28 @@ int bbs_gen(bbs_priv *bp, unsigned nbits, grand *r, unsigned n,
   g.r = mp_lsr(MP_NEW, bp->p, 1);
   g.g = MP_NEW;
   g.max = MP_ONE;
+  t = mp_lsl(t, MP_ONE, nbits - 1);
+  mp_div(&t, 0, t, bp->p);
+  if (MP_CMP(x, <, t)) x = mp_leastcongruent(x, t, x, g.jp.m);
   bp->q = pgen("q", MP_NEWSEC, x, event, ectx, n, pgen_gcdstep, &g,
 	       rabin_iters(nb), pgen_test, &rb);
   pfilt_destroy(&g.jp);
   mp_drop(g.r);
   mp_drop(g.g);
+  mp_drop(t);
   if (!bp->q) goto fail_q;
 
   /* --- 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: