3 * Random-number generators
5 * (c) 2004 Straylight/Edgeware
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of the Python interface to Catacomb.
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.
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.
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.
27 /*----- Header files ------------------------------------------------------*/
29 #include "catacomb-python.h"
31 #include "algorithms.h"
34 /*----- Main code ---------------------------------------------------------*/
36 PyTypeObject
*grand_pytype
;
37 static PyTypeObject
*truerand_pytype
;
38 static PyTypeObject
*lcrand_pytype
, *fibrand_pytype
;
39 static PyTypeObject
*dsarand_pytype
, *bbs_pytype
, *bbspriv_pytype
;
40 static PyTypeObject
*sslprf_pytype
, *tlsdx_pytype
, *tlsprf_pytype
;
43 static PyObject
*gccrands_dict
;
45 static PyObject
*grand_dopywrap(PyTypeObject
*ty
, grand
*r
, unsigned f
)
49 g
= (grand_pyobj
*)ty
->tp_alloc(ty
, 0);
52 return ((PyObject
*)g
);
55 PyObject
*grand_pywrap(grand
*r
, unsigned f
)
57 PyTypeObject
*ty
= grand_pytype
;
60 if (STRCMP(r
->ops
->name
, ==, "rand")) ty
= truerand_pytype
;
61 else if (STRCMP(r
->ops
->name
, ==, "lcrand")) ty
= lcrand_pytype
;
62 else if (STRCMP(r
->ops
->name
, ==, "fibrand")) ty
= fibrand_pytype
;
63 else if (STRCMP(r
->ops
->name
, ==, "dsarand")) ty
= dsarand_pytype
;
64 else if (STRCMP(r
->ops
->name
, ==, "bbs")) ty
= bbs_pytype
;
65 else if (STRCMP(r
->ops
->name
, ==, "sslprf")) ty
= sslprf_pytype
;
66 else if (STRCMP(r
->ops
->name
, ==, "tlsdx")) ty
= tlsdx_pytype
;
67 else if (STRCMP(r
->ops
->name
, ==, "tlsprf")) ty
= tlsprf_pytype
;
68 else if ((ob
= PyMapping_GetItemString
69 (gccrands_dict
, (/*unconst*/ char *)r
->ops
->name
)) != 0)
70 ty
= (PyTypeObject
*)ob
;
71 return (grand_dopywrap(ty
, r
, f
));
74 CONVFUNC(grand
, grand
*, GRAND_R
)
76 static int grand_check(PyObject
*me
)
78 if (!GRAND_R(me
)) VALERR("random generator object is no longer valid");
84 static PyObject
*grmeth_byte(PyObject
*me
)
86 if (grand_check(me
)) return (0);
87 return (PyInt_FromLong(grand_byte(GRAND_R(me
))));
90 static PyObject
*grmeth_word(PyObject
*me
)
92 if (grand_check(me
)) return (0);
93 return (getulong(grand_word(GRAND_R(me
))));
96 static PyObject
*grmeth_range(PyObject
*me
, PyObject
*arg
)
102 if (!PyArg_ParseTuple(arg
, "O:range", &m
)) return (0);
103 if (grand_check(me
)) return (0);
104 if (PyInt_Check(m
)) {
105 long mm
= PyInt_AsLong(m
);
106 if (mm
== -1 && PyErr_Occurred()) PyErr_Clear();
107 else if (mm
<= 0) goto notpos
;
108 else if (mm
<= 0xffffffff)
109 return (PyInt_FromLong(grand_range(GRAND_R(me
), mm
)));
111 if ((x
= getmp(m
)) == 0) goto end
;
112 if (!MP_POSP(x
)) goto notpos
;
113 y
= mprand_range(MP_NEW
, x
, GRAND_R(me
), 0);
115 return (mp_pywrap(y
));
117 VALERR("range must be strictly positive");
123 static PyObject
*grmeth_mp(PyObject
*me
, PyObject
*arg
, PyObject
*kw
)
127 static const char *const kwlist
[] = { "bits", "or", 0 };
129 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:mp", KWLIST
,
130 convszt
, &l
, convmpw
, &o
))
132 if (grand_check(me
)) return (0);
133 if (l
< MPW_BITS
&& (o
>> l
)) VALERR("or mask too large");
134 return (mp_pywrap(mprand(MP_NEW
, l
, GRAND_R(me
), o
)));
139 static PyObject
*grmeth_block(PyObject
*me
, PyObject
*arg
)
144 if (!PyArg_ParseTuple(arg
, "O&:block", convulong
, &n
)) goto end
;
145 if (grand_check(me
)) return (0);
146 rc
= bytestring_pywrap(0, n
);
147 grand_fill(GRAND_R(me
), BIN_PTR(rc
), n
);
152 static int checkop(grand
*r
, unsigned op
, const char *what
)
154 if (r
->ops
->misc(r
, GRAND_CHECK
, op
)) return (0);
155 PyErr_Format(PyExc_TypeError
, "operation %s not supported", what
);
159 static PyObject
*grmeth_seedint(PyObject
*me
, PyObject
*arg
)
162 grand
*r
= GRAND_R(me
);
163 if (!PyArg_ParseTuple(arg
, "i:seedint", &i
) ||
164 grand_check(me
) || checkop(r
, GRAND_SEEDINT
, "seedint"))
166 r
->ops
->misc(r
, GRAND_SEEDINT
, i
);
172 static PyObject
*grmeth_seedword(PyObject
*me
, PyObject
*arg
)
175 grand
*r
= GRAND_R(me
);
176 if (!PyArg_ParseTuple(arg
, "O&:seedword", convu32
, &u
) ||
177 grand_check(me
) || checkop(r
, GRAND_SEEDUINT32
, "seedword"))
179 r
->ops
->misc(r
, GRAND_SEEDUINT32
, u
);
185 static PyObject
*grmeth_seedblock(PyObject
*me
, PyObject
*arg
)
188 grand
*r
= GRAND_R(me
);
189 if (!PyArg_ParseTuple(arg
, "O&:seedblock", convbin
, &in
) ||
190 grand_check(me
) || checkop(r
, GRAND_SEEDBLOCK
, "seedblock"))
192 r
->ops
->misc(r
, GRAND_SEEDBLOCK
, in
.p
, (size_t)in
.sz
);
198 static PyObject
*grmeth_seedmp(PyObject
*me
, PyObject
*arg
)
202 grand
*r
= GRAND_R(me
);
203 if (!PyArg_ParseTuple(arg
, "O:seedmp", &x
) ||
204 grand_check(me
) || checkop(r
, GRAND_SEEDMP
, "seedmp") ||
205 (xx
= getmp(x
)) == 0)
207 r
->ops
->misc(r
, GRAND_SEEDMP
, xx
);
214 static PyObject
*grmeth_seedrand(PyObject
*me
, PyObject
*arg
, PyObject
*kw
)
216 static const char *const kwlist
[] = { "rng", 0 };
217 grand
*r
= GRAND_R(me
);
218 grand
*rr
= &rand_global
;
219 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "|O&:seedrand", KWLIST
,
221 grand_check(me
) || checkop(r
, GRAND_SEEDRAND
, "seedrand"))
223 r
->ops
->misc(r
, GRAND_SEEDRAND
, rr
);
229 static PyObject
*grmeth_mask(PyObject
*me
, PyObject
*arg
)
231 grand
*r
= GRAND_R(me
);
233 const octet
*p
; size_t n
;
237 if (!PyArg_ParseTuple(arg
, "O&:mask", convbin
, &in
)) return (0);
238 if (grand_check(me
)) return (0);
239 rc
= bytestring_pywrap(0, in
.sz
);
240 q
= (octet
*)BIN_PTR(rc
);
241 GR_FILL(r
, q
, in
.sz
);
242 p
= in
.p
; n
= in
.sz
; while (n
--) *q
++ ^= *p
++;
246 static void grand_pydealloc(PyObject
*me
)
248 grand_pyobj
*g
= (grand_pyobj
*)me
;
249 if ((g
->f
& f_freeme
) && g
->r
) GR_DESTROY(g
->r
);
253 static PyObject
*grget_name(PyObject
*me
, void *hunoz
)
254 { return (grand_check(me
) ?
0 : TEXT_FROMSTR(GRAND_R(me
)->ops
->name
)); }
256 static PyObject
*grget_cryptop(PyObject
*me
, void *hunoz
)
257 { return (grand_check(me
) ?
0 : getbool(GRAND_R(me
)->ops
->f
& GRAND_CRYPTO
)); }
259 static const PyGetSetDef grand_pygetset
[] = {
260 #define GETSETNAME(op, name) gr##op##_##name
261 GET (name
, "R.name -> name of this kind of generator")
262 GET (cryptop
, "R.cryptop -> flag: cryptographically strong?")
267 static const PyMethodDef grand_pymethods
[] = {
268 #define METHNAME(name) grmeth_##name
269 NAMETH(byte
, "R.byte() -> BYTE")
270 NAMETH(word
, "R.word() -> WORD")
271 METH (block
, "R.block(N) -> STRING")
272 KWMETH(mp
, "R.mp(bits, [or = 0]) -> MP")
273 METH (range
, "R.range(MAX) -> INT")
274 METH (mask
, "R.mask(STR) -> STR")
275 METH (seedint
, "R.seedint(I)")
276 METH (seedword
, "R.seedword(I)")
277 METH (seedblock
, "R.seedblock(BYTES)")
278 METH (seedmp
, "R.seedmp(X)")
279 KWMETH(seedrand
, "R.seedrand(RR)")
284 static const PyTypeObject grand_pytype_skel
= {
285 PyVarObject_HEAD_INIT(0, 0) /* Header */
286 "GRand", /* @tp_name@ */
287 sizeof(grand_pyobj
), /* @tp_basicsize@ */
288 0, /* @tp_itemsize@ */
290 grand_pydealloc
, /* @tp_dealloc@ */
292 0, /* @tp_getattr@ */
293 0, /* @tp_setattr@ */
294 0, /* @tp_compare@ */
296 0, /* @tp_as_number@ */
297 0, /* @tp_as_sequence@ */
298 0, /* @tp_as_mapping@ */
302 0, /* @tp_getattro@ */
303 0, /* @tp_setattro@ */
304 0, /* @tp_as_buffer@ */
305 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
309 "Generic random number source.",
311 0, /* @tp_traverse@ */
313 0, /* @tp_richcompare@ */
314 0, /* @tp_weaklistoffset@ */
316 0, /* @tp_iternext@ */
317 PYMETHODS(grand
), /* @tp_methods@ */
318 0, /* @tp_members@ */
319 PYGETSET(grand
), /* @tp_getset@ */
322 0, /* @tp_descr_get@ */
323 0, /* @tp_descr_set@ */
324 0, /* @tp_dictoffset@ */
326 PyType_GenericAlloc
, /* @tp_alloc@ */
327 abstract_pynew
, /* @tp_new@ */
332 static PyObject
*lcrand_pynew(PyTypeObject
*me
, PyObject
*arg
, PyObject
*kw
)
335 static const char *const kwlist
[] = { "seed", 0 };
336 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "|O&:new", KWLIST
, convu32
, &n
))
338 return (grand_dopywrap(lcrand_pytype
, lcrand_create(n
), f_freeme
));
341 static const PyTypeObject lcrand_pytype_skel
= {
342 PyVarObject_HEAD_INIT(0, 0) /* Header */
343 "LCRand", /* @tp_name@ */
344 sizeof(grand_pyobj
), /* @tp_basicsize@ */
345 0, /* @tp_itemsize@ */
347 grand_pydealloc
, /* @tp_dealloc@ */
349 0, /* @tp_getattr@ */
350 0, /* @tp_setattr@ */
351 0, /* @tp_compare@ */
353 0, /* @tp_as_number@ */
354 0, /* @tp_as_sequence@ */
355 0, /* @tp_as_mapping@ */
359 0, /* @tp_getattro@ */
360 0, /* @tp_setattro@ */
361 0, /* @tp_as_buffer@ */
362 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
366 "LCRand([seed = 0]): linear congruential generator.",
368 0, /* @tp_traverse@ */
370 0, /* @tp_richcompare@ */
371 0, /* @tp_weaklistoffset@ */
373 0, /* @tp_iternext@ */
374 0, /* @tp_methods@ */
375 0, /* @tp_members@ */
379 0, /* @tp_descr_get@ */
380 0, /* @tp_descr_set@ */
381 0, /* @tp_dictoffset@ */
383 PyType_GenericAlloc
, /* @tp_alloc@ */
384 lcrand_pynew
, /* @tp_new@ */
389 static PyObject
*fibrand_pynew(PyTypeObject
*me
, PyObject
*arg
, PyObject
*kw
)
392 static const char *const kwlist
[] = { "seed", 0 };
393 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "|O&:new", KWLIST
, convu32
, &n
))
395 return (grand_dopywrap(fibrand_pytype
, fibrand_create(n
), f_freeme
));
398 static const PyTypeObject fibrand_pytype_skel
= {
399 PyVarObject_HEAD_INIT(0, 0) /* Header */
400 "FibRand", /* @tp_name@ */
401 sizeof(grand_pyobj
), /* @tp_basicsize@ */
402 0, /* @tp_itemsize@ */
404 grand_pydealloc
, /* @tp_dealloc@ */
406 0, /* @tp_getattr@ */
407 0, /* @tp_setattr@ */
408 0, /* @tp_compare@ */
410 0, /* @tp_as_number@ */
411 0, /* @tp_as_sequence@ */
412 0, /* @tp_as_mapping@ */
416 0, /* @tp_getattro@ */
417 0, /* @tp_setattro@ */
418 0, /* @tp_as_buffer@ */
419 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
423 "FibRand([seed = 0]): Fibonacci generator.",
425 0, /* @tp_traverse@ */
427 0, /* @tp_richcompare@ */
428 0, /* @tp_weaklistoffset@ */
430 0, /* @tp_iternext@ */
431 0, /* @tp_methods@ */
432 0, /* @tp_members@ */
436 0, /* @tp_descr_get@ */
437 0, /* @tp_descr_set@ */
438 0, /* @tp_dictoffset@ */
440 PyType_GenericAlloc
, /* @tp_alloc@ */
441 fibrand_pynew
, /* @tp_new@ */
446 /*----- True random generator ---------------------------------------------*/
448 static PyObject
*trmeth_gate(PyObject
*me
)
449 { grand
*r
= GRAND_R(me
); r
->ops
->misc(GRAND_R(me
), RAND_GATE
); RETURN_ME
; }
451 static PyObject
*trmeth_stretch(PyObject
*me
)
452 { grand
*r
= GRAND_R(me
); r
->ops
->misc(r
, RAND_STRETCH
); RETURN_ME
; }
454 static PyObject
*trmeth_add(PyObject
*me
, PyObject
*arg
)
456 grand
*r
= GRAND_R(me
);
457 struct bin in
; unsigned goodbits
;
458 if (!PyArg_ParseTuple(arg
, "O&O&:add", convbin
, &in
, convuint
, &goodbits
))
460 r
->ops
->misc(r
, RAND_ADD
, in
.p
, (size_t)in
.sz
, goodbits
);
464 static PyObject
*trmeth_key(PyObject
*me
, PyObject
*arg
)
466 grand
*r
= GRAND_R(me
);
468 if (!PyArg_ParseTuple(arg
, "O&:key", convbin
, &k
)) return (0);
469 r
->ops
->misc(r
, RAND_KEY
, k
.p
, (size_t)k
.sz
);
473 static PyObject
*trmeth_seed(PyObject
*me
, PyObject
*arg
)
475 grand
*r
= GRAND_R(me
);
477 if (!PyArg_ParseTuple(arg
, "O&:seed", convuint
, &u
)) return (0);
478 if (u
> RAND_IBITS
) VALERR("pointlessly large");
479 r
->ops
->misc(r
, RAND_SEED
, u
);
485 static PyObject
*trmeth_timer(PyObject
*me
)
486 { grand
*r
= GRAND_R(me
); r
->ops
->misc(r
, RAND_TIMER
); RETURN_ME
; }
488 static PyObject
*truerand_pynew(PyTypeObject
*ty
,
489 PyObject
*arg
, PyObject
*kw
)
491 static const char *const kwlist
[] = { 0 };
494 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, ":new", KWLIST
)) goto end
;
496 r
->ops
->misc(r
, RAND_NOISESRC
, &noise_source
);
497 r
->ops
->misc(r
, RAND_SEED
, 160);
498 rc
= grand_dopywrap(ty
, r
, f_freeme
);
503 static const PyMethodDef truerand_pymethods
[] = {
504 #define METHNAME(name) trmeth_##name
505 NAMETH(gate
, "R.gate()")
506 NAMETH(stretch
, "R.stretch()")
507 METH (key
, "R.key(BYTES)")
508 METH (seed
, "R.seed(NBITS)")
509 METH (add
, "R.add(BYTES, GOODBITS")
510 NAMETH(timer
, "R.timer()")
515 static PyObject
*trget_goodbits(PyObject
*me
, void *hunoz
)
517 grand
*r
= GRAND_R(me
);
518 return (PyInt_FromLong(r
->ops
->misc(r
, RAND_GOODBITS
)));
521 static const PyGetSetDef truerand_pygetset
[] = {
522 #define GETSETNAME(op, name) tr##op##_##name
523 GET (goodbits
, "R.goodbits -> good bits of entropy remaining")
528 static const PyTypeObject truerand_pytype_skel
= {
529 PyVarObject_HEAD_INIT(0, 0) /* Header */
530 "TrueRand", /* @tp_name@ */
531 sizeof(grand_pyobj
), /* @tp_basicsize@ */
532 0, /* @tp_itemsize@ */
534 grand_pydealloc
, /* @tp_dealloc@ */
536 0, /* @tp_getattr@ */
537 0, /* @tp_setattr@ */
538 0, /* @tp_compare@ */
540 0, /* @tp_as_number@ */
541 0, /* @tp_as_sequence@ */
542 0, /* @tp_as_mapping@ */
546 0, /* @tp_getattro@ */
547 0, /* @tp_setattro@ */
548 0, /* @tp_as_buffer@ */
549 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
553 "TrueRand(): true random number source.",
555 0, /* @tp_traverse@ */
557 0, /* @tp_richcompare@ */
558 0, /* @tp_weaklistoffset@ */
560 0, /* @tp_iternext@ */
561 PYMETHODS(truerand
), /* @tp_methods@ */
562 0, /* @tp_members@ */
563 PYGETSET(truerand
), /* @tp_getset@ */
566 0, /* @tp_descr_get@ */
567 0, /* @tp_descr_set@ */
568 0, /* @tp_dictoffset@ */
570 PyType_GenericAlloc
, /* @tp_alloc@ */
571 truerand_pynew
, /* @tp_new@ */
576 /*----- Generators from symmetric encryption algorithms -------------------*/
578 static PyTypeObject
*gccrand_pytype
, *gcrand_pytype
, *gclatinrand_pytype
;
580 typedef grand
*gcrand_func(const void *, size_t sz
);
581 typedef grand
*gcirand_func(const void *, size_t sz
, uint32
);
582 typedef grand
*gcnrand_func(const void *, size_t sz
, const void *);
583 typedef grand
*gcshakerand_func(const void *, size_t,
584 const void *, size_t,
585 const void *, size_t);
586 typedef grand
*gcshafuncrand_func(const void *, size_t,
587 const void *, size_t);
588 typedef grand
*gckmacrand_func(const void *, size_t, const void *, size_t);
589 typedef struct gccrand_info
{
597 #define RNGF_MASK 255u
607 typedef struct gccrand_pyobj
{
609 const gccrand_info
*info
;
611 #define GCCRAND_INFO(o) (((gccrand_pyobj *)(o))->info)
613 #define GCCRAND_DEF(name, ksz, func, f, nsz) \
614 static const gccrand_info func##_info = \
615 { name, ksz, f, nsz, (gcrand_func *)func };
618 static const gccrand_info
*const gcrandtab
[] = {
619 #define GCCRAND_ENTRY(name, ksz, func, f, nsz) &func##_info,
624 static PyObject
*gcrand_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
626 const gccrand_info
*info
= GCCRAND_INFO(ty
);
627 static const char *const kwlist
[] = { "key", 0 };
630 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&:new", KWLIST
, convbin
, &k
))
632 if (keysz(k
.sz
, info
->keysz
) != k
.sz
) VALERR("bad key length");
633 return (grand_dopywrap(ty
, info
->func(k
.p
, k
.sz
), f_freeme
));
638 static PyObject
*gcirand_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
640 const gccrand_info
*info
= GCCRAND_INFO(ty
);
642 static const char *const kwlist
[] = { "key", "i", 0 };
645 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:new", KWLIST
,
646 convbin
, &k
, convu32
, &i
))
648 if (keysz(k
.sz
, info
->keysz
) != k
.sz
) VALERR("bad key length");
649 return (grand_dopywrap(ty
,
650 ((gcirand_func
*)info
->func
)(k
.p
, k
.sz
, i
),
656 static PyObject
*gcnrand_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
658 const gccrand_info
*info
= GCCRAND_INFO(ty
);
659 static const char *const kwlist
[] = { "key", "nonce", 0 };
660 static const octet zn
[24] = { 0 };
663 n
.p
= zn
; n
.sz
= info
->noncesz
; assert(info
->noncesz
<= sizeof(zn
));
664 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:new", KWLIST
,
665 convbin
, &k
, convbin
, &n
))
667 if (keysz(k
.sz
, info
->keysz
) != k
.sz
) VALERR("bad key length");
668 if (n
.sz
!= info
->noncesz
) VALERR("bad nonce length");
669 return (grand_dopywrap(ty
,
670 ((gcnrand_func
*)info
->func
)(k
.p
, k
.sz
, n
.p
),
676 static PyObject
*gcshakyrand_pynew(PyTypeObject
*ty
,
677 PyObject
*arg
, PyObject
*kw
)
679 const gccrand_info
*info
= GCCRAND_INFO(ty
);
681 *const kwlist_shake
[] = { "key", "func", "perso", 0 },
682 *const kwlist_func
[] = { "key", "perso", 0 };
683 struct bin k
, f
= { 0, 0 }, p
= { 0, 0 };
685 if ((info
->f
&RNGF_MASK
) == RNG_SHAKE
686 ?
!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&O&:new",
687 (/*unconst*/ char **)kwlist_shake
,
689 convbin
, &f
, convbin
, &p
)
690 : !PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:new",
691 (/*unconst*/ char **)kwlist_func
,
692 convbin
, &k
, convbin
, &p
))
694 if (keysz(k
.sz
, info
->keysz
) != k
.sz
) VALERR("bad key length");
695 return (grand_dopywrap(ty
,
696 (info
->f
&RNGF_MASK
) == RNG_SHAKE
697 ?
((gcshakerand_func
*)info
->func
)(f
.p
, f
.sz
,
700 : ((gcshafuncrand_func
*)info
->func
)(p
.p
, p
.sz
,
707 static PyObject
*gccrand_pywrap(const gccrand_info
*info
)
709 gccrand_pyobj
*g
= newtype(gccrand_pytype
, 0, info
->name
);
711 g
->ty
.ht_type
.tp_basicsize
= sizeof(grand_pyobj
);
712 switch (info
->f
&RNGF_MASK
) {
713 case RNG_LATIN
: g
->ty
.ht_type
.tp_base
= gclatinrand_pytype
; break;
714 default: g
->ty
.ht_type
.tp_base
= gcrand_pytype
; break;
716 Py_INCREF(g
->ty
.ht_type
.tp_base
);
717 g
->ty
.ht_type
.tp_flags
= (Py_TPFLAGS_DEFAULT
|
718 Py_TPFLAGS_BASETYPE
|
719 Py_TPFLAGS_HEAPTYPE
);
720 g
->ty
.ht_type
.tp_alloc
= PyType_GenericAlloc
;
721 g
->ty
.ht_type
.tp_free
= 0;
722 switch (info
->f
&RNGF_MASK
) {
723 case RNG_LATIN
: g
->ty
.ht_type
.tp_new
= gcnrand_pynew
; break;
724 case RNG_SEAL
: g
->ty
.ht_type
.tp_new
= gcirand_pynew
; break;
725 case RNG_SHAKE
: case RNG_KMAC
:
726 g
->ty
.ht_type
.tp_new
= gcshakyrand_pynew
; break;
727 default: g
->ty
.ht_type
.tp_new
= gcrand_pynew
; break;
729 typeready(&g
->ty
.ht_type
);
730 return ((PyObject
*)g
);
733 static PyObject
*gccrget_name(PyObject
*me
, void *hunoz
)
734 { return (TEXT_FROMSTR(GCCRAND_INFO(me
)->name
)); }
735 static PyObject
*gccrget_keysz(PyObject
*me
, void *hunoz
)
736 { return (keysz_pywrap(GCCRAND_INFO(me
)->keysz
)); }
738 static PyObject
*gclrmeth_tell(PyObject
*me
)
740 grand
*r
= GRAND_R(me
);
744 r
->ops
->misc(r
, SALSA20_TELLU64
, &off
);
749 static PyObject
*gclrmeth_seek(PyObject
*me
, PyObject
*arg
)
751 grand
*r
= GRAND_R(me
);
754 if (!PyArg_ParseTuple(arg
, "O&:seek", convk64
, &off
)) return (0);
755 r
->ops
->misc(r
, SALSA20_SEEKU64
, off
);
759 static const PyGetSetDef gccrand_pygetset
[] = {
760 #define GETSETNAME(op, name) gccr##op##_##name
761 GET (keysz
, "CR.keysz -> acceptable key sizes")
762 GET (name
, "CR.name -> name of this kind of generator")
767 static const PyMethodDef gclatinrand_pymethods
[] = {
768 #define METHNAME(name) gclrmeth_##name
769 NAMETH(tell
, "R.tell() -> OFF")
770 METH (seek
, "R.seek(OFF)")
775 static const PyTypeObject gccrand_pytype_skel
= {
776 PyVarObject_HEAD_INIT(0, 0) /* Header */
777 "GCCRand", /* @tp_name@ */
778 sizeof(gccrand_pyobj
), /* @tp_basicsize@ */
779 0, /* @tp_itemsize@ */
781 0, /* @tp_dealloc@ */
783 0, /* @tp_getattr@ */
784 0, /* @tp_setattr@ */
785 0, /* @tp_compare@ */
787 0, /* @tp_as_number@ */
788 0, /* @tp_as_sequence@ */
789 0, /* @tp_as_mapping@ */
793 0, /* @tp_getattro@ */
794 0, /* @tp_setattro@ */
795 0, /* @tp_as_buffer@ */
796 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
800 "Metaclass for symmetric crypto-based generators.",
802 0, /* @tp_traverse@ */
804 0, /* @tp_richcompare@ */
805 0, /* @tp_weaklistoffset@ */
807 0, /* @tp_iternext@ */
808 0, /* @tp_methods@ */
809 0, /* @tp_members@ */
810 PYGETSET(gccrand
), /* @tp_getset@ */
813 0, /* @tp_descr_get@ */
814 0, /* @tp_descr_set@ */
815 0, /* @tp_dictoffset@ */
817 PyType_GenericAlloc
, /* @tp_alloc@ */
818 abstract_pynew
, /* @tp_new@ */
823 static const PyTypeObject gcrand_pytype_skel
= {
824 PyVarObject_HEAD_INIT(0, 0) /* Header */
825 "GCRand", /* @tp_name@ */
826 sizeof(grand_pyobj
), /* @tp_basicsize@ */
827 0, /* @tp_itemsize@ */
829 grand_pydealloc
, /* @tp_dealloc@ */
831 0, /* @tp_getattr@ */
832 0, /* @tp_setattr@ */
833 0, /* @tp_compare@ */
835 0, /* @tp_as_number@ */
836 0, /* @tp_as_sequence@ */
837 0, /* @tp_as_mapping@ */
841 0, /* @tp_getattro@ */
842 0, /* @tp_setattro@ */
843 0, /* @tp_as_buffer@ */
844 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
848 "Abstract base class for symmetric crypto-based generators.",
850 0, /* @tp_traverse@ */
852 0, /* @tp_richcompare@ */
853 0, /* @tp_weaklistoffset@ */
855 0, /* @tp_iternext@ */
856 0, /* @tp_methods@ */
857 0, /* @tp_members@ */
861 0, /* @tp_descr_get@ */
862 0, /* @tp_descr_set@ */
863 0, /* @tp_dictoffset@ */
865 PyType_GenericAlloc
, /* @tp_alloc@ */
866 abstract_pynew
, /* @tp_new@ */
871 static const PyTypeObject gclatinrand_pytype_skel
= {
872 PyVarObject_HEAD_INIT(0, 0) /* Header */
873 "GCLatinRand", /* @tp_name@ */
874 sizeof(grand_pyobj
), /* @tp_basicsize@ */
875 0, /* @tp_itemsize@ */
877 grand_pydealloc
, /* @tp_dealloc@ */
879 0, /* @tp_getattr@ */
880 0, /* @tp_setattr@ */
881 0, /* @tp_compare@ */
883 0, /* @tp_as_number@ */
884 0, /* @tp_as_sequence@ */
885 0, /* @tp_as_mapping@ */
889 0, /* @tp_getattro@ */
890 0, /* @tp_setattro@ */
891 0, /* @tp_as_buffer@ */
892 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
896 "Abstract base class for symmetric crypto-based generators.",
898 0, /* @tp_traverse@ */
900 0, /* @tp_richcompare@ */
901 0, /* @tp_weaklistoffset@ */
903 0, /* @tp_iternext@ */
904 PYMETHODS(gclatinrand
), /* @tp_methods@ */
905 0, /* @tp_members@ */
909 0, /* @tp_descr_get@ */
910 0, /* @tp_descr_set@ */
911 0, /* @tp_dictoffset@ */
913 PyType_GenericAlloc
, /* @tp_alloc@ */
914 abstract_pynew
, /* @tp_new@ */
919 /*----- SSL and TLS generators --------------------------------------------*/
921 static PyObject
*sslprf_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
924 const gchash
*hco
= &md5
, *hci
= &sha
;
926 static const char *const kwlist
[] = { "key", "seed", "ohash", "ihash", 0 };
928 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&O&|O&O&:new", KWLIST
,
929 convbin
, &k
, convbin
, &s
,
930 convgchash
, &hco
, convgchash
, &hci
))
932 rc
= grand_dopywrap(ty
, sslprf_rand(hco
, hci
, k
.p
, k
.sz
, s
.p
, s
.sz
),
938 static PyObject
*tlsdx_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
941 const gcmac
*mc
= &sha_hmac
;
943 static const char *const kwlist
[] = { "key", "seed", "mac", 0 };
945 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&O&|O&:new", KWLIST
,
946 convbin
, &k
, convbin
, &s
,
949 rc
= grand_dopywrap(ty
, tlsdx_rand(mc
, k
.p
, k
.sz
, s
.p
, s
.sz
), f_freeme
);
954 static PyObject
*tlsprf_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
957 const gcmac
*mcl
= &md5_hmac
, *mcr
= &sha_hmac
;
959 static const char *const kwlist
[] = { "key", "seed", "lmac", "rmac", 0 };
961 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&O&|O&O&:new", KWLIST
,
962 convbin
, &k
, convbin
, &s
,
963 convgcmac
, &mcl
, convgcmac
, &mcr
))
965 rc
= grand_dopywrap(ty
, tlsprf_rand(mcl
, mcr
, k
.p
, k
.sz
, s
.p
, s
.sz
),
971 static const PyTypeObject sslprf_pytype_skel
= {
972 PyVarObject_HEAD_INIT(0, 0) /* Header */
973 "SSLRand", /* @tp_name@ */
974 sizeof(grand_pyobj
), /* @tp_basicsize@ */
975 0, /* @tp_itemsize@ */
977 grand_pydealloc
, /* @tp_dealloc@ */
979 0, /* @tp_getattr@ */
980 0, /* @tp_setattr@ */
981 0, /* @tp_compare@ */
983 0, /* @tp_as_number@ */
984 0, /* @tp_as_sequence@ */
985 0, /* @tp_as_mapping@ */
989 0, /* @tp_getattro@ */
990 0, /* @tp_setattro@ */
991 0, /* @tp_as_buffer@ */
992 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
996 "SSLRand(KEY, SEED, [ohash = md5], [ihash = sha]):\n"
997 " RNG for SSL master secret.",
999 0, /* @tp_traverse@ */
1001 0, /* @tp_richcompare@ */
1002 0, /* @tp_weaklistoffset@ */
1004 0, /* @tp_iternext@ */
1005 0, /* @tp_methods@ */
1006 0, /* @tp_members@ */
1007 0, /* @tp_getset@ */
1010 0, /* @tp_descr_get@ */
1011 0, /* @tp_descr_set@ */
1012 0, /* @tp_dictoffset@ */
1014 PyType_GenericAlloc
, /* @tp_alloc@ */
1015 sslprf_pynew
, /* @tp_new@ */
1020 static const PyTypeObject tlsdx_pytype_skel
= {
1021 PyVarObject_HEAD_INIT(0, 0) /* Header */
1022 "TLSDataExpansion", /* @tp_name@ */
1023 sizeof(grand_pyobj
), /* @tp_basicsize@ */
1024 0, /* @tp_itemsize@ */
1026 grand_pydealloc
, /* @tp_dealloc@ */
1028 0, /* @tp_getattr@ */
1029 0, /* @tp_setattr@ */
1030 0, /* @tp_compare@ */
1032 0, /* @tp_as_number@ */
1033 0, /* @tp_as_sequence@ */
1034 0, /* @tp_as_mapping@ */
1038 0, /* @tp_getattro@ */
1039 0, /* @tp_setattro@ */
1040 0, /* @tp_as_buffer@ */
1041 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
1042 Py_TPFLAGS_BASETYPE
,
1045 "TLSDataExpansion(KEY, SEED, [mac = sha_hmac]):\n"
1046 " TLS data expansion function.",
1048 0, /* @tp_traverse@ */
1050 0, /* @tp_richcompare@ */
1051 0, /* @tp_weaklistoffset@ */
1053 0, /* @tp_iternext@ */
1054 0, /* @tp_methods@ */
1055 0, /* @tp_members@ */
1056 0, /* @tp_getset@ */
1059 0, /* @tp_descr_get@ */
1060 0, /* @tp_descr_set@ */
1061 0, /* @tp_dictoffset@ */
1063 PyType_GenericAlloc
, /* @tp_alloc@ */
1064 tlsdx_pynew
, /* @tp_new@ */
1069 static const PyTypeObject tlsprf_pytype_skel
= {
1070 PyVarObject_HEAD_INIT(0, 0) /* Header */
1071 "TLSPRF", /* @tp_name@ */
1072 sizeof(grand_pyobj
), /* @tp_basicsize@ */
1073 0, /* @tp_itemsize@ */
1075 grand_pydealloc
, /* @tp_dealloc@ */
1077 0, /* @tp_getattr@ */
1078 0, /* @tp_setattr@ */
1079 0, /* @tp_compare@ */
1081 0, /* @tp_as_number@ */
1082 0, /* @tp_as_sequence@ */
1083 0, /* @tp_as_mapping@ */
1087 0, /* @tp_getattro@ */
1088 0, /* @tp_setattro@ */
1089 0, /* @tp_as_buffer@ */
1090 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
1091 Py_TPFLAGS_BASETYPE
,
1094 "TLSPRF(KEY, SEED, [lmac = md5_hmac], [rmac = sha_hmac]):\n"
1095 " TLS pseudorandom function.",
1097 0, /* @tp_traverse@ */
1099 0, /* @tp_richcompare@ */
1100 0, /* @tp_weaklistoffset@ */
1102 0, /* @tp_iternext@ */
1103 0, /* @tp_methods@ */
1104 0, /* @tp_members@ */
1105 0, /* @tp_getset@ */
1108 0, /* @tp_descr_get@ */
1109 0, /* @tp_descr_set@ */
1110 0, /* @tp_dictoffset@ */
1112 PyType_GenericAlloc
, /* @tp_alloc@ */
1113 tlsprf_pynew
, /* @tp_new@ */
1118 /*----- DSA generator -----------------------------------------------------*/
1120 static PyObject
*dsarand_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
1123 unsigned passes
= 1;
1126 static const char *const kwlist
[] = { "seed", "passes", 0 };
1128 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:new", KWLIST
,
1129 convbin
, &in
, convuint
, &passes
))
1131 if (!passes
) VALERR("must be positive");
1132 r
= dsarand_create(in
.p
, in
.sz
);
1133 if (passes
!= 1) r
->ops
->misc(r
, DSARAND_PASSES
, passes
);
1134 rc
= grand_dopywrap(ty
, r
, f_freeme
);
1139 static PyObject
*drget_seed(PyObject
*me
, void *hunoz
)
1141 grand
*r
= GRAND_R(me
);
1142 int n
= r
->ops
->misc(r
, DSARAND_SEEDSZ
);
1143 PyObject
*rc
= bytestring_pywrap(0, n
);
1144 r
->ops
->misc(r
, DSARAND_GETSEED
, BIN_PTR(rc
));
1148 static PyObject
*drget_passes(PyObject
*me
, void *hunoz
)
1150 grand
*r
= GRAND_R(me
);
1151 return (PyInt_FromLong(r
->ops
->misc(r
, DSARAND_PASSES
, 0)));
1154 static int drset_passes(PyObject
*me
, PyObject
*val
, void *hunoz
)
1156 grand
*r
= GRAND_R(me
);
1160 if (!val
) NIERR("__del__");
1161 n
= PyInt_AsLong(val
); if (n
== -1 && PyErr_Occurred()) goto end
;
1162 if (n
<= 0) VALERR("must be positive");
1163 if (n
> ULONG_MAX
) VALERR("out of range");
1164 r
->ops
->misc(r
, DSARAND_PASSES
, (unsigned)n
);
1170 static const PyGetSetDef dsarand_pygetset
[] = {
1171 #define GETSETNAME(op, name) dr##op##_##name
1172 GET (seed
, "R.seed -> current generator seed")
1173 GETSET(passes
, "R.passes -> number of passes to create output")
1178 static const PyTypeObject dsarand_pytype_skel
= {
1179 PyVarObject_HEAD_INIT(0, 0) /* Header */
1180 "DSARand", /* @tp_name@ */
1181 sizeof(grand_pyobj
), /* @tp_basicsize@ */
1182 0, /* @tp_itemsize@ */
1184 grand_pydealloc
, /* @tp_dealloc@ */
1186 0, /* @tp_getattr@ */
1187 0, /* @tp_setattr@ */
1188 0, /* @tp_compare@ */
1190 0, /* @tp_as_number@ */
1191 0, /* @tp_as_sequence@ */
1192 0, /* @tp_as_mapping@ */
1196 0, /* @tp_getattro@ */
1197 0, /* @tp_setattro@ */
1198 0, /* @tp_as_buffer@ */
1199 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
1200 Py_TPFLAGS_BASETYPE
,
1203 "DSARand(SEED): pseudorandom number generator for DSA parameters.",
1205 0, /* @tp_traverse@ */
1207 0, /* @tp_richcompare@ */
1208 0, /* @tp_weaklistoffset@ */
1210 0, /* @tp_iternext@ */
1211 0, /* @tp_methods@ */
1212 0, /* @tp_members@ */
1213 PYGETSET(dsarand
), /* @tp_getset@ */
1216 0, /* @tp_descr_get@ */
1217 0, /* @tp_descr_set@ */
1218 0, /* @tp_dictoffset@ */
1220 PyType_GenericAlloc
, /* @tp_alloc@ */
1221 dsarand_pynew
, /* @tp_new@ */
1226 /*----- Blum-Blum-Shub generator ------------------------------------------*/
1228 static PyObject
*bbs_pynew(PyTypeObject
*ty
, PyObject
*arg
, PyObject
*kw
)
1230 mp
*n
= 0, *x
= MP_TWO
;
1232 static const char *const kwlist
[] = { "n", "x", 0 };
1234 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&:new", KWLIST
,
1235 convmp
, &n
, convmp
, &x
))
1237 rc
= grand_dopywrap(ty
, bbs_rand(n
, x
), f_freeme
);
1244 static PyObject
*bbsmeth_step(PyObject
*me
)
1245 { grand
*r
= GRAND_R(me
); r
->ops
->misc(r
, BBS_STEP
); RETURN_ME
; }
1247 static PyObject
*bbsmeth_bits(PyObject
*me
, PyObject
*arg
)
1249 grand
*r
= GRAND_R(me
); unsigned n
; uint32 w
;
1250 if (!PyArg_ParseTuple(arg
, "O&:bits", convuint
, &n
)) goto end
;
1251 if (n
> 32) VALERR("can't get more than 32 bits");
1252 r
->ops
->misc(r
, BBS_BITS
, n
, &w
); return (getulong(w
));
1257 static PyObject
*bbsmeth_wrap(PyObject
*me
)
1258 { grand
*r
= GRAND_R(me
); r
->ops
->misc(r
, BBS_WRAP
); RETURN_ME
; }
1260 static PyObject
*bbsget_n(PyObject
*me
, void *hunoz
)
1262 mp
*x
= MP_NEW
; grand
*r
= GRAND_R(me
);
1263 r
->ops
->misc(r
, BBS_MOD
, &x
); return (mp_pywrap(x
));
1266 static PyObject
*bbsget_x(PyObject
*me
, void *hunoz
)
1268 mp
*x
= MP_NEW
; grand
*r
= GRAND_R(me
);
1269 r
->ops
->misc(r
, BBS_STATE
, &x
); return (mp_pywrap(x
));
1272 static int bbsset_x(PyObject
*me
, PyObject
*val
, void *hunoz
)
1274 mp
*x
= 0; grand
*r
= GRAND_R(me
); int rc
= -1; if (!val
) NIERR("__del__");
1275 if ((x
= getmp(val
)) == 0) goto end
;
1276 r
->ops
->misc(r
, BBS_SET
, x
); rc
= 0;
1277 end
: mp_drop(x
); return (rc
);
1280 static PyObject
*bbsget_stepsz(PyObject
*me
, void *hunoz
)
1282 grand
*r
= GRAND_R(me
);
1283 return (PyInt_FromLong(r
->ops
->misc(r
, BBS_STEPSZ
)));
1286 static const PyMethodDef bbs_pymethods
[] = {
1287 #define METHNAME(name) bbsmeth_##name
1288 NAMETH(step
, "R.step(): steps the generator (not useful)")
1289 METH (bits
, "R.bits(N) -> W: returns N bits (<= 32) from the generator")
1290 NAMETH(wrap
, "R.wrap(): flushes unused bits in internal buffer")
1295 static const PyGetSetDef bbs_pygetset
[] = {
1296 #define GETSETNAME(op, name) bbs##op##_##name
1297 GET (n
, "R.n -> Blum modulus")
1298 GETSET(x
, "R.x -> current seed value")
1299 GET (stepsz
, "R.stepsz -> number of bits generated per step")
1304 static const PyTypeObject bbs_pytype_skel
= {
1305 PyVarObject_HEAD_INIT(0, 0) /* Header */
1306 "BlumBlumShub", /* @tp_name@ */
1307 sizeof(grand_pyobj
), /* @tp_basicsize@ */
1308 0, /* @tp_itemsize@ */
1310 grand_pydealloc
, /* @tp_dealloc@ */
1312 0, /* @tp_getattr@ */
1313 0, /* @tp_setattr@ */
1314 0, /* @tp_compare@ */
1316 0, /* @tp_as_number@ */
1317 0, /* @tp_as_sequence@ */
1318 0, /* @tp_as_mapping@ */
1322 0, /* @tp_getattro@ */
1323 0, /* @tp_setattro@ */
1324 0, /* @tp_as_buffer@ */
1325 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
1326 Py_TPFLAGS_BASETYPE
,
1329 "BlumBlumShub(N, [x = 2]): Blum-Blum-Shub pseudorandom number generator.",
1331 0, /* @tp_traverse@ */
1333 0, /* @tp_richcompare@ */
1334 0, /* @tp_weaklistoffset@ */
1336 0, /* @tp_iternext@ */
1337 PYMETHODS(bbs
), /* @tp_methods@ */
1338 0, /* @tp_members@ */
1339 PYGETSET(bbs
), /* @tp_getset@ */
1342 0, /* @tp_descr_get@ */
1343 0, /* @tp_descr_set@ */
1344 0, /* @tp_dictoffset@ */
1346 PyType_GenericAlloc
, /* @tp_alloc@ */
1347 bbs_pynew
, /* @tp_new@ */
1352 typedef struct bbspriv_pyobj
{
1357 #define BBSPRIV_BP(o) (&((bbspriv_pyobj *)(o))->bp)
1359 static PyObject
*bbspriv_pynew(PyTypeObject
*ty
,
1360 PyObject
*arg
, PyObject
*kw
)
1362 mp
*p
= 0, *q
= 0, *n
= 0, *x
= MP_TWO
;
1363 bbspriv_pyobj
*rc
= 0;
1364 static const char *const kwlist
[] = { "n", "p", "q", "seed", 0 };
1366 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "|O&O&O&O&:new", KWLIST
,
1367 convmp
, &n
, convmp
, &p
, convmp
, &q
,
1370 if (!n
+ !p
+ !q
> 1) VALERR("must specify at least two of n, p, q");
1371 if (!n
) n
= mp_mul(MP_NEW
, p
, q
);
1372 else if (!p
) mp_div(&p
, 0, n
, q
);
1373 else if (!q
) mp_div(&q
, 0, n
, p
);
1374 rc
= (bbspriv_pyobj
*)ty
->tp_alloc(ty
, 0);
1375 rc
->gr
.r
= bbs_rand(n
, x
);
1376 rc
->gr
.f
= f_freeme
;
1377 rc
->bp
.p
= MP_COPY(p
);
1378 rc
->bp
.q
= MP_COPY(q
);
1379 rc
->bp
.n
= MP_COPY(n
);
1381 mp_drop(p
); mp_drop(q
); mp_drop(n
); mp_drop(x
);
1382 return ((PyObject
*)rc
);
1385 static PyObject
*bpmeth_generate(PyObject
*me
, PyObject
*arg
, PyObject
*kw
)
1387 bbs_priv bp
= { 0 };
1389 struct excinfo exc
= EXCINFO_INIT
;
1390 pypgev evt
= { { 0 } };
1391 unsigned nbits
, n
= 0;
1392 grand
*r
= &rand_global
;
1393 static const char *const kwlist
[] =
1394 { "nbits", "event", "rng", "nsteps", "seed", 0 };
1395 bbspriv_pyobj
*rc
= 0;
1398 if (!PyArg_ParseTupleAndKeywords(arg
, kw
, "O&|O&O&O&O&:generate", KWLIST
,
1399 convuint
, &nbits
, convpgev
, &evt
,
1400 convgrand
, &r
, convuint
, &n
, convmp
, &x
))
1402 if (bbs_gen(&bp
, nbits
, r
, n
, evt
.ev
.proc
, evt
.ev
.ctx
))
1404 rc
= PyObject_New(bbspriv_pyobj
, bbspriv_pytype
);
1405 rc
->gr
.r
= bbs_rand(bp
.n
, x
);
1406 rc
->gr
.f
= f_freeme
;
1407 rc
->bp
.p
= MP_COPY(bp
.p
);
1408 rc
->bp
.q
= MP_COPY(bp
.q
);
1409 rc
->bp
.n
= MP_COPY(bp
.n
);
1411 mp_drop(bp
.p
); mp_drop(bp
.q
); mp_drop(bp
.n
); mp_drop(x
);
1412 return ((PyObject
*)rc
);
1415 static void bbspriv_pydealloc(PyObject
*me
)
1417 bbs_priv
*bp
= BBSPRIV_BP(me
);
1421 grand_pydealloc(me
);
1424 static PyObject
*bpmeth_ff(PyObject
*me
, PyObject
*arg
)
1426 mp
*n
= 0; grand
*r
= GRAND_R(me
); bbs_priv
*bp
= BBSPRIV_BP(me
);
1427 if (!PyArg_ParseTuple(arg
, "O&:ff", convmp
, &n
)) return (0);
1428 r
->ops
->misc(r
, BBS_FF
, bp
, n
); RETURN_ME
;
1431 static PyObject
*bpmeth_rew(PyObject
*me
, PyObject
*arg
)
1433 mp
*n
= 0; grand
*r
= GRAND_R(me
); bbs_priv
*bp
= BBSPRIV_BP(me
);
1434 if (!PyArg_ParseTuple(arg
, "O&:rew", convmp
, &n
)) return (0);
1435 r
->ops
->misc(r
, BBS_REW
, bp
, n
); RETURN_ME
;
1438 static PyObject
*bpget_n(PyObject
*me
, void *hunoz
)
1439 { return (mp_pywrap(MP_COPY(BBSPRIV_BP(me
)->n
))); }
1441 static PyObject
*bpget_p(PyObject
*me
, void *hunoz
)
1442 { return (mp_pywrap(MP_COPY(BBSPRIV_BP(me
)->p
))); }
1444 static PyObject
*bpget_q(PyObject
*me
, void *hunoz
)
1445 { return (mp_pywrap(MP_COPY(BBSPRIV_BP(me
)->q
))); }
1447 static const PyMethodDef bbspriv_pymethods
[] = {
1448 #define METHNAME(name) bpmeth_##name
1449 METH (ff
, "R.ff(N): fast-forward N places")
1450 METH (rew
, "R.rew(N): rewind N places")
1451 KWSMTH(generate
, "generate(NBITS, [event = pgen_nullev], "
1452 "[rng = rand], [nsteps = 0], [seed = 2]) -> R")
1457 static const PyGetSetDef bbspriv_pygetset
[] = {
1458 #define GETSETNAME(op, name) bp##op##_##name
1459 GET (n
, "R.n -> Blum modulus")
1460 GET (p
, "R.p -> one of the factors of the modulus")
1461 GET (q
, "R.q -> one of the factors of the modulus")
1466 static const PyTypeObject bbspriv_pytype_skel
= {
1467 PyVarObject_HEAD_INIT(0, 0) /* Header */
1468 "BBSPriv", /* @tp_name@ */
1469 sizeof(bbspriv_pyobj
), /* @tp_basicsize@ */
1470 0, /* @tp_itemsize@ */
1472 bbspriv_pydealloc
, /* @tp_dealloc@ */
1474 0, /* @tp_getattr@ */
1475 0, /* @tp_setattr@ */
1476 0, /* @tp_compare@ */
1478 0, /* @tp_as_number@ */
1479 0, /* @tp_as_sequence@ */
1480 0, /* @tp_as_mapping@ */
1484 0, /* @tp_getattro@ */
1485 0, /* @tp_setattro@ */
1486 0, /* @tp_as_buffer@ */
1487 Py_TPFLAGS_DEFAULT
| /* @tp_flags@ */
1488 Py_TPFLAGS_BASETYPE
,
1491 "BBSPriv(..., [seed = 2]): Blum-Blum-Shub, with private key.\n"
1492 " Keywords: n, p, q; must provide at least two",
1494 0, /* @tp_traverse@ */
1496 0, /* @tp_richcompare@ */
1497 0, /* @tp_weaklistoffset@ */
1499 0, /* @tp_iternext@ */
1500 PYMETHODS(bbspriv
), /* @tp_methods@ */
1501 0, /* @tp_members@ */
1502 PYGETSET(bbspriv
), /* @tp_getset@ */
1505 0, /* @tp_descr_get@ */
1506 0, /* @tp_descr_set@ */
1507 0, /* @tp_dictoffset@ */
1509 PyType_GenericAlloc
, /* @tp_alloc@ */
1510 bbspriv_pynew
, /* @tp_new@ */
1515 /*----- Global stuff ------------------------------------------------------*/
1517 static const struct nameval consts
[] = {
1522 void rand_pyinit(void)
1524 INITTYPE(grand
, root
);
1525 INITTYPE(truerand
, grand
);
1526 INITTYPE(fibrand
, grand
);
1527 INITTYPE(lcrand
, grand
);
1528 INITTYPE(dsarand
, grand
);
1529 INITTYPE(bbs
, grand
);
1530 INITTYPE(bbspriv
, bbs
);
1531 INITTYPE(sslprf
, grand
);
1532 INITTYPE(tlsdx
, grand
);
1533 INITTYPE(tlsprf
, grand
);
1534 INITTYPE(gccrand
, type
);
1535 INITTYPE(gcrand
, grand
);
1536 INITTYPE(gclatinrand
, gcrand
);
1537 rand_noisesrc(RAND_GLOBAL
, &noise_source
);
1538 rand_seed(RAND_GLOBAL
, 160);
1541 static const char *crand_namefn(const void *p
)
1542 { const gccrand_info
*const *cls
= p
; return (*cls ?
(*cls
)->name
: 0); }
1543 static PyObject
*crand_valfn(const void *p
)
1544 { const gccrand_info
*const *cls
= p
; return (gccrand_pywrap(*cls
)); }
1546 void rand_pyinsert(PyObject
*mod
)
1548 INSERT("GRand", grand_pytype
);
1549 INSERT("TrueRand", truerand_pytype
);
1550 INSERT("LCRand", lcrand_pytype
);
1551 INSERT("FibRand", fibrand_pytype
);
1552 INSERT("SSLRand", sslprf_pytype
);
1553 INSERT("TLSDataExpansion", tlsdx_pytype
);
1554 INSERT("TLSPRF", tlsprf_pytype
);
1555 INSERT("DSARand", dsarand_pytype
);
1556 INSERT("BlumBlumShub", bbs_pytype
);
1557 INSERT("BBSPriv", bbspriv_pytype
);
1558 INSERT("GCCRand", gccrand_pytype
);
1559 INSERT("GCRand", gcrand_pytype
);
1560 INSERT("GCLatinRand", gclatinrand_pytype
);
1561 rand_pyobj
= grand_pywrap(&rand_global
, 0); Py_INCREF(rand_pyobj
);
1562 gccrands_dict
= make_algtab(gcrandtab
, sizeof(gccrand_info
*),
1563 crand_namefn
, crand_valfn
);
1564 INSERT("gccrands", gccrands_dict
); Py_INCREF(gccrands_dict
);
1565 INSERT("rand", rand_pyobj
);
1566 setconstants(mod
, consts
);
1569 /*----- That's all, folks -------------------------------------------------*/