~mdw / catacomb-python / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
catacomb-python.h (GEN): Remove `const' from input table type carefully.
[catacomb-python] / pgen.c
CommitLineData
d7ab1bab 1/* -*-c-*-
2 *
d7ab1bab 3 * Prime number generation
4 *
5 * (c) 2005 Straylight/Edgeware
6 */
7
b2687a0a 8/*----- Licensing notice --------------------------------------------------*
d7ab1bab 9 *
10 * This file is part of the Python interface to Catacomb.
11 *
12 * Catacomb/Python is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License as published by
14 * the Free Software Foundation; either version 2 of the License, or
15 * (at your option) any later version.
b2687a0a 16 *
d7ab1bab 17 * Catacomb/Python is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU General Public License for more details.
b2687a0a 21 *
d7ab1bab 22 * You should have received a copy of the GNU General Public License
23 * along with Catacomb/Python; if not, write to the Free Software Foundation,
24 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25 */
26
27/*----- Header files ------------------------------------------------------*/
28
29#include "catacomb-python.h"
30
31/*----- Filters -----------------------------------------------------------*/
32
33PyTypeObject *pfilt_pytype;
34
35static PyObject *pfilt_pywrap(pfilt *f)
36{
37 pfilt_pyobj *o = 0;
38 o = PyObject_New(pfilt_pyobj, pfilt_pytype);
39 o->f = *f;
40 o->st = pfilt_step(f, 0);
b2687a0a 41 return ((PyObject *)o);
d7ab1bab 42}
43
44static PyObject *pfilt_pymake(PyTypeObject *ty, PyObject *xobj)
45{
46 mp *x = 0;
47 pfilt_pyobj *o = 0;
48
49 if (PFILT_PYCHECK(xobj)) RETURN_OBJ(xobj);
50 if ((x = getmp(xobj)) == 0) goto end;
51 o = (pfilt_pyobj *)ty->tp_alloc(ty, 0);
52 o->st = pfilt_create(&o->f, x);
53end:
54 mp_drop(x);
55 return ((PyObject *)o);
56}
57
58static PyObject *pfilt_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
59{
827f89d7 60 static const char *const kwlist[] = { "x", 0 };
d7ab1bab 61 PyObject *xobj;
62
827f89d7 63 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O:new", KWLIST, &xobj))
d7ab1bab 64 return (0);
65 return (pfilt_pymake(ty, xobj));
66}
67
68static void pfilt_pydealloc(PyObject *me)
3aa33042 69 { pfilt_destroy(PFILT_F(me)); FREEOBJ(me); }
d7ab1bab 70
71static PyObject *pfmeth_step(PyObject *me, PyObject *arg)
72{
73 mpw x;
74
75 if (!PyArg_ParseTuple(arg, "O&:step", convmpw, &x)) return (0);
76 PFILT_ST(me) = pfilt_step(PFILT_F(me), x);
77 RETURN_ME;
78}
79
80static PyObject *pfmeth_muladd(PyObject *me, PyObject *arg)
81{
82 mpw m, a;
83 pfilt_pyobj *o = 0;
84
85 if (!PyArg_ParseTuple(arg, "O&O&:muladd", convmpw, &m, convmpw, &a))
86 return (0);
87 o = PyObject_New(pfilt_pyobj, pfilt_pytype);
88 o->st = pfilt_muladd(&o->f, PFILT_F(me), m, a);
89 return ((PyObject *)o);
90}
91
92static CONVFUNC(pfilt, pfilt *, PFILT_F)
93
94static PyObject *pfmeth_jump(PyObject *me, PyObject *arg)
95{
96 pfilt *f;
97
98 if (!PyArg_ParseTuple(arg, "O&:jump", convpfilt, &f)) return (0);
99 PFILT_ST(me) = pfilt_jump(PFILT_F(me), f);
100 RETURN_ME;
101}
102
103static PyObject *meth__PrimeFilter_smallfactor(PyObject *me, PyObject *arg)
104{
105 mp *x = 0;
106 PyObject *rc = 0;
107
108 if (!PyArg_ParseTuple(arg, "OO&:smallfactor", &me, convmp, &x)) goto end;
109 rc = PyInt_FromLong(pfilt_smallfactor(x));
110end:
111 mp_drop(x);
112 return (rc);
113}
114
115static int pfilt_pynonzerop(PyObject *me)
116 { return (PFILT_ST(me) != PGEN_FAIL); }
117
118static PyObject *pfilt_pyint(PyObject *me)
119{
120 long l;
121 PyObject *rc = 0;
122
bc243788
MW		 123 if (!mp_tolong_checked(PFILT_F(me)->m, &l, 0)) rc = PyInt_FromLong(l);
124 else rc = mp_topylong(PFILT_F(me)->m);
d7ab1bab 125 return (rc);
126}
127
128static PyObject *pfilt_pylong(PyObject *me)
f368b46e 129 { return (mp_topylong(PFILT_F(me)->m)); }
d7ab1bab 130
131static PyObject *pfget_x(PyObject *me, void *hunoz)
132 { return (mp_pywrap(MP_COPY(PFILT_F(me)->m))); }
133
134static PyObject *pfget_status(PyObject *me, void *hunoz)
135 { return (PyInt_FromLong(PFILT_ST(me))); }
136
137static PyGetSetDef pfilt_pygetset[] = {
138#define GETSETNAME(op, name) pf##op##_##name
139 GET (x, "F.x -> current position of filter")
140 GET (status, "F.status -> primality status of filter")
141#undef GETSETNAME
142 { 0 }
143};
144
145static PyMethodDef pfilt_pymethods[] = {
146#define METHNAME(name) pfmeth_##name
147 METH (step, "F.step(N)")
148 METH (muladd, "F.muladd(M, A)")
149 METH (jump, "F.jump(FF)")
150#undef METHNAME
151 { 0 }
152};
153
154static PyNumberMethods pfilt_pynumber = {
155 0, /* @nb_add@ */
156 0, /* @nb_subtract@ */
157 0, /* @nb_multiply@ */
158 0, /* @nb_divide@ */
159 0, /* @nb_remainder@ */
160 0, /* @nb_divmod@ */
161 0, /* @nb_power@ */
162 0, /* @nb_negative@ */
163 0, /* @nb_positive@ */
164 0, /* @nb_absolute@ */
165 pfilt_pynonzerop, /* @nb_nonzero@ */
166 0, /* @nb_invert@ */
167 0, /* @nb_lshift@ */
168 0, /* @nb_rshift@ */
169 0, /* @nb_and@ */
170 0, /* @nb_xor@ */
171 0, /* @nb_or@ */
172 0, /* @nb_coerce@ */
173 pfilt_pyint, /* @nb_int@ */
174 pfilt_pylong, /* @nb_long@ */
175 0, /* @nb_float@ */
176 0, /* @nb_oct@ */
177 0, /* @nb_hex@ */
178
179 0, /* @nb_inplace_add@ */
180 0, /* @nb_inplace_subtract@ */
181 0, /* @nb_inplace_multiply@ */
182 0, /* @nb_inplace_divide@ */
183 0, /* @nb_inplace_remainder@ */
184 0, /* @nb_inplace_power@ */
185 0, /* @nb_inplace_lshift@ */
186 0, /* @nb_inplace_rshift@ */
187 0, /* @nb_inplace_and@ */
188 0, /* @nb_inplace_xor@ */
189 0, /* @nb_inplace_or@ */
190
191 0, /* @nb_floor_divide@ */
192 0, /* @nb_true_divide@ */
193 0, /* @nb_inplace_floor_divide@ */
194 0, /* @nb_inplace_true_divide@ */
195};
196
197static PyTypeObject pfilt_pytype_skel = {
6d4db0bf 198 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 199 "PrimeFilter", /* @tp_name@ */
d7ab1bab 200 sizeof(pfilt_pyobj), /* @tp_basicsize@ */
201 0, /* @tp_itemsize@ */
202
203 pfilt_pydealloc, /* @tp_dealloc@ */
204 0, /* @tp_print@ */
205 0, /* @tp_getattr@ */
206 0, /* @tp_setattr@ */
207 0, /* @tp_compare@ */
208 0, /* @tp_repr@ */
209 &pfilt_pynumber, /* @tp_as_number@ */
210 0, /* @tp_as_sequence@ */
211 0, /* @tp_as_mapping@ */
212 0, /* @tp_hash@ */
213 0, /* @tp_call@ */
214 0, /* @tp_str@ */
215 0, /* @tp_getattro@ */
216 0, /* @tp_setattro@ */
217 0, /* @tp_as_buffer@ */
218 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
219 Py_TPFLAGS_BASETYPE,
220
221 /* @tp_doc@ */
d96c882e 222 "PrimeFilter(X): small-primes filter.",
d7ab1bab 223
224 0, /* @tp_traverse@ */
225 0, /* @tp_clear@ */
226 0, /* @tp_richcompare@ */
227 0, /* @tp_weaklistoffset@ */
228 0, /* @tp_iter@ */
963a6148 229 0, /* @tp_iternext@ */
d7ab1bab 230 pfilt_pymethods, /* @tp_methods@ */
231 0, /* @tp_members@ */
232 pfilt_pygetset, /* @tp_getset@ */
233 0, /* @tp_base@ */
234 0, /* @tp_dict@ */
235 0, /* @tp_descr_get@ */
236 0, /* @tp_descr_set@ */
237 0, /* @tp_dictoffset@ */
238 0, /* @tp_init@ */
239 PyType_GenericAlloc, /* @tp_alloc@ */
240 pfilt_pynew, /* @tp_new@ */
3aa33042 241 0, /* @tp_free@ */
d7ab1bab 242 0 /* @tp_is_gc@ */
243};
244
245/*----- Rabin-Miller testing ----------------------------------------------*/
246
247typedef struct rabin_pyobj {
248 PyObject_HEAD
249 rabin r;
250} rabin_pyobj;
251
252static PyTypeObject *rabin_pytype;
253#define RABIN_R(o) (&((rabin_pyobj *)(o))->r)
254
255static PyObject *rabin_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
256{
257 mp *x = 0;
258 rabin_pyobj *o = 0;
827f89d7 259 static const char *const kwlist[] = { "x", 0 };
d7ab1bab 260
827f89d7 261 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&:new", KWLIST, convmp, &x))
d7ab1bab 262 goto end;
263 if (!MP_POSP(x) || MP_EVENP(x)) VALERR("must be positive and odd");
264 o = (rabin_pyobj *)ty->tp_alloc(ty, 0);
265 rabin_create(&o->r, x);
266end:
267 return ((PyObject *)o);
268}
269
270static void rabin_pydealloc(PyObject *me)
271{
272 rabin_destroy(RABIN_R(me));
3aa33042 273 FREEOBJ(me);
d7ab1bab 274}
275
276static PyObject *rmeth_test(PyObject *me, PyObject *arg)
277{
278 mp *w = 0;
279 PyObject *rc = 0;
280
281 if (!PyArg_ParseTuple(arg, "O&:test", convmp, &w)) goto end;
282 rc = PyInt_FromLong(rabin_test(RABIN_R(me), w));
283end:
284 mp_drop(w);
285 return (rc);
286}
287
288static PyObject *rmeth_rtest(PyObject *me, PyObject *arg)
289{
290 mp *w = 0;
291 PyObject *rc = 0;
292
293 if (!PyArg_ParseTuple(arg, "O&:rtest", convmp, &w)) goto end;
294 rc = PyInt_FromLong(rabin_rtest(RABIN_R(me), w));
295end:
296 mp_drop(w);
297 return (rc);
298}
299
300static PyObject *rget_niters(PyObject *me, void *hunoz)
301 { return (PyInt_FromLong(rabin_iters(mp_bits(RABIN_R(me)->mm.m)))); }
302
303static PyObject *rget_x(PyObject *me, void *hunoz)
304 { return (mp_pywrap(MP_COPY(RABIN_R(me)->mm.m))); }
305
306static PyObject *meth__RabinMiller_iters(PyObject *me, PyObject *arg)
307{
308 unsigned n;
309
310 if (!PyArg_ParseTuple(arg, "OO&:iters", &me, convuint, &n)) return (0);
311 return (PyInt_FromLong(rabin_iters(n)));
312}
313
314static PyGetSetDef rabin_pygetset[] = {
315#define GETSETNAME(op, name) r##op##_##name
316 GET (x, "R.x -> number under test")
317 GET (niters, "R.niters -> suggested number of tests")
318#undef GETSETNAME
319 { 0 }
320};
321
322static PyMethodDef rabin_pymethods[] = {
323#define METHNAME(name) rmeth_##name
324 METH (test, "R.test(W) -> PGST")
325 METH (rtest, "R.rtest(W) -> PGST")
326#undef METHNAME
327 { 0 }
328};
329
330static PyTypeObject rabin_pytype_skel = {
6d4db0bf 331 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 332 "RabinMiller", /* @tp_name@ */
d7ab1bab 333 sizeof(rabin_pyobj), /* @tp_basicsize@ */
334 0, /* @tp_itemsize@ */
335
336 rabin_pydealloc, /* @tp_dealloc@ */
337 0, /* @tp_print@ */
338 0, /* @tp_getattr@ */
339 0, /* @tp_setattr@ */
340 0, /* @tp_compare@ */
341 0, /* @tp_repr@ */
342 0, /* @tp_as_number@ */
343 0, /* @tp_as_sequence@ */
344 0, /* @tp_as_mapping@ */
345 0, /* @tp_hash@ */
346 0, /* @tp_call@ */
347 0, /* @tp_str@ */
348 0, /* @tp_getattro@ */
349 0, /* @tp_setattro@ */
350 0, /* @tp_as_buffer@ */
351 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
352 Py_TPFLAGS_BASETYPE,
353
354 /* @tp_doc@ */
d96c882e 355 "RabinMiller(X): Rabin-Miller strong primality test.",
d7ab1bab 356
357 0, /* @tp_traverse@ */
358 0, /* @tp_clear@ */
359 0, /* @tp_richcompare@ */
360 0, /* @tp_weaklistoffset@ */
361 0, /* @tp_iter@ */
963a6148 362 0, /* @tp_iternext@ */
d7ab1bab 363 rabin_pymethods, /* @tp_methods@ */
364 0, /* @tp_members@ */
365 rabin_pygetset, /* @tp_getset@ */
366 0, /* @tp_base@ */
367 0, /* @tp_dict@ */
368 0, /* @tp_descr_get@ */
369 0, /* @tp_descr_set@ */
370 0, /* @tp_dictoffset@ */
371 0, /* @tp_init@ */
372 PyType_GenericAlloc, /* @tp_alloc@ */
373 rabin_pynew, /* @tp_new@ */
3aa33042 374 0, /* @tp_free@ */
d7ab1bab 375 0 /* @tp_is_gc@ */
376};
377
378/*----- Events ------------------------------------------------------------*/
379
380typedef struct pgevent_pyobj {
381 PyObject_HEAD
d65d80d7 382 PyObject *r;
d7ab1bab 383 pgen_event *ev;
384} pgevent_pyobj;
385
386static PyTypeObject *pgevent_pytype;
387#define PGEVENT_EV(o) (((pgevent_pyobj *)(o))->ev)
388
389static PyObject *pgevent_pywrap(pgen_event *ev)
390{
391 pgevent_pyobj *o = PyObject_New(pgevent_pyobj, pgevent_pytype);
d65d80d7 392 o->ev = ev; o->r = 0;
d7ab1bab 393 return ((PyObject *)o);
394}
395
396static CONVFUNC(pgevent, pgen_event *, PGEVENT_EV)
397
d65d80d7
MW		 398static void pgevent_kill(PyObject *me)
399{
400 pgevent_pyobj *ev = (pgevent_pyobj *)me;
401
402 ev->ev = 0;
403 if (ev->r) GRAND_R(ev->r) = 0;
404}
405
406static void pgevent_pydealloc(PyObject *me)
407{
408 pgevent_pyobj *ev = (pgevent_pyobj *)me;
19a4c416 409 Py_XDECREF(ev->r); FREEOBJ(me);
d65d80d7 410}
d7ab1bab 411
412#define PGEVENT_CHECK(me) do { \
413 if (!PGEVENT_EV(me)) { \
d65d80d7 414 PyErr_SetString(PyExc_ValueError, "event object is no longer valid"); \
d7ab1bab 415 return (0); \
416 } \
417} while (0)
418
419static PyObject *peget_name(PyObject *me, void *hunoz)
420 { PGEVENT_CHECK(me); return (PyString_FromString(PGEVENT_EV(me)->name)); }
421
422static PyObject *peget_x(PyObject *me, void *hunoz)
423 { PGEVENT_CHECK(me); return (mp_pywrap(MP_COPY(PGEVENT_EV(me)->m))); }
424
425static PyObject *peget_steps(PyObject *me, void *hunoz)
426 { PGEVENT_CHECK(me); return (PyInt_FromLong(PGEVENT_EV(me)->steps)); }
427
428static PyObject *peget_tests(PyObject *me, void *hunoz)
429 { PGEVENT_CHECK(me); return (PyInt_FromLong(PGEVENT_EV(me)->tests)); }
430
431static PyObject *peget_rng(PyObject *me, void *hunoz)
d65d80d7
MW		 432{
433 pgevent_pyobj *ev = (pgevent_pyobj *)me;
434
435 PGEVENT_CHECK(me);
436 if (!ev->r) ev->r = grand_pywrap(ev->ev->r, 0);
437 Py_INCREF(ev->r); return ((PyObject *)ev->r);
438}
d7ab1bab 439
440static int peset_x(PyObject *me, PyObject *xobj, void *hunoz)
441{
442 mp *x = 0;
443 pgen_event *ev = PGEVENT_EV(me);
444 int rc = -1;
03e1fedf 445 if (!xobj) NIERR("__del__");
d7ab1bab 446 PGEVENT_CHECK(me);
447 if ((x = getmp(xobj)) == 0) goto end;
448 mp_drop(ev->m);
449 ev->m = MP_COPY(x);
450 rc = 0;
451end:
452 mp_drop(x);
453 return (rc);
454}
455
456static PyGetSetDef pgevent_pygetset[] = {
457#define GETSETNAME(op, name) pe##op##_##name
458 GET (name, "EV.name -> value being generated")
459 GETSET(x, "EV.x -> value under test")
460 GET (steps, "EV.steps -> number of steps left")
461 GET (tests, "EV.tests -> tests before passing")
462 GET (rng, "EV.rng -> (noncrypto) random number generator")
463#undef GETSETNAME
464 { 0 }
465};
466
467static PyTypeObject pgevent_pytype_skel = {
6d4db0bf 468 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 469 "PrimeGenEvent", /* @tp_name@ */
d7ab1bab 470 sizeof(pgevent_pyobj), /* @tp_basicsize@ */
471 0, /* @tp_itemsize@ */
472
473 pgevent_pydealloc, /* @tp_dealloc@ */
474 0, /* @tp_print@ */
475 0, /* @tp_getattr@ */
476 0, /* @tp_setattr@ */
477 0, /* @tp_compare@ */
478 0, /* @tp_repr@ */
479 0, /* @tp_as_number@ */
480 0, /* @tp_as_sequence@ */
481 0, /* @tp_as_mapping@ */
482 0, /* @tp_hash@ */
483 0, /* @tp_call@ */
484 0, /* @tp_str@ */
485 0, /* @tp_getattro@ */
486 0, /* @tp_setattro@ */
487 0, /* @tp_as_buffer@ */
488 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
489 Py_TPFLAGS_BASETYPE,
490
491 /* @tp_doc@ */
d96c882e 492 "Prime-generation event.",
d7ab1bab 493
494 0, /* @tp_traverse@ */
495 0, /* @tp_clear@ */
496 0, /* @tp_richcompare@ */
497 0, /* @tp_weaklistoffset@ */
498 0, /* @tp_iter@ */
963a6148 499 0, /* @tp_iternext@ */
d7ab1bab 500 0, /* @tp_methods@ */
501 0, /* @tp_members@ */
502 pgevent_pygetset, /* @tp_getset@ */
503 0, /* @tp_base@ */
504 0, /* @tp_dict@ */
505 0, /* @tp_descr_get@ */
506 0, /* @tp_descr_set@ */
507 0, /* @tp_dictoffset@ */
508 0, /* @tp_init@ */
509 PyType_GenericAlloc, /* @tp_alloc@ */
510 abstract_pynew, /* @tp_new@ */
3aa33042 511 0, /* @tp_free@ */
d7ab1bab 512 0 /* @tp_is_gc@ */
513};
514
515/*----- Event handlers ----------------------------------------------------*/
516
517PyTypeObject *pgev_pytype;
518
519typedef struct pgstep_pyobj {
520 PGEV_HEAD
521 pgen_filterctx f;
522} pgstep_pyobj;
523
524static PyTypeObject *pgstep_pytype;
525#define PGSTEP_STEP(o) (((pgstep_pyobj *)(o))->f.step)
526
527typedef struct pgjump_pyobj {
528 PGEV_HEAD
529 PyObject *fobj;
530 pgen_jumpctx j;
531} pgjump_pyobj;
532
533static PyTypeObject *pgjump_pytype;
534#define PGJUMP_FOBJ(o) (((pgjump_pyobj *)(o))->fobj)
535#define PGJUMP_J(o) (&((pgjump_pyobj *)(o))->j)
536
537typedef struct pgtest_pyobj {
538 PGEV_HEAD
539 rabin r;
540} pgtest_pyobj;
541
542static PyTypeObject *pgtest_pytype;
543
544static int pgev_python(int rq, pgen_event *ev, void *p)
545{
930d78e3 546 pypgev *pg = p;
d7ab1bab 547 PyObject *pyev = 0;
548 PyObject *rc = 0;
549 int st = PGEN_ABORT;
550 long l;
827f89d7
MW		 551 static const char *const meth[] =
552 { "pg_abort", "pg_done", "pg_begin", "pg_try", "pg_fail", "pg_pass" };
d7ab1bab 553
d7ab1bab 554 rq++;
555 if (rq > N(meth)) SYSERR("event code out of range");
556 pyev = pgevent_pywrap(ev);
7f38dc76 557 if ((rc = PyObject_CallMethod(pg->obj, (/*unconst*/ char *)meth[rq],
827f89d7 558 "(O)", pyev)) == 0)
d7ab1bab 559 goto end;
560 if (rc == Py_None)
561 st = PGEN_TRY;
562 else if ((l = PyInt_AsLong(rc)) == -1 && PyErr_Occurred())
563 goto end;
564 else if (l < PGEN_ABORT || l > PGEN_PASS)
565 VALERR("return code out of range");
b2687a0a 566 else
d7ab1bab 567 st = l;
568end:
930d78e3
MW		 569 if (PyErr_Occurred())
570 stash_exception(pg->exc, "exception from `pgen' handler");
d7ab1bab 571 if (pyev) {
572 pgevent_kill(pyev);
573 Py_DECREF(pyev);
574 }
575 Py_XDECREF(rc);
d7ab1bab 576 return (st);
577}
578
579static PyObject *pgev_pywrap(const pgev *pg)
580{
581 pgev_pyobj *o;
582
583 o = PyObject_New(pgev_pyobj, pgev_pytype);
584 o->pg = *pg;
585 return ((PyObject *)o);
586}
587
588int convpgev(PyObject *o, void *p)
589{
930d78e3 590 pypgev *pg = p;
d7ab1bab 591
592 if (PGEV_PYCHECK(o))
930d78e3 593 pg->ev = *PGEV_PG(o);
d7ab1bab 594 else {
930d78e3
MW		 595 pg->ev.proc = pgev_python;
596 pg->ev.ctx = pg;
597 pg->obj = o; Py_INCREF(o);
d7ab1bab 598 }
599 return (1);
600}
601
930d78e3 602void droppgev(pypgev *pg)
d7ab1bab 603{
930d78e3
MW		 604 if (pg->ev.proc == pgev_python)
605 { assert(pg->ev.ctx == pg); Py_DECREF(pg->obj); }
d7ab1bab 606}
607
608static PyObject *pgmeth_common(PyObject *me, PyObject *arg, int rq)
609{
610 pgen_event *ev;
611 pgev *pg = PGEV_PG(me);
612 PyObject *rc = 0;
613
614 if (!PyArg_ParseTuple(arg, "O&", convpgevent, &ev)) goto end;
615 rc = PyInt_FromLong(!pg->proc ? rq : pg->proc(rq, ev, pg->ctx));
616end:
617 return (rc);
618}
619
620#define PGMETH(lc, uc) \
621 static PyObject *pgmeth_pg_##lc(PyObject *me, PyObject *arg) \
622 { return pgmeth_common(me, arg, PGEN_##uc); }
623PGMETH(abort, ABORT)
624PGMETH(done, DONE)
625PGMETH(begin, BEGIN)
626PGMETH(try, TRY)
627PGMETH(pass, PASS)
628PGMETH(fail, FAIL)
629#undef PGMETH
630
631static PyObject *pgev_stdev(pgen_proc *proc)
632 { pgev pg; pg.proc = proc; pg.ctx = 0; return (pgev_pywrap(&pg)); }
633
634static PyMethodDef pgev_pymethods[] = {
635#define METHNAME(name) pgmeth_##name
fb23e96c
MW		 636 METH (pg_abort, "E.pg_abort(EV) -> PGST -- prime generation aborted")
637 METH (pg_done, "E.pg_done(EV) -> PGST -- prime generation finished")
638 METH (pg_begin, "E.pg_begin(EV) -> PGST -- commence stepping/testing")
639 METH (pg_try, "E.pg_try(EV) -> PGST -- found new candidate")
640 METH (pg_pass, "E.pg_pass(EV) -> PGST -- passed primality test")
641 METH (pg_fail, "E.pg_fail(EV) -> PGST -- failed primality test")
d7ab1bab 642#undef METHNAME
643 { 0 }
644};
645
646static PyTypeObject pgev_pytype_skel = {
6d4db0bf 647 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 648 "PrimeGenBuiltinHandler", /* @tp_name@ */
d7ab1bab 649 sizeof(pgev_pyobj), /* @tp_basicsize@ */
650 0, /* @tp_itemsize@ */
651
3aa33042 652 0, /* @tp_dealloc@ */
d7ab1bab 653 0, /* @tp_print@ */
654 0, /* @tp_getattr@ */
655 0, /* @tp_setattr@ */
656 0, /* @tp_compare@ */
657 0, /* @tp_repr@ */
658 0, /* @tp_as_number@ */
659 0, /* @tp_as_sequence@ */
660 0, /* @tp_as_mapping@ */
661 0, /* @tp_hash@ */
662 0, /* @tp_call@ */
663 0, /* @tp_str@ */
664 0, /* @tp_getattro@ */
665 0, /* @tp_setattro@ */
666 0, /* @tp_as_buffer@ */
667 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
668 Py_TPFLAGS_BASETYPE,
669
670 /* @tp_doc@ */
d96c882e 671 "Built-in prime-generation event handler, base class.",
d7ab1bab 672
673 0, /* @tp_traverse@ */
674 0, /* @tp_clear@ */
675 0, /* @tp_richcompare@ */
676 0, /* @tp_weaklistoffset@ */
677 0, /* @tp_iter@ */
963a6148 678 0, /* @tp_iternext@ */
d7ab1bab 679 pgev_pymethods, /* @tp_methods@ */
680 0, /* @tp_members@ */
681 0, /* @tp_getset@ */
682 0, /* @tp_base@ */
683 0, /* @tp_dict@ */
684 0, /* @tp_descr_get@ */
685 0, /* @tp_descr_set@ */
686 0, /* @tp_dictoffset@ */
687 0, /* @tp_init@ */
688 PyType_GenericAlloc, /* @tp_alloc@ */
689 abstract_pynew, /* @tp_new@ */
3aa33042 690 0, /* @tp_free@ */
d7ab1bab 691 0 /* @tp_is_gc@ */
692};
693
694static PyObject *pgstep_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
695{
696 mpw s;
697 pgstep_pyobj *rc = 0;
827f89d7 698 static const char *const kwlist[] = { "step", 0 };
d7ab1bab 699
827f89d7 700 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&:new", KWLIST, convmpw, &s))
d7ab1bab 701 goto end;
702 rc = (pgstep_pyobj *)ty->tp_alloc(ty, 0);
703 rc->f.step = s;
704 rc->pg.proc = pgen_filter;
705 rc->pg.ctx = &rc->f;
706end:
707 return ((PyObject *)rc);
708}
709
710static PyObject *psget_step(PyObject *me, void *hunoz)
711 { return (PyInt_FromLong(PGSTEP_STEP(me))); }
712
713static PyGetSetDef pgstep_pygetset[] = {
714#define GETSETNAME(op, name) ps##op##_##name
b2687a0a 715 GET (step, "S.step -> step size for the stepper")
d7ab1bab 716#undef GETSETNAME
717 { 0 }
718};
719
720static PyTypeObject pgstep_pytype_skel = {
6d4db0bf 721 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 722 "PrimeGenStepper", /* @tp_name@ */
d7ab1bab 723 sizeof(pgstep_pyobj), /* @tp_basicsize@ */
724 0, /* @tp_itemsize@ */
725
726 0, /* @tp_dealloc@ */
727 0, /* @tp_print@ */
728 0, /* @tp_getattr@ */
729 0, /* @tp_setattr@ */
730 0, /* @tp_compare@ */
731 0, /* @tp_repr@ */
732 0, /* @tp_as_number@ */
733 0, /* @tp_as_sequence@ */
734 0, /* @tp_as_mapping@ */
735 0, /* @tp_hash@ */
736 0, /* @tp_call@ */
737 0, /* @tp_str@ */
738 0, /* @tp_getattro@ */
739 0, /* @tp_setattro@ */
740 0, /* @tp_as_buffer@ */
741 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
742 Py_TPFLAGS_BASETYPE,
743
744 /* @tp_doc@ */
d96c882e 745 "PrimeGenStepper(STEP): simple stepper with small-factors filter.",
d7ab1bab 746
747 0, /* @tp_traverse@ */
748 0, /* @tp_clear@ */
749 0, /* @tp_richcompare@ */
750 0, /* @tp_weaklistoffset@ */
751 0, /* @tp_iter@ */
963a6148 752 0, /* @tp_iternext@ */
d7ab1bab 753 0, /* @tp_methods@ */
754 0, /* @tp_members@ */
755 pgstep_pygetset, /* @tp_getset@ */
756 0, /* @tp_base@ */
757 0, /* @tp_dict@ */
758 0, /* @tp_descr_get@ */
759 0, /* @tp_descr_set@ */
760 0, /* @tp_dictoffset@ */
761 0, /* @tp_init@ */
762 PyType_GenericAlloc, /* @tp_alloc@ */
763 pgstep_pynew, /* @tp_new@ */
3aa33042 764 0, /* @tp_free@ */
d7ab1bab 765 0 /* @tp_is_gc@ */
766};
767
768static PyObject *pgjump_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
769{
770 PyObject *o, *fobj;
771 pgjump_pyobj *rc = 0;
827f89d7 772 static const char *const kwlist[] = { "jump", 0 };
d7ab1bab 773
827f89d7 774 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O:new", KWLIST, &o) ||
d7ab1bab 775 (fobj = pfilt_pymake(pfilt_pytype, o)) == 0)
776 goto end;
777 rc = (pgjump_pyobj *)ty->tp_alloc(ty, 0);
778 rc->fobj = fobj;
779 rc->j.j = PFILT_F(fobj);
780 rc->pg.proc = pgen_jump;
781 rc->pg.ctx = &rc->j;
782end:
783 return ((PyObject *)rc);
784}
785
786static void pgjump_pydealloc(PyObject *me)
787{
788 Py_DECREF(PGJUMP_FOBJ(me));
3aa33042 789 FREEOBJ(me);
d7ab1bab 790}
791
792static PyObject *pjget_jump(PyObject *me, void *hunoz)
793 { RETURN_OBJ(PGJUMP_FOBJ(me)); }
794
795static PyGetSetDef pgjump_pygetset[] = {
796#define GETSETNAME(op, name) pj##op##_##name
b2687a0a 797 GET (jump, "S.jump -> jump size for the stepper")
d7ab1bab 798#undef GETSETNAME
799 { 0 }
800};
801
802static PyTypeObject pgjump_pytype_skel = {
6d4db0bf 803 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 804 "PrimeGenJumper", /* @tp_name@ */
d7ab1bab 805 sizeof(pgjump_pyobj), /* @tp_basicsize@ */
806 0, /* @tp_itemsize@ */
807
808 pgjump_pydealloc, /* @tp_dealloc@ */
809 0, /* @tp_print@ */
810 0, /* @tp_getattr@ */
811 0, /* @tp_setattr@ */
812 0, /* @tp_compare@ */
813 0, /* @tp_repr@ */
814 0, /* @tp_as_number@ */
815 0, /* @tp_as_sequence@ */
816 0, /* @tp_as_mapping@ */
817 0, /* @tp_hash@ */
818 0, /* @tp_call@ */
819 0, /* @tp_str@ */
820 0, /* @tp_getattro@ */
821 0, /* @tp_setattro@ */
822 0, /* @tp_as_buffer@ */
823 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
824 Py_TPFLAGS_BASETYPE,
825
826 /* @tp_doc@ */
d96c882e
MW		 827 "PrimeGenJumper(JUMP): "
828 "stepper for larger steps with small-factors filter.",
d7ab1bab 829
830 0, /* @tp_traverse@ */
831 0, /* @tp_clear@ */
832 0, /* @tp_richcompare@ */
833 0, /* @tp_weaklistoffset@ */
834 0, /* @tp_iter@ */
963a6148 835 0, /* @tp_iternext@ */
d7ab1bab 836 0, /* @tp_methods@ */
837 0, /* @tp_members@ */
838 pgjump_pygetset, /* @tp_getset@ */
839 0, /* @tp_base@ */
840 0, /* @tp_dict@ */
841 0, /* @tp_descr_get@ */
842 0, /* @tp_descr_set@ */
843 0, /* @tp_dictoffset@ */
844 0, /* @tp_init@ */
845 PyType_GenericAlloc, /* @tp_alloc@ */
846 pgjump_pynew, /* @tp_new@ */
3aa33042 847 0, /* @tp_free@ */
d7ab1bab 848 0 /* @tp_is_gc@ */
849};
850
851static PyObject *pgtest_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
852{
853 pgtest_pyobj *rc = 0;
827f89d7 854 static const char *const kwlist[] = { 0 };
d7ab1bab 855
827f89d7 856 if (!PyArg_ParseTupleAndKeywords(arg, kw, ":new", KWLIST)) goto end;
d7ab1bab 857 rc = (pgtest_pyobj *)ty->tp_alloc(ty, 0);
858 rc->pg.proc = pgen_test;
859 rc->pg.ctx = &rc->r;
860end:
861 return ((PyObject *)rc);
862}
863
864static PyTypeObject pgtest_pytype_skel = {
6d4db0bf 865 PyObject_HEAD_INIT(0) 0, /* Header */
c461c9b3 866 "PrimeGenTester", /* @tp_name@ */
d7ab1bab 867 sizeof(pgtest_pyobj), /* @tp_basicsize@ */
868 0, /* @tp_itemsize@ */
869
870 0, /* @tp_dealloc@ */
871 0, /* @tp_print@ */
872 0, /* @tp_getattr@ */
873 0, /* @tp_setattr@ */
874 0, /* @tp_compare@ */
875 0, /* @tp_repr@ */
876 0, /* @tp_as_number@ */
877 0, /* @tp_as_sequence@ */
878 0, /* @tp_as_mapping@ */
879 0, /* @tp_hash@ */
880 0, /* @tp_call@ */
881 0, /* @tp_str@ */
882 0, /* @tp_getattro@ */
883 0, /* @tp_setattro@ */
884 0, /* @tp_as_buffer@ */
885 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
886 Py_TPFLAGS_BASETYPE,
887
888 /* @tp_doc@ */
d96c882e 889 "PrimeGenTester(): Rabin-Miller tester.",
d7ab1bab 890
891 0, /* @tp_traverse@ */
892 0, /* @tp_clear@ */
893 0, /* @tp_richcompare@ */
894 0, /* @tp_weaklistoffset@ */
895 0, /* @tp_iter@ */
963a6148 896 0, /* @tp_iternext@ */
d7ab1bab 897 0, /* @tp_methods@ */
898 0, /* @tp_members@ */
899 0, /* @tp_getset@ */
900 0, /* @tp_base@ */
901 0, /* @tp_dict@ */
902 0, /* @tp_descr_get@ */
903 0, /* @tp_descr_set@ */
904 0, /* @tp_dictoffset@ */
905 0, /* @tp_init@ */
906 PyType_GenericAlloc, /* @tp_alloc@ */
907 pgtest_pynew, /* @tp_new@ */
3aa33042 908 0, /* @tp_free@ */
d7ab1bab 909 0 /* @tp_is_gc@ */
910};
911
912/*----- Prime generation functions ----------------------------------------*/
913
930d78e3 914void pgenerr(struct excinfo *exc)
d7ab1bab 915{
930d78e3
MW		 916 if (exc->ty) RESTORE_EXCINFO(exc);
917 else PyErr_SetString(PyExc_ValueError, "prime generation failed");
d7ab1bab 918}
919
920static PyObject *meth_pgen(PyObject *me, PyObject *arg, PyObject *kw)
921{
922 mp *x = 0;
923 mp *r = 0;
924 PyObject *rc = 0;
925 char *p = "p";
926 pgen_filterctx fc = { 2 };
927 rabin tc;
930d78e3
MW		 928 struct excinfo exc = EXCINFO_INIT;
929 pypgev step = { { 0 } }, test = { { 0 } }, evt = { { 0 } };
d7ab1bab 930 unsigned nsteps = 0, ntests = 0;
827f89d7
MW		 931 static const char *const kwlist[] =
932 { "start", "name", "stepper", "tester", "event", "nsteps", "ntests", 0 };
d7ab1bab 933
930d78e3
MW		 934 step.exc = &exc; step.ev.proc = pgen_filter; step.ev.ctx = &fc;
935 test.exc = &exc; test.ev.proc = pgen_test; test.ev.ctx = &tc;
936 evt.exc = &exc;
827f89d7 937 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&O&O&:pgen", KWLIST,
d7ab1bab 938 convmp, &x, &p, convpgev, &step,
939 convpgev, &test, convpgev, &evt,
940 convuint, &nsteps, convuint, &ntests))
941 goto end;
942 if (!ntests) ntests = rabin_iters(mp_bits(x));
930d78e3
MW		 943 if ((r = pgen(p, MP_NEW, x, evt.ev.proc, evt.ev.ctx,
944 nsteps, step.ev.proc, step.ev.ctx,
945 ntests, test.ev.proc, test.ev.ctx)) == 0)
946 PGENERR(&exc);
947 rc = mp_pywrap(r); r = 0;
d7ab1bab 948end:
949 mp_drop(r); mp_drop(x);
950 droppgev(&step); droppgev(&test); droppgev(&evt);
b2687a0a 951 return (rc);
d7ab1bab 952}
953
954static PyObject *meth_strongprime_setup(PyObject *me,
955 PyObject *arg, PyObject *kw)
956{
957 mp *x = 0;
958 pfilt f;
959 grand *r = &rand_global;
960 unsigned nbits;
961 char *name = "p";
962 unsigned n = 0;
930d78e3
MW		 963 struct excinfo exc = EXCINFO_INIT;
964 pypgev evt = { { 0 } };
d7ab1bab 965 PyObject *rc = 0;
827f89d7
MW		 966 static const char *const kwlist[] =
967 { "nbits", "name", "event", "rng", "nsteps", 0 };
d7ab1bab 968
930d78e3 969 evt.exc = &exc;
827f89d7 970 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&", KWLIST,
d7ab1bab 971 convuint, &nbits, &name,
972 convpgev, &evt, convgrand, &r,
973 convuint, &n))
974 goto end;
975 if ((x = strongprime_setup(name, MP_NEW, &f, nbits,
930d78e3
MW		 976 r, n, evt.ev.proc, evt.ev.ctx)) == 0)
977 PGENERR(&exc);
d7ab1bab 978 rc = Py_BuildValue("(NN)", mp_pywrap(x), pfilt_pywrap(&f));
979 x = 0;
980end:
981 mp_drop(x);
982 droppgev(&evt);
983 return (rc);
984}
985
986static PyObject *meth_strongprime(PyObject *me, PyObject *arg, PyObject *kw)
987{
988 mp *x = 0;
989 grand *r = &rand_global;
990 unsigned nbits;
991 char *name = "p";
992 unsigned n = 0;
930d78e3
MW		 993 struct excinfo exc = EXCINFO_INIT;
994 pypgev evt = { { 0 } };
d7ab1bab 995 PyObject *rc = 0;
827f89d7
MW		 996 static const char *const kwlist[] =
997 { "nbits", "name", "event", "rng", "nsteps", 0 };
d7ab1bab 998
930d78e3 999 evt.exc = &exc;
827f89d7 1000 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&", KWLIST,
d7ab1bab 1001 convuint, &nbits, &name,
1002 convpgev, &evt, convgrand, &r,
1003 convuint, &n))
1004 goto end;
1005 if ((x = strongprime(name, MP_NEW, nbits,
930d78e3
MW		 1006 r, n, evt.ev.proc, evt.ev.ctx)) == 0)
1007 PGENERR(&exc);
d7ab1bab 1008 rc = mp_pywrap(x);
1009 x = 0;
1010end:
1011 mp_drop(x);
1012 droppgev(&evt);
1013 return (rc);
1014}
1015
1016static PyObject *meth_limlee(PyObject *me, PyObject *arg, PyObject *kw)
1017{
1018 char *p = "p";
930d78e3
MW		 1019 struct excinfo exc = EXCINFO_INIT;
1020 pypgev ie = { { 0 } }, oe = { { 0 } };
d7ab1bab 1021 unsigned ql, pl;
1022 grand *r = &rand_global;
1023 unsigned on = 0;
1024 size_t i, nf = 0;
1025 PyObject *rc = 0, *vec;
827f89d7
MW		 1026 static const char *const kwlist[] =
1027 { "pbits", "qbits", "name", "event", "ievent", "rng", "nsteps", 0 };
d7ab1bab 1028 mp *x = 0, **v = 0;
1029
930d78e3 1030 ie.exc = oe.exc = &exc;
827f89d7 1031 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&O&|sO&O&O&O&:limlee", KWLIST,
d7ab1bab 1032 convuint, &pl, convuint, &ql,
1033 &p, convpgev, &oe, convpgev, &ie,
1034 convgrand, &r, convuint, &on))
1035 goto end;
1036 if ((x = limlee(p, MP_NEW, MP_NEW, ql, pl, r, on,
930d78e3
MW		 1037 oe.ev.proc, oe.ev.ctx, ie.ev.proc, ie.ev.ctx,
1038 &nf, &v)) == 0)
1039 PGENERR(&exc);;
d7ab1bab 1040 vec = PyList_New(nf);
1041 for (i = 0; i < nf; i++)
c51a597d 1042 PyList_SET_ITEM(vec, i, mp_pywrap(v[i]));
d7ab1bab 1043 xfree(v);
1044 rc = Py_BuildValue("(NN)", mp_pywrap(x), vec);
1045end:
1046 droppgev(&oe); droppgev(&ie);
1047 return (rc);
1048}
1049
1050/*----- Global stuff ------------------------------------------------------*/
1051
1052static PyMethodDef methods[] = {
1053#define METHNAME(name) meth_##name
d96c882e
MW		 1054 METH (_PrimeFilter_smallfactor, "smallfactor(X) -> PGRC")
1055 METH (_RabinMiller_iters, "iters(NBITS) -> NITERS")
1056 KWMETH(pgen,
1057 "pgen(START, [name = 'p'], [stepper = PrimeGenStepper(2)],\n"
1058 " [tester = PrimeGenTester()], [event = pgen_nullev],\n"
1059 " [nsteps = 0], [ntests = RabinMiller.iters(START.nbits)]) -> P")
1060 KWMETH(strongprime_setup,
1061 "strongprime_setup(NBITS, [name = 'p'], [event = pgen_nullev],\n"
1062 " [rng = rand], [nsteps = 0]) -> (START, JUMP)")
1063 KWMETH(strongprime,
1064 "strongprime(NBITS, [name = 'p'], [event = pgen_nullev],\n"
1065 " [rng = rand], [nsteps = 0]) -> P")
1066 KWMETH(limlee,
1067 "limlee(PBITS, QBITS, [name = 'p'], [event = pgen_nullev],\n"
1068 " [ievent = pgen_nullev], [rng = rand], [nsteps = 0]) "
1069 "-> (P, [Q, ...])")
d7ab1bab 1070#undef METHNAME
1071 { 0 }
1072};
1073
1074void pgen_pyinit(void)
1075{
1076 INITTYPE(pfilt, root);
1077 INITTYPE(rabin, root);
1078 INITTYPE(pgevent, root);
1079 INITTYPE(pgev, root);
1080 INITTYPE(pgstep, pgev);
1081 INITTYPE(pgjump, pgev);
1082 INITTYPE(pgtest, pgev);
1083 addmethods(methods);
1084}
1085
1086static PyObject *obj;
1087
1088void pgen_pyinsert(PyObject *mod)
1089{
1090 INSERT("PrimeFilter", pfilt_pytype);
1091 INSERT("RabinMiller", rabin_pytype);
1092 INSERT("PrimeGenEvent", pgevent_pytype);
1093 INSERT("PrimeGenBuiltinHandler", pgev_pytype);
1094 INSERT("PrimeGenStepper", pgstep_pytype);
1095 INSERT("PrimeGenJumper", pgjump_pytype);
1096 INSERT("PrimeGenTester", pgtest_pytype);
1097 INSERT("pgen_nullev", obj = pgev_stdev(0));
1098 INSERT("pgen_stdev", pgev_stdev(pgen_ev));
1099 INSERT("pgen_spinev", pgev_stdev(pgen_evspin));
1100 INSERT("pgen_subev", pgev_stdev(pgen_subev));
1101}
1102
1103/*----- That's all, folks -------------------------------------------------*/
Catacomb cryptographic library -- Python bindings