chiark / gitweb /
*.c: Use Python `METH_NOARGS' methods where applicable.
[catacomb-python] / group.c
1 /* -*-c-*-
2  *
3  * Abstract group inteface
4  *
5  * (c) 2004 Straylight/Edgeware
6  */
7
8 /*----- Licensing notice --------------------------------------------------*
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.
16  *
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.
21  *
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 /*----- DH and binary group infos -----------------------------------------*/
32
33 static PyTypeObject *fginfo_pytype, *dhinfo_pytype, *bindhinfo_pytype;
34
35 typedef struct fginfo_pyobj {
36   PyObject_HEAD
37   gprime_param dp;
38 } fginfo_pyobj;
39
40 #define FGINFO_DP(fg) (&((fginfo_pyobj *)(fg))->dp)
41
42 static PyObject *fginfo_pywrap(gprime_param *dp, PyTypeObject *ty)
43 {
44   fginfo_pyobj *z = PyObject_New(fginfo_pyobj, ty);
45   z->dp = *dp;
46   return ((PyObject *)z);
47 }
48
49 static PyObject *fginfo_pynew(PyTypeObject *ty,
50                               PyObject *arg, PyObject *kw)
51 {
52   static const char *const kwlist[] = { "p", "r", "g", 0 };
53   gprime_param dp = { 0 };
54   fginfo_pyobj *z = 0;
55
56   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&O&O&:new", KWLIST,
57                                    convmp, &dp.p,
58                                    convmp, &dp.q,
59                                    convmp, &dp.g))
60     goto end;
61   z = PyObject_New(fginfo_pyobj, ty);
62   z->dp = dp;
63   return ((PyObject *)z);
64 end:
65   mp_drop(dp.p);
66   mp_drop(dp.q);
67   mp_drop(dp.g);
68   return (0);
69 }
70
71 static PyObject *figet_r(PyObject *me, void *hunoz)
72   { return mp_pywrap(MP_COPY(FGINFO_DP(me)->q)); }
73
74 static PyObject *diget_p(PyObject *me, void *hunoz)
75   { return mp_pywrap(MP_COPY(FGINFO_DP(me)->p)); }
76
77 static PyObject *diget_g(PyObject *me, void *hunoz)
78   { return mp_pywrap(MP_COPY(FGINFO_DP(me)->g)); }
79
80 static PyObject *biget_p(PyObject *me, void *hunoz)
81   { return gf_pywrap(MP_COPY(FGINFO_DP(me)->p)); }
82
83 static PyObject *biget_m(PyObject *me, void *hunoz)
84   { return PyInt_FromLong(mp_octets(FGINFO_DP(me)->p) - 1); }
85
86 static PyObject *biget_g(PyObject *me, void *hunoz)
87   { return gf_pywrap(MP_COPY(FGINFO_DP(me)->g)); }
88
89 static void fginfo_pydealloc(PyObject *me)
90 {
91   mp_drop(FGINFO_DP(me)->p);
92   mp_drop(FGINFO_DP(me)->q);
93   mp_drop(FGINFO_DP(me)->g);
94   FREEOBJ(me);
95 }
96
97 static PyObject *dimeth_generate(PyObject *me, PyObject *arg, PyObject *kw)
98 {
99   dh_param dp;
100   unsigned ql = 0, pl;
101   unsigned steps = 0;
102   grand *r = &rand_global;
103   struct excinfo exc = EXCINFO_INIT;
104   pypgev evt = { { 0 } };
105   static const char *const kwlist[] =
106     { "pbits", "qbits", "event", "rng", "nsteps", 0 };
107   PyObject *rc = 0;
108
109   evt.exc = &exc;
110   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|O&O&O&O&:generate", KWLIST,
111                                    convuint, &pl, convuint, &ql,
112                                    convpgev, &evt, convgrand, &r,
113                                    convuint, &steps))
114     goto end;
115   if (dh_gen(&dp, ql, pl, steps, r, evt.ev.proc, evt.ev.ctx))
116     PGENERR(&exc);
117   rc = fginfo_pywrap(&dp, dhinfo_pytype);
118 end:
119   droppgev(&evt);
120   return (rc);
121 }
122
123 static PyObject *dimeth_genlimlee(PyObject *me, PyObject *arg, PyObject *kw)
124 {
125   dh_param dp;
126   unsigned ql, pl;
127   unsigned steps = 0;
128   grand *r = &rand_global;
129   struct excinfo exc = EXCINFO_INIT;
130   pypgev oe = { { 0 } }, ie = { { 0 } };
131   int subgroupp = 1;
132   unsigned f = 0;
133   static const char *const kwlist[] = {
134     "pbits", "qbits", "event", "ievent",
135     "rng", "nsteps", "subgroupp", 0
136   };
137   size_t i, nf;
138   mp **v = 0;
139   PyObject *rc = 0, *vec = 0;
140
141   oe.exc = ie.exc = &exc;
142   if (!PyArg_ParseTupleAndKeywords(arg, kw,
143                                    "O&O&|O&O&O&O&O&:genlimlee", KWLIST,
144                                    convuint, &pl, convuint, &ql,
145                                    convpgev, &oe, convpgev, &ie,
146                                    convgrand, &r, convuint, &steps,
147                                    convbool, &subgroupp))
148     goto end;
149   if (subgroupp) f |= DH_SUBGROUP;
150   if (dh_limlee(&dp, ql, pl, f, steps, r,
151                 oe.ev.proc, oe.ev.ctx, ie.ev.proc, ie.ev.ctx, &nf, &v))
152     PGENERR(&exc);
153   vec = PyList_New(nf);
154   for (i = 0; i < nf; i++)
155     PyList_SET_ITEM(vec, i, mp_pywrap(v[i]));
156   xfree(v);
157   rc = Py_BuildValue("(NN)", fginfo_pywrap(&dp, dhinfo_pytype), vec);
158 end:
159   droppgev(&oe); droppgev(&ie);
160   return (rc);
161 }
162
163 static PyObject *dimeth_genkcdsa(PyObject *me, PyObject *arg, PyObject *kw)
164 {
165   dh_param dp;
166   unsigned ql, pl;
167   unsigned steps = 0;
168   grand *r = &rand_global;
169   struct excinfo exc = EXCINFO_INIT;
170   pypgev evt = { { 0 } };
171   static const char *const kwlist[] =
172     { "pbits", "qbits", "event", "rng", "nsteps", 0 };
173   mp *v = MP_NEW;
174   PyObject *rc = 0;
175
176   evt.exc = &exc;
177   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&O&|O&O&O&:genkcdsa", KWLIST,
178                                    convuint, &pl, convuint, &ql,
179                                    convpgev, &evt, convgrand, &r,
180                                    convuint, &steps))
181     goto end;
182   if (dh_kcdsagen(&dp, ql, pl, 0, steps, r, evt.ev.proc, evt.ev.ctx))
183     PGENERR(&exc);
184   mp_div(&v, 0, dp.p, dp.q);
185   v = mp_lsr(v, v, 1);
186   rc = Py_BuildValue("(NN)", fginfo_pywrap(&dp, dhinfo_pytype),
187                      mp_pywrap(v));
188 end:
189   droppgev(&evt);
190   return (rc);
191 }
192
193 static PyObject *dimeth_gendsa(PyObject *me, PyObject *arg, PyObject *kw)
194 {
195   dsa_param dp;
196   unsigned ql, pl;
197   unsigned steps = 0;
198   dsa_seed ds;
199   char *k;
200   Py_ssize_t ksz;
201   struct excinfo exc = EXCINFO_INIT;
202   pypgev evt = { { 0 } };
203   static const char *const kwlist[] =
204     { "pbits", "qbits", "seed", "event", "nsteps", 0 };
205   PyObject *rc = 0;
206
207   evt.exc = &exc;
208   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&O&s#|O&O&:gendsa", KWLIST,
209                                    convuint, &pl, convuint, &ql,
210                                    &k, &ksz, convpgev, &evt,
211                                    convuint, &steps))
212     goto end;
213   if (dsa_gen(&dp, ql, pl, steps, k, ksz, &ds, evt.ev.proc, evt.ev.ctx))
214     PGENERR(&exc);
215   rc = Py_BuildValue("(NNl)", fginfo_pywrap(&dp, dhinfo_pytype),
216                      bytestring_pywrap(ds.p, ds.sz), (long)ds.count);
217   xfree(ds.p);
218 end:
219   droppgev(&evt);
220   return (rc);
221 }
222
223 static int npgroups = -1, nbingroups = -1;
224
225 static PyObject *namedgroups(const pentry *pp, int *ne)
226 {
227   int i, j;
228   const char *p;
229   PyObject *d, *c;
230
231   d = PyDict_New();
232   for (i = 0; pp[i].name; i++) {
233     p = pp[i].name;
234     for (j = 0; j < i; j++) {
235       if (pp[i].data == pp[j].data) {
236         c = PyDict_GetItemString(d, (/*unconst*/ char *)pp[j].name);
237         Py_INCREF(c);
238         goto found;
239       }
240     }
241     c = PyInt_FromLong(i);
242   found:
243     PyDict_SetItemString(d, (/*unconst*/ char *)p, c);
244     Py_DECREF(c);
245   }
246   *ne = i;
247   return (d);
248 }
249
250 static PyObject *meth__groupn(PyObject *me, PyObject *arg,
251                               PyTypeObject *ty, const pentry *pp, int ne)
252 {
253   int i;
254   gprime_param gp;
255   PyObject *rc = 0;
256
257   if (!PyArg_ParseTuple(arg, "i:_groupn", &i)) goto end;
258   if (i < 0 || i >= ne) VALERR("group index out of range");
259   dh_infofromdata(&gp, pp[i].data);
260   rc = fginfo_pywrap(&gp, ty);
261 end:
262   return (rc);
263 }
264
265 static PyObject *dimeth__groupn(PyObject *me, PyObject *arg)
266   { return (meth__groupn(me, arg, dhinfo_pytype, ptab, npgroups)); }
267
268 static PyObject *bimeth__groupn(PyObject *me, PyObject *arg)
269   { return (meth__groupn(me, arg, bindhinfo_pytype, bintab, nbingroups)); }
270
271 static PyObject *meth__parse(PyObject *me, PyObject *arg, PyTypeObject *ty,
272                              int (*parse)(qd_parse *, gprime_param *))
273 {
274   qd_parse qd;
275   char *p;
276   gprime_param gp;
277   PyObject *rc = 0;
278
279   if (!PyArg_ParseTuple(arg, "s:parse", &p)) goto end;
280   qd.p = p; qd.e = 0;
281   if (parse(&qd, &gp)) VALERR(qd.e);
282   rc = fginfo_pywrap(&gp, ty);
283 end:
284   return (rc);
285 }
286
287 static PyObject *dimeth_parse(PyObject *me, PyObject *arg)
288   { return (meth__parse(me, arg, dhinfo_pytype, dh_parse)); }
289
290 static PyObject *bimeth_parse(PyObject *me, PyObject *arg)
291   { return (meth__parse(me, arg, bindhinfo_pytype, dhbin_parse)); }
292
293 static const PyGetSetDef fginfo_pygetset[] = {
294 #define GETSETNAME(op, name) fi##op##_##name
295   GET   (r,             "I.r -> group order")
296 #undef GETSETNAME
297   { 0 }
298 };
299
300 static const PyGetSetDef dhinfo_pygetset[] = {
301 #define GETSETNAME(op, name) di##op##_##name
302   GET   (p,             "I.p -> prime")
303   GET   (g,             "I.g -> generator")
304 #undef GETSETNAME
305   { 0 }
306 };
307
308 static const PyMethodDef dhinfo_pymethods[] = {
309 #define METHNAME(name) dimeth_##name
310   SMTH  (parse,         "parse(STR) -> D, REST")
311   SMTH  (_groupn,       0)
312   KWSMTH(generate,
313        "generate(PBITS, [qbits = 0], [event = pgen_nullev],\n"
314        "         [rng = rand], [nsteps = 0]) -> D")
315   KWSMTH(genlimlee,
316        "genlimlee(PBITS, QBITS, [event = pgen_nullev], "
317                                                   "[ievent = pgen_nullev],\n"
318        "          [rng = rand], [nsteps = 0], [subgroupp = True]) "
319                                                           "-> (D, [Q, ...])")
320   KWSMTH(gendsa,
321        "gendsa(PBITS, QBITS, SEED, [event = pgen_nullev], [nsteps = 0])\n"
322        "  -> (D, SEED, COUNT)")
323   KWSMTH(genkcdsa,
324        "gendsa(PBITS, QBITS, [event = pgen_nullev], "
325                                               "[rng = rand], [nsteps = 0])\n"
326        "  -> (D, V)")
327 #undef METHNAME
328   { 0 }
329 };
330
331 static const PyGetSetDef bindhinfo_pygetset[] = {
332 #define GETSETNAME(op, name) bi##op##_##name
333   GET   (p,             "I.p -> irreducible polynomial")
334   GET   (m,             "I.m -> degree of polynomial")
335   GET   (g,             "I.g -> generator")
336 #undef GETSETNAME
337   { 0 }
338 };
339
340 static const PyMethodDef bindhinfo_pymethods[] = {
341 #define METHNAME(name) bimeth_##name
342   SMTH  (parse,         "parse(STR) -> D, REST")
343   SMTH  (_groupn,       0)
344 #undef METHNAME
345   { 0 }
346 };
347
348 static PyTypeObject fginfo_pytype_skel = {
349   PyObject_HEAD_INIT(0) 0,              /* Header */
350   "FGInfo",                             /* @tp_name@ */
351   sizeof(fginfo_pyobj),                 /* @tp_basicsize@ */
352   0,                                    /* @tp_itemsize@ */
353
354   fginfo_pydealloc,                     /* @tp_dealloc@ */
355   0,                                    /* @tp_print@ */
356   0,                                    /* @tp_getattr@ */
357   0,                                    /* @tp_setattr@ */
358   0,                                    /* @tp_compare@ */
359   0,                                    /* @tp_repr@ */
360   0,                                    /* @tp_as_number@ */
361   0,                                    /* @tp_as_sequence@ */
362   0,                                    /* @tp_as_mapping@ */
363   0,                                    /* @tp_hash@ */
364   0,                                    /* @tp_call@ */
365   0,                                    /* @tp_str@ */
366   0,                                    /* @tp_getattro@ */
367   0,                                    /* @tp_setattro@ */
368   0,                                    /* @tp_as_buffer@ */
369   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
370     Py_TPFLAGS_BASETYPE,
371
372   /* @tp_doc@ */
373   "Abstract base class for field-group information objects.",
374
375   0,                                    /* @tp_traverse@ */
376   0,                                    /* @tp_clear@ */
377   0,                                    /* @tp_richcompare@ */
378   0,                                    /* @tp_weaklistoffset@ */
379   0,                                    /* @tp_iter@ */
380   0,                                    /* @tp_iternext@ */
381   0,                                    /* @tp_methods@ */
382   0,                                    /* @tp_members@ */
383   PYGETSET(fginfo),                     /* @tp_getset@ */
384   0,                                    /* @tp_base@ */
385   0,                                    /* @tp_dict@ */
386   0,                                    /* @tp_descr_get@ */
387   0,                                    /* @tp_descr_set@ */
388   0,                                    /* @tp_dictoffset@ */
389   0,                                    /* @tp_init@ */
390   PyType_GenericAlloc,                  /* @tp_alloc@ */
391   abstract_pynew,                       /* @tp_new@ */
392   0,                                    /* @tp_free@ */
393   0                                     /* @tp_is_gc@ */
394 };
395
396 static PyTypeObject dhinfo_pytype_skel = {
397   PyObject_HEAD_INIT(0) 0,              /* Header */
398   "DHInfo",                             /* @tp_name@ */
399   sizeof(fginfo_pyobj),                 /* @tp_basicsize@ */
400   0,                                    /* @tp_itemsize@ */
401
402   fginfo_pydealloc,                     /* @tp_dealloc@ */
403   0,                                    /* @tp_print@ */
404   0,                                    /* @tp_getattr@ */
405   0,                                    /* @tp_setattr@ */
406   0,                                    /* @tp_compare@ */
407   0,                                    /* @tp_repr@ */
408   0,                                    /* @tp_as_number@ */
409   0,                                    /* @tp_as_sequence@ */
410   0,                                    /* @tp_as_mapping@ */
411   0,                                    /* @tp_hash@ */
412   0,                                    /* @tp_call@ */
413   0,                                    /* @tp_str@ */
414   0,                                    /* @tp_getattro@ */
415   0,                                    /* @tp_setattro@ */
416   0,                                    /* @tp_as_buffer@ */
417   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
418     Py_TPFLAGS_BASETYPE,
419
420   /* @tp_doc@ */
421   "DHInfo(P, R, G): standard (integer) Diffie-Hellman group information.",
422
423   0,                                    /* @tp_traverse@ */
424   0,                                    /* @tp_clear@ */
425   0,                                    /* @tp_richcompare@ */
426   0,                                    /* @tp_weaklistoffset@ */
427   0,                                    /* @tp_iter@ */
428   0,                                    /* @tp_iternext@ */
429   PYMETHODS(dhinfo),                    /* @tp_methods@ */
430   0,                                    /* @tp_members@ */
431   PYGETSET(dhinfo),                     /* @tp_getset@ */
432   0,                                    /* @tp_base@ */
433   0,                                    /* @tp_dict@ */
434   0,                                    /* @tp_descr_get@ */
435   0,                                    /* @tp_descr_set@ */
436   0,                                    /* @tp_dictoffset@ */
437   0,                                    /* @tp_init@ */
438   PyType_GenericAlloc,                  /* @tp_alloc@ */
439   fginfo_pynew,                         /* @tp_new@ */
440   0,                                    /* @tp_free@ */
441   0                                     /* @tp_is_gc@ */
442 };
443
444 static PyTypeObject bindhinfo_pytype_skel = {
445   PyObject_HEAD_INIT(0) 0,              /* Header */
446   "BinDHInfo",                          /* @tp_name@ */
447   sizeof(fginfo_pyobj),                 /* @tp_basicsize@ */
448   0,                                    /* @tp_itemsize@ */
449
450   fginfo_pydealloc,                     /* @tp_dealloc@ */
451   0,                                    /* @tp_print@ */
452   0,                                    /* @tp_getattr@ */
453   0,                                    /* @tp_setattr@ */
454   0,                                    /* @tp_compare@ */
455   0,                                    /* @tp_repr@ */
456   0,                                    /* @tp_as_number@ */
457   0,                                    /* @tp_as_sequence@ */
458   0,                                    /* @tp_as_mapping@ */
459   0,                                    /* @tp_hash@ */
460   0,                                    /* @tp_call@ */
461   0,                                    /* @tp_str@ */
462   0,                                    /* @tp_getattro@ */
463   0,                                    /* @tp_setattro@ */
464   0,                                    /* @tp_as_buffer@ */
465   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
466     Py_TPFLAGS_BASETYPE,
467
468   /* @tp_doc@ */
469   "BinDHInfo(P, R, G): binary-field Diffie-Hellman group information.",
470
471   0,                                    /* @tp_traverse@ */
472   0,                                    /* @tp_clear@ */
473   0,                                    /* @tp_richcompare@ */
474   0,                                    /* @tp_weaklistoffset@ */
475   0,                                    /* @tp_iter@ */
476   0,                                    /* @tp_iternext@ */
477   PYMETHODS(bindhinfo),                 /* @tp_methods@ */
478   0,                                    /* @tp_members@ */
479   PYGETSET(bindhinfo),                  /* @tp_getset@ */
480   0,                                    /* @tp_base@ */
481   0,                                    /* @tp_dict@ */
482   0,                                    /* @tp_descr_get@ */
483   0,                                    /* @tp_descr_set@ */
484   0,                                    /* @tp_dictoffset@ */
485   0,                                    /* @tp_init@ */
486   PyType_GenericAlloc,                  /* @tp_alloc@ */
487   fginfo_pynew,                         /* @tp_new@ */
488   0,                                    /* @tp_free@ */
489   0                                     /* @tp_is_gc@ */
490 };
491
492 /*----- General utilities -------------------------------------------------*/
493
494 PyTypeObject *ge_pytype, *group_pytype;
495 static PyTypeObject *primegroup_pytype, *bingroup_pytype, *ecgroup_pytype;
496
497 group *group_copy(group *g)
498 {
499   if (STRCMP(G_NAME(g), ==, "prime")) {
500     gctx_prime *gc = (gctx_prime *)g;
501     gprime_param gp;
502     gp.g = G_TOINT(g, MP_NEW, g->g);
503     gp.p = gc->mm.m;
504     gp.q = gc->g.r;
505     g = group_prime(&gp);
506     MP_DROP(gp.g);
507   } else if (STRCMP(G_NAME(g), ==, "bin")) {
508     gctx_bin *gc = (gctx_bin *)g;
509     gbin_param gb;
510     gb.g = G_TOINT(g, MP_NEW, g->g);
511     gb.p = gc->r.p;
512     gb.q = gc->g.r;
513     g = group_binary(&gb);
514     MP_DROP(gb.g);
515   } else if (STRCMP(G_NAME(g), ==, "ec")) {
516     gctx_ec *gc = (gctx_ec *)g;
517     ec_info ei;
518     if ((ei.c = eccurve_copy(gc->ei.c)) == 0)
519       return (0);
520     EC_CREATE(&ei.g);
521     EC_COPY(&ei.g, &gc->ei.g);
522     ei.r = MP_COPY(gc->ei.r);
523     ei.h = MP_COPY(gc->ei.h);
524     g = group_ec(&ei);
525   } else
526     g = 0;
527   return (g);
528 }
529
530 PyObject *ge_pywrap(PyObject *gobj, ge *x)
531 {
532   ge_pyobj *z = PyObject_New(ge_pyobj, (PyTypeObject *)gobj);
533   z->x = x;
534   z->g = GROUP_G(gobj);
535   Py_INCREF(gobj);
536   return ((PyObject *)z);
537 }
538
539 static PyObject *ge_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
540 {
541   static const char *const kwlist[] = { "x", 0 };
542   PyObject *x;
543   group *g;
544   ec p = EC_INIT;
545   mp *y = 0;
546   ge *xx = 0;
547   mptext_stringctx sc;
548
549   g = GROUP_G(ty);
550   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O:new", KWLIST, &x)) goto end;
551   xx = G_CREATE(g);
552   if (ECPT_PYCHECK(x)) {
553     getecptout(&p, x);
554     if (G_FROMEC(g, xx, &p))
555       TYERR("can't convert from elliptic curve point");
556     EC_DESTROY(&p);
557   } else if ((y = tomp(x)) != 0) {
558     if (G_FROMINT(g, xx, y))
559       TYERR("can't convert from integer");
560     MP_DROP(y);
561   } else if (PyString_Check(x)) {
562     sc.buf = PyString_AS_STRING(x);
563     sc.lim = sc.buf + PyString_GET_SIZE(x);
564     if (G_READ(g, xx, &mptext_stringops, &sc) || sc.buf < sc.lim)
565       VALERR("malformed group element string");
566   } else
567     TYERR("can't convert to group element");
568   return (ge_pywrap((PyObject *)ty, xx));
569 end:
570   mp_drop(y);
571   EC_DESTROY(&p);
572   if (xx) G_DESTROY(g, xx);
573   return (0);
574 }
575
576 static PyObject *group_dopywrap(PyTypeObject *ty, group *g)
577 {
578   group_pyobj *gobj = newtype(ty, 0, g->ops->name);
579   gobj->g = g;
580   gobj->ty.ht_type.tp_basicsize = sizeof(ge_pyobj);
581   gobj->ty.ht_type.tp_base = ge_pytype;
582   Py_INCREF(group_pytype);
583   gobj->ty.ht_type.tp_flags = (Py_TPFLAGS_DEFAULT |
584                                Py_TPFLAGS_BASETYPE |
585                                Py_TPFLAGS_CHECKTYPES |
586                                Py_TPFLAGS_HEAPTYPE);
587   gobj->ty.ht_type.tp_alloc = PyType_GenericAlloc;
588   gobj->ty.ht_type.tp_free = 0;
589   gobj->ty.ht_type.tp_new = ge_pynew;
590   typeready(&gobj->ty.ht_type);
591   return ((PyObject *)gobj);
592 }
593
594 PyObject *group_pywrap(group *g)
595 {
596   PyTypeObject *ty;
597
598   if (STRCMP(G_NAME(g), ==, "prime")) ty = primegroup_pytype;
599   else if (STRCMP(G_NAME(g), ==, "bin")) ty = bingroup_pytype;
600   else if (STRCMP(G_NAME(g), ==, "ec")) ty = ecgroup_pytype;
601   else abort();
602   return (group_dopywrap(ty, g));
603 }
604
605 /*----- Group elements ----------------------------------------------------*/
606
607 #define BINOP(name)                                                     \
608   static PyObject *ge_py##name(PyObject *x, PyObject *y)                \
609   {                                                                     \
610     ge *z;                                                              \
611     group *g;                                                           \
612     if (!GE_PYCHECK(x) || !GE_PYCHECK(y) ||                             \
613         (GE_G(x) != GE_G(y) && !group_samep(GE_G(x), GE_G(y))))         \
614       RETURN_NOTIMPL;                                                   \
615     g = GE_G(x);                                                        \
616     z = G_CREATE(g);                                                    \
617     g->ops->name(g, z, GE_X(x), GE_X(y));                               \
618     return (ge_pywrap(GE_GOBJ(x), z));                                  \
619   }
620 BINOP(mul)
621 BINOP(div)
622 #undef BINOP
623
624 #define UNOP(name)                                                      \
625   static PyObject *gemeth_##name(PyObject *me)                          \
626   {                                                                     \
627     group *g;                                                           \
628     ge *z;                                                              \
629     g = GE_G(me);                                                       \
630     z = G_CREATE(g);                                                    \
631     g->ops->name(g, z, GE_X(me));                                       \
632     return (ge_pywrap(GE_GOBJ(me), z));                                 \
633   }
634 UNOP(sqr)
635 UNOP(inv)
636 #undef UNOP
637
638 static PyObject *ge_pyexp(PyObject *x, PyObject *n, PyObject *m)
639 {
640   mp *nn;
641   ge *z;
642
643   if (m != Py_None || !GE_PYCHECK(x) || (nn = getmp(n)) == 0)
644     RETURN_NOTIMPL;
645   z = G_CREATE(GE_G(x));
646   G_EXP(GE_G(x), z, GE_X(x), nn);
647   MP_DROP(nn);
648   return (ge_pywrap(GE_GOBJ(x), z));
649 }
650
651 static void ge_pydealloc(PyObject *me)
652 {
653   G_DESTROY(GE_G(me), GE_X(me));
654   Py_DECREF(GE_GOBJ(me));
655   FREEOBJ(me);
656 }
657
658 static void group_pydealloc(PyObject *me)
659 {
660   G_DESTROYGROUP(GROUP_G(me));
661   PyType_Type.tp_dealloc(me);
662 }
663
664 static PyObject *gmexp_id(PyObject *me)
665 {
666   group *g = GROUP_G(me); ge *x = G_CREATE(g);
667   G_COPY(g, x, g->i); return (ge_pywrap(me, x));
668 }
669
670 static int gmexp_fill(void *pp, PyObject  *me, PyObject *x, PyObject *m)
671 {
672   group_expfactor *f = pp;
673
674   if (!GE_PYCHECK(x) || GE_G(x) != GROUP_G(me) || (f->exp = getmp(m)) == 0)
675     return (-1);
676   f->base = GE_X(x);
677   return (0);
678 }
679
680 static PyObject *ge_pyrichcompare(PyObject *x, PyObject *y, int op)
681 {
682   int b;
683   PyObject *rc = 0;
684
685   if (!GE_PYCHECK(x) || !GE_PYCHECK(y) ||
686       (GE_G(x) != GE_G(y) && !group_samep(GE_G(x), GE_G(y))))
687     RETURN_NOTIMPL;
688   switch (op) {
689     case Py_EQ: b = G_EQ(GE_G(x), GE_X(x), GE_X(y)); break;
690     case Py_NE: b = !G_EQ(GE_G(x), GE_X(x), GE_X(y)); break;
691     default: TYERR("group elements are unordered");
692   }
693   rc = getbool(b);
694 end:
695   return (rc);
696 }
697
698 static PyObject *gemeth_check(PyObject *me)
699 {
700   if (group_check(GE_G(me), GE_X(me))) VALERR("bad group element");
701   RETURN_OBJ(me);
702 end:
703   return (0);
704 }
705
706 static int ge_pynonzerop(PyObject *x)
707   { return (!G_IDENTP(GE_G(x), GE_X(x))); }
708
709 static PyObject *ge_pystr(PyObject *me)
710 {
711   dstr d = DSTR_INIT;
712   PyObject *rc;
713
714   group_writedstr(GE_G(me), GE_X(me), &d);
715   rc = PyString_FromStringAndSize(d.buf, d.len);
716   DDESTROY(&d);
717   return (rc);
718 }
719
720 static PyObject *ge_pylong(PyObject *me)
721 {
722   mp *x = 0;
723   PyObject *rc = 0;
724
725   if ((x = G_TOINT(GE_G(me), MP_NEW, GE_X(me))) == 0)
726     TYERR("can't convert to integer");
727   rc = mp_topylong(x);
728 end:
729   mp_drop(x);
730   return (rc);
731 }
732
733 static PyObject *ge_pyint(PyObject *me)
734 {
735   mp *x = 0;
736   PyObject *rc = 0;
737   long l;
738
739   if ((x = G_TOINT(GE_G(me), MP_NEW, GE_X(me))) == 0)
740     TYERR("can't convert to integer");
741   if (!mp_tolong_checked(x, &l, 0)) rc = PyInt_FromLong(l);
742   else rc = mp_topylong(x);
743 end:
744   mp_drop(x);
745   return (rc);
746 }
747
748 static PyObject *gemeth_toint(PyObject *me)
749 {
750   mp *x;
751
752   if ((x = G_TOINT(GE_G(me), MP_NEW, GE_X(me))) == 0)
753     TYERR("can't convert to integer");
754   return (mp_pywrap(x));
755 end:
756   return (0);
757 }
758
759 static PyObject *gemeth_toec(PyObject *me, PyObject *arg, PyObject *kw)
760 {
761   static const char *const kwlist[] = { "curve", 0 };
762   PyTypeObject *cty = 0;
763   PyObject *rc = 0;
764   group *g;
765   ec_curve *c;
766   ec p = EC_INIT;
767
768   if (!PyArg_ParseTupleAndKeywords(arg, kw, "|O:toec", KWLIST,
769                                    &cty)) goto end;
770   g = GROUP_G(GE_GOBJ(me));
771   if (cty) {
772     if (!PyType_Check(cty) || !PyType_IsSubtype(cty, ecpt_pytype))
773       TYERR("want subtype of catacomb.ECPt");
774     Py_INCREF((PyObject *)cty);
775   } else if (STRCMP(G_NAME(g), ==, "ec")) {
776     c = eccurve_copy(((gctx_ec *)g)->ei.c);
777     cty = (PyTypeObject *)eccurve_pywrap(0, c);
778   } else  {
779     cty = ecpt_pytype;
780     Py_INCREF((PyObject *)cty);
781   }
782   if (G_TOEC(GE_G(me), &p, GE_X(me))) {
783     Py_DECREF((PyObject *)cty);
784     TYERR("can't convert to ec point");
785   }
786   rc = ecpt_pywrapout(cty, &p);
787   Py_DECREF((PyObject *)cty);
788 end:
789   return (rc);
790 }
791
792 static PyObject *gemeth_tobuf(PyObject *me)
793 {
794   buf b;
795   PyObject *rc;
796   size_t n;
797
798   n = GE_G(me)->noctets + 4;
799   rc = bytestring_pywrap(0, n);
800   buf_init(&b, PyString_AS_STRING(rc), n);
801   G_TOBUF(GE_G(me), &b, GE_X(me));
802   assert(BOK(&b));
803   _PyString_Resize(&rc, BLEN(&b));
804   return (rc);
805 }
806
807 static PyObject *gemeth_toraw(PyObject *me)
808 {
809   buf b;
810   PyObject *rc;
811   size_t n;
812
813   n = GE_G(me)->noctets;
814   rc = bytestring_pywrap(0, n);
815   buf_init(&b, PyString_AS_STRING(rc), n);
816   G_TORAW(GE_G(me), &b, GE_X(me));
817   assert(BOK(&b));
818   _PyString_Resize(&rc, BLEN(&b));
819   return (rc);
820 }
821
822 static PyObject *gmexp_exp(PyObject *me, void *pp, int n)
823 {
824   ge *z = G_CREATE(GROUP_G(me));
825   G_MEXP(GROUP_G(me), z, pp, n);
826   return (ge_pywrap(me, z));
827 }
828
829 static void gmexp_drop(void *pp)
830 {
831   group_expfactor *f = pp;
832   MP_DROP(f->exp);
833 }
834
835 static PyObject *gmeth_mexp(PyObject *me, PyObject *arg)
836 {
837   return (mexp_common(me, arg, sizeof(group_expfactor),
838                       gmexp_id, gmexp_fill, gmexp_exp, gmexp_drop));
839 }
840
841 static PyObject *gmeth_checkgroup(PyObject *me, PyObject *arg, PyObject *kw)
842 {
843   static const char *const kwlist[] = { "rng", 0 };
844   grand *r = &rand_global;
845   const char *p;
846
847   if (!PyArg_ParseTupleAndKeywords(arg, kw, "|O&:checkgroup", KWLIST,
848                                    convgrand, &r))
849     goto end;
850   if ((p = G_CHECK(GROUP_G(me), r)) != 0)
851     VALERR(p);
852   RETURN_OBJ(me);
853 end:
854   return (0);
855 }
856
857 static PyObject *group_pyrichcompare(PyObject *x, PyObject *y, int op)
858 {
859   int b = group_samep(GROUP_G(x), GROUP_G(y));
860   switch (op) {
861     case Py_EQ: break;
862     case Py_NE: b = !b;
863     default: TYERR("can't order groups");
864   }
865   return (getbool(b));
866 end:
867   return (0);
868 }
869
870 static PyObject *gemeth_frombuf(PyObject *me, PyObject *arg)
871 {
872   buf b;
873   char *p;
874   Py_ssize_t n;
875   group *g;
876   ge *x = 0;
877
878   if (!PyArg_ParseTuple(arg, "s#:frombuf", &p, &n)) return (0);
879   g = GROUP_G(me);
880   buf_init(&b, p, n);
881   x = G_CREATE(g);
882   if (G_FROMBUF(g, &b, x)) VALERR("invalid data");
883   return (Py_BuildValue("(NN)", ge_pywrap(me, x), bytestring_pywrapbuf(&b)));
884 end:
885   if (x) G_DESTROY(g, x);
886   return (0);
887 }
888
889 static PyObject *gemeth_fromraw(PyObject *me, PyObject *arg)
890 {
891   buf b;
892   char *p;
893   Py_ssize_t n;
894   group *g;
895   ge *x = 0;
896
897   if (!PyArg_ParseTuple(arg, "s#:fromraw", &p, &n)) return (0);
898   g = GROUP_G(me);
899   buf_init(&b, p, n);
900   x = G_CREATE(g);
901   if (G_FROMRAW(g, &b, x)) VALERR("invalid data");
902   return (Py_BuildValue("(NN)", ge_pywrap(me, x), bytestring_pywrapbuf(&b)));
903 end:
904   if (x) G_DESTROY(g, x);
905   return (0);
906 }
907
908 static PyObject *gemeth_fromstring(PyObject *me, PyObject *arg)
909 {
910   mptext_stringctx sc;
911   char *p;
912   Py_ssize_t n;
913   group *g;
914   ge *x = 0;
915
916   if (!PyArg_ParseTuple(arg, "s#:fromstring", &p, &n)) return (0);
917   sc.buf = p;
918   sc.lim = sc.buf + n;
919   g = GROUP_G(me);
920   x = G_CREATE(g);
921   if (G_READ(g, x, &mptext_stringops, &sc))
922     VALERR("bad group element string");
923   return (Py_BuildValue("(Ns#)", ge_pywrap(me, x),
924                         sc.buf, (Py_ssize_t)(sc.lim - sc.buf)));
925 end:
926   if (x) G_DESTROY(g, x);
927   return (0);
928 }
929
930 static PyObject *gmeth_parse(PyObject *me, PyObject *arg)
931 {
932   char *p;
933   qd_parse qd;
934   group *g;
935
936   if (!PyArg_ParseTuple(arg, "s:parse", &p)) goto end;
937   qd.p = p; qd.e = 0;
938   if ((g = group_parse(&qd)) == 0) VALERR(qd.e);
939   return (group_pywrap(g));
940 end:
941   return (0);
942 }
943
944 static PyObject *geget_group(PyObject *me, void *hunoz)
945   { RETURN_OBJ(GE_GOBJ(me)); }
946
947 static PyObject *gget_nbits(PyObject *me, void *hunoz)
948   { return (PyInt_FromLong(GROUP_G(me)->nbits)); }
949
950 static PyObject *gget_noctets(PyObject *me, void *hunoz)
951   { return (PyInt_FromLong(GROUP_G(me)->noctets)); }
952
953 static PyObject *gget_i(PyObject *me, void *hunoz)
954 {
955   group *g = GROUP_G(me); ge *x = G_CREATE(g);
956   G_COPY(g, x, g->i); return (ge_pywrap(me, x));
957 }
958
959 static PyObject *gget_g(PyObject *me, void *hunoz)
960 {
961   group *g = GROUP_G(me); ge *x = G_CREATE(g);
962   G_COPY(g, x, g->g); return (ge_pywrap(me, x));
963 }
964
965 static long ge_pyhash(PyObject *me)
966 {
967   buf b;
968   size_t sz = GE_G(me)->noctets + 4;
969   uint32 h = 0xf672c776 + GE_G(me)->ops->ty;
970   octet *p = xmalloc(sz);
971   buf_init(&b, p, sz);
972   G_TOBUF(GE_G(me), &b, GE_X(me));
973   assert(BOK(&b));
974   h = unihash_hash(&unihash_global, h, BBASE(&b), BLEN(&b));
975   xfree(p);
976   return (h % LONG_MAX);
977 }
978
979 static PyObject *gget_r(PyObject *me, void *hunoz)
980   { return (mp_pywrap(MP_COPY(GROUP_G(me)->r))); }
981
982 static PyObject *gget_h(PyObject *me, void *hunoz)
983   { return (mp_pywrap(MP_COPY(GROUP_G(me)->h))); }
984
985 static const PyGetSetDef ge_pygetset[] = {
986 #define GETSETNAME(op, name) ge##op##_##name
987   GET   (group,         "X.group -> group containing X")
988 #undef GETSETNAME
989   { 0 }
990 };
991
992 static const PyMethodDef ge_pymethods[] = {
993 #define METHNAME(name) gemeth_##name
994   NAMETH(inv,           "X.inv() -> inverse element of X")
995   NAMETH(sqr,           "X.sqr() -> X^2 = X * X")
996   NAMETH(check,         "X.check() -> check X really belongs to its group")
997   NAMETH(toint,         "X.toint() -> X converted to an integer")
998   KWMETH(toec,          "X.toec([curve = ECPt]) -> "
999                                        "X converted to elliptic curve point")
1000   NAMETH(tobuf,         "X.tobuf() -> X in buffer representation")
1001   NAMETH(toraw,         "X.toraw() -> X in raw representation")
1002   CMTH  (frombuf,       "frombuf(BUF) -> X, REST")
1003   CMTH  (fromraw,       "fromraw(BUF) -> X, REST")
1004   CMTH  (fromstring,    "fromstring(STR) -> X, REST")
1005 #undef METHNAME
1006   { 0 }
1007 };
1008
1009 static const PyNumberMethods ge_pynumber = {
1010   0,                                    /* @nb_add@ */
1011   0,                                    /* @nb_subtract@ */
1012   ge_pymul,                             /* @nb_multiply@ */
1013   ge_pydiv,                             /* @nb_divide@ */
1014   0,                                    /* @nb_remainder@ */
1015   0,                                    /* @nb_divmod@ */
1016   ge_pyexp,                             /* @nb_power@ */
1017   0,                                    /* @nb_negative@ */
1018   0,                                    /* @nb_positive@ */
1019   0,                                    /* @nb_absolute@ */
1020   ge_pynonzerop,                        /* @nb_nonzero@ */
1021   0,                                    /* @nb_invert@ */
1022   0,                                    /* @nb_lshift@ */
1023   0,                                    /* @nb_rshift@ */
1024   0,                                    /* @nb_and@ */
1025   0,                                    /* @nb_xor@ */
1026   0,                                    /* @nb_or@ */
1027   0,                                    /* @nb_coerce@ */
1028   ge_pyint,                             /* @nb_int@ */
1029   ge_pylong,                            /* @nb_long@ */
1030   0 /* meaningless */,                  /* @nb_float@ */
1031   0,                                    /* @nb_oct@ */
1032   0,                                    /* @nb_hex@ */
1033
1034   0,                                    /* @nb_inplace_add@ */
1035   0,                                    /* @nb_inplace_subtract@ */
1036   0,                                    /* @nb_inplace_multiply@ */
1037   0,                                    /* @nb_inplace_divide@ */
1038   0,                                    /* @nb_inplace_remainder@ */
1039   0,                                    /* @nb_inplace_power@ */
1040   0,                                    /* @nb_inplace_lshift@ */
1041   0,                                    /* @nb_inplace_rshift@ */
1042   0,                                    /* @nb_inplace_and@ */
1043   0,                                    /* @nb_inplace_xor@ */
1044   0,                                    /* @nb_inplace_or@ */
1045
1046   0,                                    /* @nb_floor_divide@ */
1047   ge_pydiv,                             /* @nb_true_divide@ */
1048   0,                                    /* @nb_inplace_floor_divide@ */
1049   0,                                    /* @nb_inplace_true_divide@ */
1050 };
1051
1052 static PyTypeObject ge_pytype_skel = {
1053   PyObject_HEAD_INIT(0) 0,              /* Header */
1054   "GE",                                 /* @tp_name@ */
1055   sizeof(ge_pyobj),                     /* @tp_basicsize@ */
1056   0,                                    /* @tp_itemsize@ */
1057
1058   ge_pydealloc,                         /* @tp_dealloc@ */
1059   0,                                    /* @tp_print@ */
1060   0,                                    /* @tp_getattr@ */
1061   0,                                    /* @tp_setattr@ */
1062   0,                                    /* @tp_compare@ */
1063   0,                                    /* @tp_repr@ */
1064   PYNUMBER(ge),                         /* @tp_as_number@ */
1065   0,                                    /* @tp_as_sequence@ */
1066   0,                                    /* @tp_as_mapping@ */
1067   ge_pyhash,                            /* @tp_hash@ */
1068   0,                                    /* @tp_call@ */
1069   ge_pystr,                             /* @tp_str@ */
1070   0,                                    /* @tp_getattro@ */
1071   0,                                    /* @tp_setattro@ */
1072   0,                                    /* @tp_as_buffer@ */
1073   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
1074     Py_TPFLAGS_CHECKTYPES |
1075     Py_TPFLAGS_BASETYPE,
1076
1077   /* @tp_doc@ */
1078   "Group elements, abstract base class.",
1079
1080   0,                                    /* @tp_traverse@ */
1081   0,                                    /* @tp_clear@ */
1082   ge_pyrichcompare,                     /* @tp_richcompare@ */
1083   0,                                    /* @tp_weaklistoffset@ */
1084   0,                                    /* @tp_iter@ */
1085   0,                                    /* @tp_iternext@ */
1086   PYMETHODS(ge),                        /* @tp_methods@ */
1087   0,                                    /* @tp_members@ */
1088   PYGETSET(ge),                         /* @tp_getset@ */
1089   0,                                    /* @tp_base@ */
1090   0,                                    /* @tp_dict@ */
1091   0,                                    /* @tp_descr_get@ */
1092   0,                                    /* @tp_descr_set@ */
1093   0,                                    /* @tp_dictoffset@ */
1094   0,                                    /* @tp_init@ */
1095   PyType_GenericAlloc,                  /* @tp_alloc@ */
1096   abstract_pynew,                       /* @tp_new@ */
1097   0,                                    /* @tp_free@ */
1098   0                                     /* @tp_is_gc@ */
1099 };
1100
1101 static const PyGetSetDef group_pygetset[] = {
1102 #define GETSETNAME(op, name) g##op##_##name
1103   GET   (noctets,       "G.noctets -> size in octets of element")
1104   GET   (nbits,         "G.nbits -> size in bits of element")
1105   GET   (i,             "G.i -> group identity")
1106   GET   (g,             "G.g -> group generator")
1107   GET   (r,             "G.r -> group order")
1108   GET   (h,             "G.h -> group cofactor")
1109 #undef GETSETNAME
1110   { 0 }
1111 };
1112
1113 static const PyMethodDef group_pymethods[] = {
1114 #define METHNAME(name) gmeth_##name
1115   METH  (mexp,        "G.mexp([(X0, N0), (X1, N1), ...]) -> X0^N0 X1^N1 ...")
1116   KWMETH(checkgroup,    "G.checkgroup([rng = rand]): check group is good")
1117   SMTH  (parse,         "parse(STR) -> G, REST")
1118 #undef METHNAME
1119   { 0 }
1120 };
1121
1122 static PyTypeObject group_pytype_skel = {
1123   PyObject_HEAD_INIT(0) 0,              /* Header */
1124   "Group",                              /* @tp_name@ */
1125   sizeof(group_pyobj),                  /* @tp_basicsize@ */
1126   0,                                    /* @tp_itemsize@ */
1127
1128   group_pydealloc,                      /* @tp_dealloc@ */
1129   0,                                    /* @tp_print@ */
1130   0,                                    /* @tp_getattr@ */
1131   0,                                    /* @tp_setattr@ */
1132   0,                                    /* @tp_compare@ */
1133   0,                                    /* @tp_repr@ */
1134   0,                                    /* @tp_as_number@ */
1135   0,                                    /* @tp_as_sequence@ */
1136   0,                                    /* @tp_as_mapping@ */
1137   0,                                    /* @tp_hash@ */
1138   0,                                    /* @tp_call@ */
1139   0,                                    /* @tp_str@ */
1140   0,                                    /* @tp_getattro@ */
1141   0,                                    /* @tp_setattro@ */
1142   0,                                    /* @tp_as_buffer@ */
1143   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
1144     Py_TPFLAGS_BASETYPE,
1145
1146   /* @tp_doc@ */
1147   "Abstract base class for groups.",
1148
1149   0,                                    /* @tp_traverse@ */
1150   0,                                    /* @tp_clear@ */
1151   group_pyrichcompare,                  /* @tp_richcompare@ */
1152   0,                                    /* @tp_weaklistoffset@ */
1153   0,                                    /* @tp_iter@ */
1154   0,                                    /* @tp_iternext@ */
1155   PYMETHODS(group),                     /* @tp_methods@ */
1156   0,                                    /* @tp_members@ */
1157   PYGETSET(group),                      /* @tp_getset@ */
1158   0,                                    /* @tp_base@ */
1159   0,                                    /* @tp_dict@ */
1160   0,                                    /* @tp_descr_get@ */
1161   0,                                    /* @tp_descr_set@ */
1162   0,                                    /* @tp_dictoffset@ */
1163   0,                                    /* @tp_init@ */
1164   PyType_GenericAlloc,                  /* @tp_alloc@ */
1165   abstract_pynew,                       /* @tp_new@ */
1166   0,                                    /* @tp_free@ */
1167   0                                     /* @tp_is_gc@ */
1168 };
1169
1170 static PyObject *pgget_info(PyObject *me, void *hunoz)
1171 {
1172   gprime_param dp;
1173   gctx_prime *gg = (gctx_prime *)GROUP_G(me);
1174   dp.p = MP_COPY(gg->mm.m);
1175   dp.q = MP_COPY(gg->g.r);
1176   dp.g = mpmont_reduce(&gg->mm, MP_NEW, gg->gen.x);
1177   return (fginfo_pywrap(&dp, dhinfo_pytype));
1178 }
1179
1180 static const PyGetSetDef primegroup_pygetset[] = {
1181 #define GETSETNAME(op, name) pg##op##_##name
1182   GET   (info,          "G.info -> information about the group")
1183 #undef GETSETNAME
1184   { 0 }
1185 };
1186
1187 static PyObject *primegroup_pynew(PyTypeObject *ty,
1188                                   PyObject *arg, PyObject *kw)
1189 {
1190   PyObject *i;
1191   static const char *const kwlist[] = { "info", 0 };
1192
1193   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O!:new", KWLIST,
1194                                    dhinfo_pytype, &i))
1195     return (0);
1196   return (group_dopywrap(ty, group_prime(FGINFO_DP(i))));
1197 }
1198
1199 static PyTypeObject primegroup_pytype_skel = {
1200   PyObject_HEAD_INIT(0) 0,              /* Header */
1201   "PrimeGroup",                         /* @tp_name@ */
1202   sizeof(group_pyobj),                  /* @tp_basicsize@ */
1203   0,                                    /* @tp_itemsize@ */
1204
1205   group_pydealloc,                      /* @tp_dealloc@ */
1206   0,                                    /* @tp_print@ */
1207   0,                                    /* @tp_getattr@ */
1208   0,                                    /* @tp_setattr@ */
1209   0,                                    /* @tp_compare@ */
1210   0,                                    /* @tp_repr@ */
1211   0,                                    /* @tp_as_number@ */
1212   0,                                    /* @tp_as_sequence@ */
1213   0,                                    /* @tp_as_mapping@ */
1214   0,                                    /* @tp_hash@ */
1215   0,                                    /* @tp_call@ */
1216   0,                                    /* @tp_str@ */
1217   0,                                    /* @tp_getattro@ */
1218   0,                                    /* @tp_setattro@ */
1219   0,                                    /* @tp_as_buffer@ */
1220   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
1221     Py_TPFLAGS_BASETYPE,
1222
1223   /* @tp_doc@ */
1224   "PrimeGroup(INFO): subgroups of prime fields.",
1225
1226   0,                                    /* @tp_traverse@ */
1227   0,                                    /* @tp_clear@ */
1228   0,                                    /* @tp_richcompare@ */
1229   0,                                    /* @tp_weaklistoffset@ */
1230   0,                                    /* @tp_iter@ */
1231   0,                                    /* @tp_iternext@ */
1232   0,                                    /* @tp_methods@ */
1233   0,                                    /* @tp_members@ */
1234   PYGETSET(primegroup),                 /* @tp_getset@ */
1235   0,                                    /* @tp_base@ */
1236   0,                                    /* @tp_dict@ */
1237   0,                                    /* @tp_descr_get@ */
1238   0,                                    /* @tp_descr_set@ */
1239   0,                                    /* @tp_dictoffset@ */
1240   0,                                    /* @tp_init@ */
1241   PyType_GenericAlloc,                  /* @tp_alloc@ */
1242   primegroup_pynew,                     /* @tp_new@ */
1243   0,                                    /* @tp_free@ */
1244   0                                     /* @tp_is_gc@ */
1245 };
1246
1247 static PyObject *bgget_info(PyObject *me, void *hunoz)
1248 {
1249   gbin_param dp;
1250   gctx_bin *gg = (gctx_bin *)GROUP_G(me);
1251   dp.p = MP_COPY(gg->r.p);
1252   dp.q = MP_COPY(gg->g.r);
1253   dp.g = MP_COPY(gg->gen.x);
1254   return (fginfo_pywrap(&dp, bindhinfo_pytype));
1255 }
1256
1257 static const PyGetSetDef bingroup_pygetset[] = {
1258 #define GETSETNAME(op, name) bg##op##_##name
1259   GET   (info,          "G.info -> information about the group")
1260 #undef GETSETNAME
1261   { 0 }
1262 };
1263
1264 static PyObject *bingroup_pynew(PyTypeObject *ty,
1265                                 PyObject *arg, PyObject *kw)
1266 {
1267   PyObject *i;
1268   static const char *const kwlist[] = { "info", 0 };
1269
1270   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O!:new", KWLIST,
1271                                    bindhinfo_pytype, &i))
1272     return (0);
1273   return (group_dopywrap(ty, group_binary(FGINFO_DP(i))));
1274 }
1275
1276 static PyTypeObject bingroup_pytype_skel = {
1277   PyObject_HEAD_INIT(0) 0,              /* Header */
1278   "BinGroup",                           /* @tp_name@ */
1279   sizeof(group_pyobj),                  /* @tp_basicsize@ */
1280   0,                                    /* @tp_itemsize@ */
1281
1282   group_pydealloc,                      /* @tp_dealloc@ */
1283   0,                                    /* @tp_print@ */
1284   0,                                    /* @tp_getattr@ */
1285   0,                                    /* @tp_setattr@ */
1286   0,                                    /* @tp_compare@ */
1287   0,                                    /* @tp_repr@ */
1288   0,                                    /* @tp_as_number@ */
1289   0,                                    /* @tp_as_sequence@ */
1290   0,                                    /* @tp_as_mapping@ */
1291   0,                                    /* @tp_hash@ */
1292   0,                                    /* @tp_call@ */
1293   0,                                    /* @tp_str@ */
1294   0,                                    /* @tp_getattro@ */
1295   0,                                    /* @tp_setattro@ */
1296   0,                                    /* @tp_as_buffer@ */
1297   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
1298     Py_TPFLAGS_BASETYPE,
1299
1300   /* @tp_doc@ */
1301   "BinGroup(INFO): subgroups of binary fields.",
1302
1303   0,                                    /* @tp_traverse@ */
1304   0,                                    /* @tp_clear@ */
1305   0,                                    /* @tp_richcompare@ */
1306   0,                                    /* @tp_weaklistoffset@ */
1307   0,                                    /* @tp_iter@ */
1308   0,                                    /* @tp_iternext@ */
1309   0,                                    /* @tp_methods@ */
1310   0,                                    /* @tp_members@ */
1311   PYGETSET(bingroup),                   /* @tp_getset@ */
1312   0,                                    /* @tp_base@ */
1313   0,                                    /* @tp_dict@ */
1314   0,                                    /* @tp_descr_get@ */
1315   0,                                    /* @tp_descr_set@ */
1316   0,                                    /* @tp_dictoffset@ */
1317   0,                                    /* @tp_init@ */
1318   PyType_GenericAlloc,                  /* @tp_alloc@ */
1319   bingroup_pynew,                       /* @tp_new@ */
1320   0,                                    /* @tp_free@ */
1321   0                                     /* @tp_is_gc@ */
1322 };
1323
1324 static PyObject *egget_info(PyObject *me, void *hunoz)
1325 {
1326   ec_info ei;
1327   gctx_ec *gg = (gctx_ec *)GROUP_G(me);
1328
1329   ecinfo_copy(&ei, &gg->ei);
1330   return (ecinfo_pywrap(&ei));
1331 }
1332
1333 static const PyGetSetDef ecgroup_pygetset[] = {
1334 #define GETSETNAME(op, name) eg##op##_##name
1335   GET   (info,          "G.info -> information about the group")
1336 #undef GETSETNAME
1337   { 0 }
1338 };
1339
1340 static PyObject *ecgroup_pynew(PyTypeObject *ty,
1341                                PyObject *arg, PyObject *kw)
1342 {
1343   PyObject *i;
1344   ec_info ei;
1345   static const char *const kwlist[] = { "info", 0 };
1346
1347   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O!:new", KWLIST,
1348                                    ecinfo_pytype, &i))
1349     return (0);
1350   ecinfo_copy(&ei, ECINFO_EI(i));
1351   return (group_dopywrap(ty, group_ec(&ei)));
1352 }
1353
1354 static PyTypeObject ecgroup_pytype_skel = {
1355   PyObject_HEAD_INIT(0) 0,              /* Header */
1356   "ECGroup",                            /* @tp_name@ */
1357   sizeof(group_pyobj),                  /* @tp_basicsize@ */
1358   0,                                    /* @tp_itemsize@ */
1359
1360   group_pydealloc,                      /* @tp_dealloc@ */
1361   0,                                    /* @tp_print@ */
1362   0,                                    /* @tp_getattr@ */
1363   0,                                    /* @tp_setattr@ */
1364   0,                                    /* @tp_compare@ */
1365   0,                                    /* @tp_repr@ */
1366   0,                                    /* @tp_as_number@ */
1367   0,                                    /* @tp_as_sequence@ */
1368   0,                                    /* @tp_as_mapping@ */
1369   0,                                    /* @tp_hash@ */
1370   0,                                    /* @tp_call@ */
1371   0,                                    /* @tp_str@ */
1372   0,                                    /* @tp_getattro@ */
1373   0,                                    /* @tp_setattro@ */
1374   0,                                    /* @tp_as_buffer@ */
1375   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
1376     Py_TPFLAGS_BASETYPE,
1377
1378   /* @tp_doc@ */
1379   "ECGroup(INFO): elliptic curve groups.",
1380
1381   0,                                    /* @tp_traverse@ */
1382   0,                                    /* @tp_clear@ */
1383   0,                                    /* @tp_richcompare@ */
1384   0,                                    /* @tp_weaklistoffset@ */
1385   0,                                    /* @tp_iter@ */
1386   0,                                    /* @tp_iternext@ */
1387   0,                                    /* @tp_methods@ */
1388   0,                                    /* @tp_members@ */
1389   PYGETSET(ecgroup),                    /* @tp_getset@ */
1390   0,                                    /* @tp_base@ */
1391   0,                                    /* @tp_dict@ */
1392   0,                                    /* @tp_descr_get@ */
1393   0,                                    /* @tp_descr_set@ */
1394   0,                                    /* @tp_dictoffset@ */
1395   0,                                    /* @tp_init@ */
1396   PyType_GenericAlloc,                  /* @tp_alloc@ */
1397   ecgroup_pynew,                        /* @tp_new@ */
1398   0,                                    /* @tp_free@ */
1399   0                                     /* @tp_is_gc@ */
1400 };
1401
1402 /*----- Global stuff ------------------------------------------------------*/
1403
1404 void group_pyinit(void)
1405 {
1406   INITTYPE(fginfo, root);
1407   INITTYPE(dhinfo, fginfo);
1408   INITTYPE(bindhinfo, dhinfo);
1409   INITTYPE(ge, root);
1410   INITTYPE(group, type);
1411   INITTYPE(primegroup, group);
1412   INITTYPE(bingroup, group);
1413   INITTYPE(ecgroup, group);
1414 }
1415
1416 void group_pyinsert(PyObject *mod)
1417 {
1418   INSERT("FGInfo", fginfo_pytype);
1419   INSERT("DHInfo", dhinfo_pytype);
1420   INSERT("BinDHInfo", bindhinfo_pytype);
1421   INSERT("GE", ge_pytype);
1422   INSERT("Group", group_pytype);
1423   INSERT("PrimeGroup", primegroup_pytype);
1424   INSERT("BinGroup", bingroup_pytype);
1425   INSERT("ECGroup", ecgroup_pytype);
1426   INSERT("_pgroups", namedgroups(ptab, &npgroups));
1427   INSERT("_bingroups", namedgroups(bintab, &nbingroups));
1428 }
1429
1430 /*----- That's all, folks -------------------------------------------------*/