/* -*-c-*-
*
- * $Id: pgen.c,v 1.10 2004/04/08 01:36:15 mdw Exp $
+ * $Id$
*
* Prime generation glue
*
* (c) 1999 Straylight/Edgeware
*/
-/*----- Licensing notice --------------------------------------------------*
+/*----- Licensing notice --------------------------------------------------*
*
* This file is part of Catacomb.
*
* it under the terms of the GNU Library General Public License as
* published by the Free Software Foundation; either version 2 of the
* License, or (at your option) any later version.
- *
+ *
* Catacomb is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Library General Public License for more details.
- *
+ *
* You should have received a copy of the GNU Library General Public
* License along with Catacomb; if not, write to the Free
* Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
int pgen_filter(int rq, pgen_event *ev, void *p)
{
pgen_filterctx *f = p;
- int rc = PGEN_ABORT;
+ int rc = PGEN_FAIL;
switch (rq) {
case PGEN_BEGIN:
break;
case PGEN_TRY:
mp_drop(ev->m);
- if (!((f->step | f->f.m->v[0]) & 1))
- rc = pfilt_step(&f->f, 1);
- else
- rc = pfilt_step(&f->f, f->step);
break;
case PGEN_DONE:
pfilt_destroy(&f->f);
return (PGEN_DONE);
+ default:
+ rc = PGEN_ABORT;
+ break;
}
-
+
+ if (rc == PGEN_FAIL && !((f->step | f->f.m->v[0]) & 1))
+ rc = pfilt_step(&f->f, 1);
while (rc == PGEN_FAIL)
rc = pfilt_step(&f->f, f->step);
ev->m = MP_COPY(f->f.m);
pfilt_destroy(&f->f);
return (PGEN_DONE);
}
-
+
while (rc == PGEN_FAIL)
rc = pfilt_jump(&f->f, f->j);
ev->m = MP_COPY(f->f.m);
int rq, rc;
pgen_proc *proc;
void *ctx;
+ int p;
+
+ enum { P_STEP, P_TEST };
/* --- Set up the initial event block --- */
/* --- Tell the event handler we're under way --- */
- if (event && event(PGEN_BEGIN, &ev, ectx) == PGEN_ABORT)
+ if (event && event(PGEN_BEGIN, &ev, ectx) == PGEN_ABORT) {
+ ev.r->ops->destroy(ev.r);
return (0);
+ }
/* --- Set up for the initial call --- */
- proc = step; ctx = sctx; rq = PGEN_BEGIN;
+ proc = step; ctx = sctx; p = P_STEP; rq = PGEN_BEGIN;
/* --- Enter the great maelstrom of state transitions --- */
for (;;) {
unsigned act = 0;
- enum {
- A_STEP = 1u,
- A_TEST = 2u,
- A_EVENT = 4u,
- A_ENDTEST = 8u,
- A_ENDSTEP = 16u,
- A_DONE = 32u
- };
+#define A_STEP 1u
+#define A_TEST 2u
+#define A_EVENT 4u
+#define A_ENDTEST 8u
+#define A_ENDSTEP 16u
+#define A_DONE 32u
/* --- Call the procedure and decide what to do next --- */
rc = proc(rq, &ev, ctx);
switch (rc) {
case PGEN_TRY:
- if (proc == test)
+ if (p == P_TEST)
rq = PGEN_TRY;
else {
act |= A_EVENT;
- proc = test; ctx = tctx;
+ proc = test; ctx = tctx; p = P_TEST;
rq = PGEN_BEGIN;
}
break;
case PGEN_PASS:
act |= A_TEST | A_EVENT;
- if (proc == test)
+ if (p == P_TEST)
rq = PGEN_TRY;
else {
- proc = test; ctx = tctx;
+ proc = test; ctx = tctx; p = P_TEST;
rq = PGEN_BEGIN;
}
break;
case PGEN_FAIL:
act |= A_STEP;
- if (proc == test) {
+ if (p == P_TEST) {
act |= A_ENDTEST | A_EVENT;
- proc = step; ctx = sctx;
+ proc = step; ctx = sctx; p = P_STEP;
}
rq = PGEN_TRY;
break;
case PGEN_DONE:
act |= A_EVENT | A_DONE | A_ENDSTEP;
- if (proc == test)
+ if (p == P_TEST)
act |= A_ENDTEST;
break;
case PGEN_ABORT:
act |= A_EVENT | A_DONE;
- if (proc == test || rq == PGEN_TRY)
+ if (p == P_TEST || rq == PGEN_TRY)
act |= A_ENDSTEP;
- if (proc == test && rq == PGEN_BEGIN)
+ if (p == P_TEST && rq != PGEN_BEGIN)
act |= A_ENDTEST;
break;
default:
rc = PGEN_ABORT;
if (!(act & A_DONE)) {
act |= A_ENDSTEP | A_DONE;
- if (proc == test)
+ if (p == P_TEST)
act |= A_ENDTEST;
}
- }
+ }
/* --- Close down tester and stepper functions --- */
int pgen_primep(mp *p, grand *gr)
{
- int i = rabin_iters(mp_bits(p));
+ int i;
rabin r;
mp *x = MP_NEW;
- if (MP_ISNEG(p)) return (0);
+ if (MP_NEGP(p)) return (0);
switch (pfilt_smallfactor(p)) {
case PGEN_DONE: return (1);
case PGEN_FAIL: return (0);
}
rabin_create(&r, p);
- while (i) {
+ for (i = 32; i; i--) {
x = mprand_range(x, p, gr, 0);
if (rabin_rtest(&r, x) == PGEN_FAIL)
break;
- i--;
}
MP_DROP(x);
rabin_destroy(&r);
#include <mLib/testrig.h>
+static int t_primep(dstr *v)
+{
+ mp *m = *(mp **)v[0].buf;
+ int e = *(int *)v[1].buf;
+ int r;
+ grand *rng;
+ int ok = 1;
+
+ rng = fibrand_create(0);
+ r = pgen_primep(m, rng);
+ GR_DESTROY(rng);
+ if (e != r) {
+ fputs("\n*** primep failed", stderr);
+ fputs("\nm = ", stderr); mp_writefile(m, stderr, 10);
+ fprintf(stderr, "\nexpected %d", e);
+ fprintf(stderr, "\nreported %d", r);
+ fputc('\n', stderr);
+ ok = 0;
+ }
+
+ mp_drop(m);
+ assert(mparena_count(MPARENA_GLOBAL) == 0);
+ return (ok);
+}
+
static int verify(dstr *v)
{
mp *m = *(mp **)v[0].buf;
static test_chunk tests[] = {
{ "pgen", verify, { &type_mp, &type_mp, 0 } },
+ { "primep", t_primep, { &type_mp, &type_int, 0 } },
{ 0, 0, { 0 } }
};