[pyke] / catacomb-python.h

/* -*-c-*-
 *
 * $Id$
 *
 * Definitions for Catacomb bindings
 *
 * (c) 2004 Straylight/Edgeware
 */

/*----- Licensing notice --------------------------------------------------* 
 *
 * This file is part of the Python interface to Catacomb.
 *
 * Catacomb/Python is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * Catacomb/Python is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Catacomb/Python; if not, write to the Free Software Foundation,
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#ifndef CATACOMB_PYTHON_H
#define CATACOMB_PYTHON_H

#ifdef __cplusplus
  extern "C" {
#endif

/*----- Header files ------------------------------------------------------*/

#include <Python.h>
#include <longintrepr.h>
#include <structmember.h>

#undef ULLONG_MAX
#undef ULONG_LONG_MAX

#include <mLib/darray.h>
#include <mLib/dstr.h>
#include <mLib/macros.h>
#include <mLib/quis.h>

#include <catacomb/buf.h>

#include <catacomb/grand.h>
#include <catacomb/rand.h>
#include <catacomb/noise.h>
#include <catacomb/bbs.h>
#include <catacomb/mprand.h>
#include <catacomb/lcrand.h>
#include <catacomb/fibrand.h>
#include <catacomb/dsarand.h>
#include <catacomb/sslprf.h>
#include <catacomb/tlsprf.h>
#include <catacomb/blkc.h>

#include <catacomb/gcipher.h>
#include <catacomb/ghash.h>
#include <catacomb/gmac.h>
#include <catacomb/md5.h>
#include <catacomb/md5-hmac.h>
#include <catacomb/sha.h>
#include <catacomb/sha-mgf.h>
#include <catacomb/sha-hmac.h>

#include <catacomb/mp.h>
#include <catacomb/mpint.h>
#include <catacomb/mpmul.h>
#include <catacomb/mpcrt.h>
#include <catacomb/mpmont.h>
#include <catacomb/mpbarrett.h>
#include <catacomb/mpreduce.h>

#include <catacomb/pgen.h>
#include <catacomb/pfilt.h>
#include <catacomb/strongprime.h>
#include <catacomb/limlee.h>
#include <catacomb/dh.h>
#include <catacomb/ptab.h>
#include <catacomb/bintab.h>
#include <catacomb/dsa.h>

#include <catacomb/gf.h>
#include <catacomb/gfreduce.h>
#include <catacomb/gfn.h>

#include <catacomb/field.h>
#include <catacomb/field-guts.h>

#include <catacomb/ec.h>
#include <catacomb/ec-raw.h>
#include <catacomb/ectab.h>

#include <catacomb/group.h>
#include <catacomb/group-guts.h>

#include <catacomb/gdsa.h>
#include <catacomb/gkcdsa.h>
#include <catacomb/rsa.h>

#include <catacomb/key.h>
#include <catacomb/passphrase.h>
#include <catacomb/pixie.h>

#include <catacomb/share.h>
#include <catacomb/gfshare.h>

/*----- Utility macros ----------------------------------------------------*/

#define RETURN_OBJ(obj) do { Py_INCREF(obj); return (obj); } while (0)
#define RETURN_NONE RETURN_OBJ(Py_None)
#define RETURN_NOTIMPL RETURN_OBJ(Py_NotImplemented)
#define RETURN_TRUE RETURN_OBJ(Py_True)
#define RETURN_FALSE RETURN_OBJ(Py_False)
#define RETURN_ME RETURN_OBJ(me)

#define EXCERR(exc, str) do {						\
  PyErr_SetString(exc, str);						\
  goto end;								\
} while (0)
#define VALERR(str) EXCERR(PyExc_ValueError, str)
#define TYERR(str) EXCERR(PyExc_TypeError, str)
#define ZDIVERR(str) EXCERR(PyExc_ZeroDivisionError, str)
#define SYNERR(str) EXCERR(PyExc_SyntaxError, str)
#define SYSERR(str) EXCERR(PyExc_SystemError, str)
#define INDEXERR(idx) do {						\
  PyErr_SetObject(PyExc_KeyError, idx);					\
  goto end;								\
} while (0)
#define OSERR(name) do {						\
  PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);			\
  goto end;								\
} while (0)
#define PGENERR do { pgenerr(); goto end; } while (0)

#define CONVFUNC(ty, cty, ext)						\
  int conv##ty(PyObject *o, void *p)					\
  {									\
    if (!PyObject_TypeCheck(o, ty##_pytype))				\
      TYERR("wanted a " #ty);						\
    *(cty *)p = ext(o);							\
    return (1);								\
  end:									\
    return (0);								\
  }

#define root_pytype 0
#define type_pytype &PyType_Type
#define INITTYPE(ty, base) do {						\
  ty##_pytype_skel.tp_base = base##_pytype;				\
  ty##_pytype = inittype(&ty##_pytype_skel);				\
} while (0)

#define INSERT(name, ob) do {						\
  PyObject *_o = (PyObject *)(ob);					\
  Py_INCREF(_o); 							\
  PyModule_AddObject(mod, name, _o);					\
} while (0)

#define INSEXC(name, var, base, meth)					\
  INSERT(name, var = mkexc(mod, base, name, meth))

#define METH(func, doc)							\
  { #func, METHNAME(func), METH_VARARGS, doc },
#define KWMETH(func, doc)						\
  { #func, (PyCFunction)METHNAME(func),					\
    METH_VARARGS | METH_KEYWORDS, doc },

#define GET(func, doc)							\
  { #func, GETSETNAME(get, func), 0, doc },
#define GETSET(func, doc)						\
  { #func, GETSETNAME(get, func), GETSETNAME(set, func), doc },

#define MEMBER(name, ty, f, doc)					\
  { #name, ty, offsetof(MEMBERSTRUCT, name), f, doc },

#define MODULES(_)							\
  _(bytestring) _(buffer)						\
  _(rand) _(algorithms) _(pubkey) _(pgen)				\
  _(mp) _(field) _(ec) _(group)						\
  _(passphrase) _(share) _(key)
#define DOMODINIT(m) m##_pyinit();
#define DOMODINSERT(m) m##_pyinsert(mod);
#define INIT_MODULES do { MODULES(DOMODINIT) } while (0)
#define INSERT_MODULES do { MODULES(DOMODINSERT) } while (0)

#define DO(m)								\
  extern void m##_pyinit(void);						\
  extern void m##_pyinsert(PyObject *);
MODULES(DO)
#undef DO

#define FREEOBJ(obj)							\
  (((PyObject *)(obj))->ob_type->tp_free((PyObject *)(obj)))

/*----- Bytestrings -------------------------------------------------------*/

PyTypeObject *bytestring_pyobj;
PyObject *bytestring_pywrap(const void *, size_t);
PyObject *bytestring_pywrapbuf(buf *);

/*----- Multiprecision arithmetic -----------------------------------------*/

typedef struct mp_pyobj {
  PyObject_HEAD
  mp *x;
} mp_pyobj;

extern PyTypeObject *mp_pytype;
extern PyTypeObject *gf_pytype;
#define MP_X(o) (((mp_pyobj *)(o))->x)
#define MP_PYCHECK(o) PyObject_TypeCheck((o), mp_pytype)
#define GF_PYCHECK(o) PyObject_TypeCheck((o), gf_pytype)

extern mp *mp_frompylong(PyLongObject *);
extern PyLongObject *mp_topylong(mp *);
extern mp *tomp(PyObject *);
extern mp *getmp(PyObject *);
extern int convmp(PyObject *, void *);
extern mp *getgf(PyObject *);
extern int convgf(PyObject *, void *);
extern PyObject *mp_pywrap(mp *);
extern PyObject *gf_pywrap(mp *);
extern mp *mp_frompyobject(PyObject *, int);
extern PyObject *mp_topystring(mp *, int,
			       const char *, const char *, const char *);
extern int mp_tolong_checked(mp *, long *);

/*----- Abstract fields ---------------------------------------------------*/

typedef struct field_pyobj {
  PyHeapTypeObject ty;
  field *f;
} field_pyobj;

extern PyTypeObject *fe_pytype;
#define FE_PYCHECK(o) PyObject_TypeCheck((o), fe_pytype)
#define FE_F(o) (((fe_pyobj *)(o))->f)
#define FE_FOBJ(o) ((PyObject *)(o)->ob_type)
#define FE_X(o) (((fe_pyobj *)(o))->x)
extern PyObject *fe_pywrap(PyObject *, mp *);
extern mp *getfe(field *, PyObject *);

typedef struct fe_pyobj {
  PyObject_HEAD
  field *f;
  mp *x;
} fe_pyobj;
      
extern PyTypeObject *field_pytype;
extern PyTypeObject *primefield_pytype;
extern PyTypeObject *niceprimefield_pytype;
extern PyTypeObject *binfield_pytype;
extern PyTypeObject *binpolyfield_pytype;
extern PyTypeObject *binnormfield_pytype;
#define FIELD_PYCHECK(o) PyObject_TypeCheck((o), field_pytype)
#define FIELD_F(o) (((field_pyobj *)(o))->f)
extern PyObject *field_pywrap(field *);
extern field *field_copy(field *);

/*----- Elliptic curves ---------------------------------------------------*/

typedef struct ecpt_pyobj {
  PyObject_HEAD
  ec_curve *c;
  ec p;
} ecpt_pyobj;

extern PyTypeObject *ecpt_pytype, *ecptcurve_pytype;
#define ECPT_PYCHECK(o) PyObject_TypeCheck((o), ecpt_pytype)
#define ECPTCURVE_PYCHECK(o) PyObject_TypeCheck((o), ecptcurve_pytype)
#define ECPT_C(o) (((ecpt_pyobj *)(o))->c)
#define ECPT_COBJ(o) ((PyObject *)(o)->ob_type)
#define ECPT_FOBJ(o) ECCURVE_FOBJ(ECPT_COBJ((o)))
#define ECPT_P(o) (&((ecpt_pyobj *)(o))->p)
extern PyObject *ecpt_pywrap(PyObject *, ec *);
extern PyObject *ecpt_pywrapout(void *, ec *);
extern int toecpt(ec_curve *, ec *, PyObject *);
extern int getecpt(ec_curve *, ec *, PyObject *);
extern void getecptout(ec *, PyObject *);
extern int convecpt(PyObject *, void *);

typedef struct eccurve_pyobj {
  PyHeapTypeObject ty;
  ec_curve *c;
  PyObject *fobj;
} eccurve_pyobj;

extern PyTypeObject *eccurve_pytype;
extern PyTypeObject *ecprimecurve_pytype;
extern PyTypeObject *ecprimeprojcurve_pytype;
extern PyTypeObject *ecbincurve_pytype;
extern PyTypeObject *ecbinprojcurve_pytype;
#define ECCURVE_PYCHECK(o) PyObject_TypeCheck((o), eccurve_pytype)
#define ECCURVE_C(o) (((eccurve_pyobj *)(o))->c)
#define ECCURVE_FOBJ(o) (((eccurve_pyobj *)(o))->fobj)
extern PyObject *eccurve_pywrap(PyObject *, ec_curve *);
extern ec_curve *eccurve_copy(ec_curve *);

typedef struct ecinfo_pyobj {
  PyObject_HEAD
  ec_info ei;
  PyObject *cobj;
} ecinfo_pyobj;
      
extern PyTypeObject *ecinfo_pytype;
#define ECINFO_PYCHECK(o) PyObject_TypeCheck((o), ecinfo_pytype)
#define ECINFO_EI(o) (&((ecinfo_pyobj *)(o))->ei)
#define ECINFO_COBJ(o) (((ecinfo_pyobj *)(o))->cobj)
extern void ecinfo_copy(ec_info *, const ec_info *);
extern PyObject *ecinfo_pywrap(ec_info *);

/*----- Cyclic groups -----------------------------------------------------*/

typedef struct fginfo_pyobj {
  PyObject_HEAD
  gprime_param dp;
} fginfo_pyobj;

PyTypeObject *fginfo_pytype, *dhinfo_pytype, *bindhinfo_pytype;
#define FGINFO_DP(fg) (&((fginfo_pyobj *)(fg))->dp)
PyObject *fginfo_pywrap(gprime_param *, PyTypeObject *);

typedef struct ge_pyobj {
  PyObject_HEAD
  ge *x;
  group *g;
} ge_pyobj;

extern PyTypeObject *ge_pytype;
#define GE_PYCHECK(o) PyObject_TypeCheck((o), ge_pytype)
#define GE_X(o) (((ge_pyobj *)(o))->x)
#define GE_G(o) (((ge_pyobj *)(o))->g)
#define GE_GOBJ(o) ((PyObject *)(group_pyobj *)(o)->ob_type)
extern PyObject *ge_pywrap(PyObject *, ge *);

typedef struct group_pyobj {
  PyHeapTypeObject ty;
  group *g;
} group_pyobj;

extern PyTypeObject *group_pytype;
extern PyTypeObject *primegroup_pytype, *bingroup_pytype, *ecgroup_pytype;
#define GROUP_G(o) (((group_pyobj *)(o))->g)
extern PyObject *group_pywrap(group *);
extern group *group_copy(group *);

/*----- Random number generators ------------------------------------------*/

#define f_freeme 1u

typedef struct grand_pyobj {
  PyObject_HEAD
  unsigned f;
  grand *r;
} grand_pyobj;

extern PyTypeObject *grand_pytype, *truerand_pytype;
extern PyTypeObject *lcrand_pytype,* fibrand_pytype;
extern PyTypeObject *dsarand_pytype, *bbs_pytype;
extern PyObject *rand_pyobj;
#define GRAND_PYCHECK(o) PyObject_TypeCheck((o), grand_pytype)
#define GRAND_F(o) (((grand_pyobj *)(o))->f)
#define GRAND_R(o) (((grand_pyobj *)(o))->r)
extern PyObject *grand_pywrap(grand *, unsigned);
extern int convgrand(PyObject *, void *);

/*----- Key sizes ---------------------------------------------------------*/

typedef struct keysz_pyobj {
  PyObject_HEAD
  int dfl;
} keysz_pyobj;

typedef struct keyszrange_pyobj {
  PyObject_HEAD
  int dfl;
  int min, max, mod;
} keyszrange_pyobj;

typedef struct keyszset_pyobj {
  PyObject_HEAD
  int dfl;
  PyObject *set;
} keyszset_pyobj;

#define KEYSZ_PYCHECK(o) PyObject_TypeCheck((o), keysz_pytype)
extern PyObject *keysz_pywrap(const octet *);

/*----- Symmetric cryptography --------------------------------------------*/

typedef struct gccipher_pyobj {
  PyHeapTypeObject ty;
  gccipher *cc;
} gccipher_pyobj;

extern PyTypeObject *gccipher_pytype;
#define GCCIPHER_PYCHECK(o) PyObject_TypeCheck((o), gccipher_pytype)
#define GCCIPHER_CC(o) (((gccipher_pyobj *)(o))->cc)
#define GCCIPHER_F(o) (((gccipher_pyobj *)(o))->f)
extern PyObject *gccipher_pywrap(gccipher *);
extern int convgccipher(PyObject *, void *);
extern int convgcipher(PyObject *, void *);

typedef struct gcipher_pyobj {
  PyObject_HEAD
  unsigned f;
  gcipher *c;
} gcipher_pyobj;

extern PyTypeObject *gcipher_pytype;
#define GCIPHER_PYCHECK(o) PyObject_TypeCheck((o), gcipher_pytype)
#define GCIPHER_C(o) (((gcipher_pyobj *)(o))->c)
#define GCIPHER_F(o) (((gcipher_pyobj *)(o))->f)
extern PyObject *gcipher_pywrap(PyObject *, gcipher *, unsigned);
extern int convgcipher(PyObject *, void *);

typedef struct gchash_pyobj {
  PyHeapTypeObject ty;
  gchash *ch;
} gchash_pyobj;

extern PyTypeObject *gchash_pytype;
#define GCHASH_PYCHECK(o) PyObject_TypeCheck((o), gchash_pytype)
#define GCHASH_CH(o) (((gchash_pyobj *)(o))->ch)
#define GCHASH_F(o) (((gchash_pyobj *)(o))->f)
extern PyObject *gchash_pywrap(gchash *);
extern int convgchash(PyObject *, void *);

typedef struct ghash_pyobj {
  PyObject_HEAD
  unsigned f;
  ghash *h;
} ghash_pyobj;

extern PyTypeObject *ghash_pytype, *gmhash_pytype;
extern PyObject *sha_pyobj, *has160_pyobj;
#define GHASH_PYCHECK(o) PyObject_TypeCheck((o), ghash_pytype)
#define GHASH_H(o) (((ghash_pyobj *)(o))->h)
#define GHASH_F(o) (((ghash_pyobj *)(o))->f)
extern PyObject *ghash_pywrap(PyObject *, ghash *, unsigned);
extern int convghash(PyObject *, void *);
extern int convgmhash(PyObject *, void *);

typedef struct gcmac_pyobj {
  PyHeapTypeObject ty;
  gcmac *cm;
} gcmac_pyobj;

extern PyTypeObject *gcmac_pytype;
#define GCMAC_PYCHECK(o) PyObject_TypeCheck((o), gcmac_pytype)
#define GCMAC_CM(o) (((gcmac_pyobj *)(o))->cm)
#define GCMAC_F(o) (((gcmac_pyobj *)(o))->f)
extern PyObject *gcmac_pywrap(gcmac *);
extern int convgcmac(PyObject *, void *);

typedef struct gmac_pyobj {
  PyHeapTypeObject ty;
  unsigned f;
  gmac *m;
} gmac_pyobj;

extern PyTypeObject *gmac_pytype;
#define GMAC_PYCHECK(o) PyObject_TypeCheck((o), gmac_pytype)
#define GMAC_M(o) (((gmac_pyobj *)(o))->m)
#define GMAC_F(o) (((gmac_pyobj *)(o))->f)
extern PyObject *gmac_pywrap(PyObject *, gmac *, unsigned);
extern int convgmac(PyObject *, void *);

/*----- Key generation ----------------------------------------------------*/
    
typedef struct pfilt_pyobj {
  PyObject_HEAD
  pfilt f;
  int st;
} pfilt_pyobj;

extern PyTypeObject *pfilt_pytype;
#define PFILT_PYCHECK(o) PyObject_TypeCheck(o, pfilt_pytype)
#define PFILT_F(o) (&((pfilt_pyobj *)(o))->f)
#define PFILT_ST(o) (((pfilt_pyobj *)(o))->st)

typedef struct { pgen_proc *proc; void *ctx; } pgev;
#define PGEV_HEAD PyObject_HEAD pgev pg;

typedef struct pgev_pyobj {
  PGEV_HEAD
} pgev_pyobj;

extern PyTypeObject *pgev_pytype;
#define PGEV_PYCHECK(o) PyObject_TypeCheck(o, pgev_pytype)
#define PGEV_PG(o) (&((pgev_pyobj *)(o))->pg)

extern int convpgev(PyObject *, void *);
extern void droppgev(pgev *);
extern void pgenerr(void);

/*----- Core utility functions --------------------------------------------*/

extern PyObject *mexp_common(PyObject *, PyObject *, size_t,
			     PyObject *(*id)(PyObject *),
			     int (*fill)(void *, PyObject *,
					 PyObject *, PyObject *),
			     PyObject *(*exp)(PyObject *, void *, int),
			     void (*drop)(void *));

extern int convulong(PyObject *, void *);
#define DECL_CONVU_(n) extern int convu##n(PyObject *, void *);
DOUINTSZ(DECL_CONVU_)
extern int convmpw(PyObject *, void *);
extern int convuint(PyObject *, void *);
extern int convszt(PyObject *, void *);
extern int convbool(PyObject *, void *);
extern PyObject *abstract_pynew(PyTypeObject *, PyObject *, PyObject *);
extern PyObject *getbool(int);
#define DECL_GETU_(n) extern PyObject *getu##n(uint##n);
DOUINTSZ(DECL_GETU_)
extern PyObject * mkexc(PyObject *, PyObject *, const char *, PyMethodDef *);
extern void *newtype(PyTypeObject *, const PyTypeObject *, const char *);
extern PyTypeObject *inittype(PyTypeObject *);
extern void addmethods(const PyMethodDef *);

/*----- That's all, folks -------------------------------------------------*/

#ifdef __cplusplus
  }
#endif

#endif
Commit	Line	Data
b6a86d2e	1	/* --c--
	2	*
	3	* $Id$
	4	*
	5	* Definitions for Catacomb bindings
	6	*
	7	* (c) 2004 Straylight/Edgeware
	8	*/
	9
	10	/----- Licensing notice --------------------------------------------------
	11	*
	12	* This file is part of the Python interface to Catacomb.
	13	*
	14	* Catacomb/Python is free software; you can redistribute it and/or modify
	15	* it under the terms of the GNU General Public License as published by
	16	* the Free Software Foundation; either version 2 of the License, or
	17	* (at your option) any later version.
	18	*
	19	* Catacomb/Python is distributed in the hope that it will be useful,
	20	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	21	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	22	* GNU General Public License for more details.
	23	*
	24	* You should have received a copy of the GNU General Public License
	25	* along with Catacomb/Python; if not, write to the Free Software Foundation,
	26	* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	27	*/
	28
	29	#ifndef CATACOMB_PYTHON_H
	30	#define CATACOMB_PYTHON_H
	31
	32	#ifdef __cplusplus
	33	extern "C" {
	34	#endif
	35
	36	/----- Header files ------------------------------------------------------/
	37
	38	#include <Python.h>
	39	#include <longintrepr.h>
	40	#include <structmember.h>
	41
73a712fa	42	#undef ULLONG_MAX
	43	#undef ULONG_LONG_MAX
	44
b6a86d2e	45	#include <mLib/darray.h>
	46	#include <mLib/dstr.h>
	47	#include <mLib/macros.h>
73a712fa	48	#include <mLib/quis.h>
b6a86d2e	49
	50	#include <catacomb/buf.h>
	51
	52	#include <catacomb/grand.h>
	53	#include <catacomb/rand.h>
	54	#include <catacomb/noise.h>
	55	#include <catacomb/bbs.h>
	56	#include <catacomb/mprand.h>
	57	#include <catacomb/lcrand.h>
	58	#include <catacomb/fibrand.h>
	59	#include <catacomb/dsarand.h>
	60	#include <catacomb/sslprf.h>
	61	#include <catacomb/tlsprf.h>
62058916	62	#include <catacomb/blkc.h>
b6a86d2e	63
	64	#include <catacomb/gcipher.h>
	65	#include <catacomb/ghash.h>
	66	#include <catacomb/gmac.h>
	67	#include <catacomb/md5.h>
	68	#include <catacomb/md5-hmac.h>
	69	#include <catacomb/sha.h>
	70	#include <catacomb/sha-mgf.h>
	71	#include <catacomb/sha-hmac.h>
	72
	73	#include <catacomb/mp.h>
	74	#include <catacomb/mpint.h>
	75	#include <catacomb/mpmul.h>
	76	#include <catacomb/mpcrt.h>
	77	#include <catacomb/mpmont.h>
	78	#include <catacomb/mpbarrett.h>
	79	#include <catacomb/mpreduce.h>
	80
	81	#include <catacomb/pgen.h>
	82	#include <catacomb/pfilt.h>
	83	#include <catacomb/strongprime.h>
	84	#include <catacomb/limlee.h>
	85	#include <catacomb/dh.h>
	86	#include <catacomb/ptab.h>
	87	#include <catacomb/bintab.h>
	88	#include <catacomb/dsa.h>
	89
	90	#include <catacomb/gf.h>
	91	#include <catacomb/gfreduce.h>
	92	#include <catacomb/gfn.h>
	93
	94	#include <catacomb/field.h>
	95	#include <catacomb/field-guts.h>
	96
	97	#include <catacomb/ec.h>
	98	#include <catacomb/ec-raw.h>
	99	#include <catacomb/ectab.h>
	100
	101	#include <catacomb/group.h>
	102	#include <catacomb/group-guts.h>
	103
	104	#include <catacomb/gdsa.h>
	105	#include <catacomb/gkcdsa.h>
	106	#include <catacomb/rsa.h>
	107
	108	#include <catacomb/key.h>
	109	#include <catacomb/passphrase.h>
	110	#include <catacomb/pixie.h>
	111
73a712fa	112	#include <catacomb/share.h>
	113	#include <catacomb/gfshare.h>
	114
b6a86d2e	115	/----- Utility macros ----------------------------------------------------/
	116
	117	#define RETURN_OBJ(obj) do { Py_INCREF(obj); return (obj); } while (0)
	118	#define RETURN_NONE RETURN_OBJ(Py_None)
	119	#define RETURN_NOTIMPL RETURN_OBJ(Py_NotImplemented)
	120	#define RETURN_TRUE RETURN_OBJ(Py_True)
	121	#define RETURN_FALSE RETURN_OBJ(Py_False)
	122	#define RETURN_ME RETURN_OBJ(me)
	123
	124	#define EXCERR(exc, str) do { \
	125	PyErr_SetString(exc, str); \
	126	goto end; \
	127	} while (0)
	128	#define VALERR(str) EXCERR(PyExc_ValueError, str)
	129	#define TYERR(str) EXCERR(PyExc_TypeError, str)
	130	#define ZDIVERR(str) EXCERR(PyExc_ZeroDivisionError, str)
	131	#define SYNERR(str) EXCERR(PyExc_SyntaxError, str)
	132	#define SYSERR(str) EXCERR(PyExc_SystemError, str)
73a712fa	133	#define INDEXERR(idx) do { \
	134	PyErr_SetObject(PyExc_KeyError, idx); \
	135	goto end; \
	136	} while (0)
b6a86d2e	137	#define OSERR(name) do { \
	138	PyErr_SetFromErrnoWithFilename(PyExc_OSError, name); \
	139	goto end; \
	140	} while (0)
	141	#define PGENERR do { pgenerr(); goto end; } while (0)
	142
	143	#define CONVFUNC(ty, cty, ext) \
	144	int conv##ty(PyObject o, void p) \
	145	{ \
	146	if (!PyObject_TypeCheck(o, ty##_pytype)) \
	147	TYERR("wanted a " #ty); \
	148	(cty )p = ext(o); \
	149	return (1); \
	150	end: \
	151	return (0); \
	152	}
	153
	154	#define root_pytype 0
	155	#define type_pytype &PyType_Type
	156	#define INITTYPE(ty, base) do { \
	157	ty##_pytype_skel.tp_base = base##_pytype; \
	158	ty##_pytype = inittype(&ty##_pytype_skel); \
	159	} while (0)
	160
	161	#define INSERT(name, ob) do { \
	162	PyObject _o = (PyObject )(ob); \
	163	Py_INCREF(_o); \
	164	PyModule_AddObject(mod, name, _o); \
	165	} while (0)
	166
73a712fa	167	#define INSEXC(name, var, base, meth) \
	168	INSERT(name, var = mkexc(mod, base, name, meth))
	169
b6a86d2e	170	#define METH(func, doc) \
	171	{ #func, METHNAME(func), METH_VARARGS, doc },
	172	#define KWMETH(func, doc) \
	173	{ #func, (PyCFunction)METHNAME(func), \
	174	METH_VARARGS \| METH_KEYWORDS, doc },
	175
	176	#define GET(func, doc) \
	177	{ #func, GETSETNAME(get, func), 0, doc },
	178	#define GETSET(func, doc) \
	179	{ #func, GETSETNAME(get, func), GETSETNAME(set, func), doc },
	180
	181	#define MEMBER(name, ty, f, doc) \
	182	{ #name, ty, offsetof(MEMBERSTRUCT, name), f, doc },
	183
73a712fa	184	#define MODULES(_) \
	185	_(bytestring) _(buffer) \
	186	_(rand) _(algorithms) _(pubkey) _(pgen) \
	187	_(mp) _(field) _(ec) _(group) \
	188	_(passphrase) _(share) _(key)
b6a86d2e	189	#define DOMODINIT(m) m##_pyinit();
	190	#define DOMODINSERT(m) m##_pyinsert(mod);
	191	#define INIT_MODULES do { MODULES(DOMODINIT) } while (0)
	192	#define INSERT_MODULES do { MODULES(DOMODINSERT) } while (0)
	193
	194	#define DO(m) \
	195	extern void m##_pyinit(void); \
	196	extern void m##_pyinsert(PyObject *);
	197	MODULES(DO)
	198	#undef DO
	199
ba45a729	200	#define FREEOBJ(obj) \
	201	(((PyObject )(obj))->ob_type->tp_free((PyObject )(obj)))
	202
b6a86d2e	203	/----- Bytestrings -------------------------------------------------------/
	204
	205	PyTypeObject *bytestring_pyobj;
	206	PyObject bytestring_pywrap(const void , size_t);
	207	PyObject bytestring_pywrapbuf(buf );
	208
	209	/----- Multiprecision arithmetic -----------------------------------------/
	210
	211	typedef struct mp_pyobj {
	212	PyObject_HEAD
	213	mp *x;
	214	} mp_pyobj;
	215
	216	extern PyTypeObject *mp_pytype;
	217	extern PyTypeObject *gf_pytype;
	218	#define MP_X(o) (((mp_pyobj *)(o))->x)
	219	#define MP_PYCHECK(o) PyObject_TypeCheck((o), mp_pytype)
	220	#define GF_PYCHECK(o) PyObject_TypeCheck((o), gf_pytype)
	221
	222	extern mp mp_frompylong(PyLongObject );
	223	extern PyLongObject mp_topylong(mp );
	224	extern mp tomp(PyObject );
	225	extern mp getmp(PyObject );
	226	extern int convmp(PyObject , void );
	227	extern mp getgf(PyObject );
	228	extern int convgf(PyObject , void );
	229	extern PyObject mp_pywrap(mp );
	230	extern PyObject gf_pywrap(mp );
	231	extern mp mp_frompyobject(PyObject , int);
	232	extern PyObject mp_topystring(mp , int,
	233	const char , const char , const char *);
	234	extern int mp_tolong_checked(mp , long );
	235
	236	/----- Abstract fields ---------------------------------------------------/
	237
	238	typedef struct field_pyobj {
f0526039	239	PyHeapTypeObject ty;
b6a86d2e	240	field *f;
	241	} field_pyobj;
	242
	243	extern PyTypeObject *fe_pytype;
	244	#define FE_PYCHECK(o) PyObject_TypeCheck((o), fe_pytype)
	245	#define FE_F(o) (((fe_pyobj *)(o))->f)
	246	#define FE_FOBJ(o) ((PyObject *)(o)->ob_type)
	247	#define FE_X(o) (((fe_pyobj *)(o))->x)
	248	extern PyObject fe_pywrap(PyObject , mp *);
	249	extern mp getfe(field , PyObject *);
	250
	251	typedef struct fe_pyobj {
	252	PyObject_HEAD
	253	field *f;
b6a86d2e	254	mp *x;
	255	} fe_pyobj;
	256
	257	extern PyTypeObject *field_pytype;
	258	extern PyTypeObject *primefield_pytype;
	259	extern PyTypeObject *niceprimefield_pytype;
	260	extern PyTypeObject *binfield_pytype;
	261	extern PyTypeObject *binpolyfield_pytype;
	262	extern PyTypeObject *binnormfield_pytype;
	263	#define FIELD_PYCHECK(o) PyObject_TypeCheck((o), field_pytype)
	264	#define FIELD_F(o) (((field_pyobj *)(o))->f)
	265	extern PyObject field_pywrap(field );
	266	extern field field_copy(field );
	267
	268	/----- Elliptic curves ---------------------------------------------------/
	269
	270	typedef struct ecpt_pyobj {
	271	PyObject_HEAD
	272	ec_curve *c;
	273	ec p;
	274	} ecpt_pyobj;
	275
	276	extern PyTypeObject ecpt_pytype, ecptcurve_pytype;
	277	#define ECPT_PYCHECK(o) PyObject_TypeCheck((o), ecpt_pytype)
	278	#define ECPTCURVE_PYCHECK(o) PyObject_TypeCheck((o), ecptcurve_pytype)
	279	#define ECPT_C(o) (((ecpt_pyobj *)(o))->c)
	280	#define ECPT_COBJ(o) ((PyObject *)(o)->ob_type)
	281	#define ECPT_FOBJ(o) ECCURVE_FOBJ(ECPT_COBJ((o)))
	282	#define ECPT_P(o) (&((ecpt_pyobj *)(o))->p)
	283	extern PyObject ecpt_pywrap(PyObject , ec *);
	284	extern PyObject ecpt_pywrapout(void , ec *);
	285	extern int toecpt(ec_curve , ec , PyObject *);
	286	extern int getecpt(ec_curve , ec , PyObject *);
	287	extern void getecptout(ec , PyObject );
73a712fa	288	extern int convecpt(PyObject , void );
b6a86d2e	289
b6a86d2e	290	typedef struct eccurve_pyobj {
f0526039	291	PyHeapTypeObject ty;
b6a86d2e	292	ec_curve *c;
	293	PyObject *fobj;
	294	} eccurve_pyobj;
	295
	296	extern PyTypeObject *eccurve_pytype;
	297	extern PyTypeObject *ecprimecurve_pytype;
	298	extern PyTypeObject *ecprimeprojcurve_pytype;
	299	extern PyTypeObject *ecbincurve_pytype;
	300	extern PyTypeObject *ecbinprojcurve_pytype;
	301	#define ECCURVE_PYCHECK(o) PyObject_TypeCheck((o), eccurve_pytype)
	302	#define ECCURVE_C(o) (((eccurve_pyobj *)(o))->c)
	303	#define ECCURVE_FOBJ(o) (((eccurve_pyobj *)(o))->fobj)
	304	extern PyObject eccurve_pywrap(PyObject , ec_curve *);
	305	extern ec_curve eccurve_copy(ec_curve );
	306
	307	typedef struct ecinfo_pyobj {
	308	PyObject_HEAD
	309	ec_info ei;
	310	PyObject *cobj;
	311	} ecinfo_pyobj;
	312
	313	extern PyTypeObject *ecinfo_pytype;
	314	#define ECINFO_PYCHECK(o) PyObject_TypeCheck((o), ecinfo_pytype)
	315	#define ECINFO_EI(o) (&((ecinfo_pyobj *)(o))->ei)
	316	#define ECINFO_COBJ(o) (((ecinfo_pyobj *)(o))->cobj)
	317	extern void ecinfo_copy(ec_info , const ec_info );
	318	extern PyObject ecinfo_pywrap(ec_info );
	319
	320	/----- Cyclic groups -----------------------------------------------------/
	321
	322	typedef struct fginfo_pyobj {
	323	PyObject_HEAD
	324	gprime_param dp;
	325	} fginfo_pyobj;
	326
	327	PyTypeObject fginfo_pytype, dhinfo_pytype, *bindhinfo_pytype;
	328	#define FGINFO_DP(fg) (&((fginfo_pyobj *)(fg))->dp)
	329	PyObject fginfo_pywrap(gprime_param , PyTypeObject *);
	330
	331	typedef struct ge_pyobj {
	332	PyObject_HEAD
	333	ge *x;
	334	group *g;
	335	} ge_pyobj;
	336
	337	extern PyTypeObject *ge_pytype;
	338	#define GE_PYCHECK(o) PyObject_TypeCheck((o), ge_pytype)
	339	#define GE_X(o) (((ge_pyobj *)(o))->x)
	340	#define GE_G(o) (((ge_pyobj *)(o))->g)
	341	#define GE_GOBJ(o) ((PyObject )(group_pyobj )(o)->ob_type)
	342	extern PyObject ge_pywrap(PyObject , ge *);
	343
	344	typedef struct group_pyobj {
f0526039	345	PyHeapTypeObject ty;
b6a86d2e	346	group *g;
	347	} group_pyobj;
	348
	349	extern PyTypeObject *group_pytype;
	350	extern PyTypeObject primegroup_pytype, bingroup_pytype, *ecgroup_pytype;
	351	#define GROUP_G(o) (((group_pyobj *)(o))->g)
	352	extern PyObject group_pywrap(group );
	353	extern group group_copy(group );
	354
	355	/----- Random number generators ------------------------------------------/
	356
	357	#define f_freeme 1u
	358
	359	typedef struct grand_pyobj {
	360	PyObject_HEAD
	361	unsigned f;
	362	grand *r;
	363	} grand_pyobj;
	364
	365	extern PyTypeObject grand_pytype, truerand_pytype;
	366	extern PyTypeObject lcrand_pytype, fibrand_pytype;
	367	extern PyTypeObject dsarand_pytype, bbs_pytype;
	368	extern PyObject *rand_pyobj;
	369	#define GRAND_PYCHECK(o) PyObject_TypeCheck((o), grand_pytype)
	370	#define GRAND_F(o) (((grand_pyobj *)(o))->f)
	371	#define GRAND_R(o) (((grand_pyobj *)(o))->r)
	372	extern PyObject grand_pywrap(grand , unsigned);
	373	extern int convgrand(PyObject , void );
	374
	375	/----- Key sizes ---------------------------------------------------------/
	376
	377	typedef struct keysz_pyobj {
	378	PyObject_HEAD
	379	int dfl;
	380	} keysz_pyobj;
	381
	382	typedef struct keyszrange_pyobj {
	383	PyObject_HEAD
	384	int dfl;
	385	int min, max, mod;
	386	} keyszrange_pyobj;
	387
	388	typedef struct keyszset_pyobj {
	389	PyObject_HEAD
	390	int dfl;
	391	PyObject *set;
	392	} keyszset_pyobj;
	393
	394	#define KEYSZ_PYCHECK(o) PyObject_TypeCheck((o), keysz_pytype)
	395	extern PyObject keysz_pywrap(const octet );
	396
	397	/----- Symmetric cryptography --------------------------------------------/
	398
	399	typedef struct gccipher_pyobj {
f0526039	400	PyHeapTypeObject ty;
b6a86d2e	401	gccipher *cc;
	402	} gccipher_pyobj;
	403
	404	extern PyTypeObject *gccipher_pytype;
	405	#define GCCIPHER_PYCHECK(o) PyObject_TypeCheck((o), gccipher_pytype)
	406	#define GCCIPHER_CC(o) (((gccipher_pyobj *)(o))->cc)
	407	#define GCCIPHER_F(o) (((gccipher_pyobj *)(o))->f)
	408	extern PyObject gccipher_pywrap(gccipher );
	409	extern int convgccipher(PyObject , void );
	410	extern int convgcipher(PyObject , void );
	411
	412	typedef struct gcipher_pyobj {
	413	PyObject_HEAD
	414	unsigned f;
	415	gcipher *c;
	416	} gcipher_pyobj;
	417
	418	extern PyTypeObject *gcipher_pytype;
	419	#define GCIPHER_PYCHECK(o) PyObject_TypeCheck((o), gcipher_pytype)
	420	#define GCIPHER_C(o) (((gcipher_pyobj *)(o))->c)
	421	#define GCIPHER_F(o) (((gcipher_pyobj *)(o))->f)
	422	extern PyObject gcipher_pywrap(PyObject , gcipher *, unsigned);
	423	extern int convgcipher(PyObject , void );
	424
	425	typedef struct gchash_pyobj {
f0526039	426	PyHeapTypeObject ty;
b6a86d2e	427	gchash *ch;
	428	} gchash_pyobj;
	429
	430	extern PyTypeObject *gchash_pytype;
	431	#define GCHASH_PYCHECK(o) PyObject_TypeCheck((o), gchash_pytype)
	432	#define GCHASH_CH(o) (((gchash_pyobj *)(o))->ch)
	433	#define GCHASH_F(o) (((gchash_pyobj *)(o))->f)
	434	extern PyObject gchash_pywrap(gchash );
	435	extern int convgchash(PyObject , void );
	436
	437	typedef struct ghash_pyobj {
	438	PyObject_HEAD
	439	unsigned f;
	440	ghash *h;
	441	} ghash_pyobj;
	442
	443	extern PyTypeObject ghash_pytype, gmhash_pytype;
	444	extern PyObject sha_pyobj, has160_pyobj;
	445	#define GHASH_PYCHECK(o) PyObject_TypeCheck((o), ghash_pytype)
	446	#define GHASH_H(o) (((ghash_pyobj *)(o))->h)
	447	#define GHASH_F(o) (((ghash_pyobj *)(o))->f)
	448	extern PyObject ghash_pywrap(PyObject , ghash *, unsigned);
	449	extern int convghash(PyObject , void );
	450	extern int convgmhash(PyObject , void );
	451
	452	typedef struct gcmac_pyobj {
f0526039	453	PyHeapTypeObject ty;
b6a86d2e	454	gcmac *cm;
	455	} gcmac_pyobj;
	456
	457	extern PyTypeObject *gcmac_pytype;
	458	#define GCMAC_PYCHECK(o) PyObject_TypeCheck((o), gcmac_pytype)
	459	#define GCMAC_CM(o) (((gcmac_pyobj *)(o))->cm)
	460	#define GCMAC_F(o) (((gcmac_pyobj *)(o))->f)
	461	extern PyObject gcmac_pywrap(gcmac );
	462	extern int convgcmac(PyObject , void );
	463
	464	typedef struct gmac_pyobj {
f0526039	465	PyHeapTypeObject ty;
b6a86d2e	466	unsigned f;
b6a86d2e	467	gmac *m;
b6a86d2e	468	} gmac_pyobj;
	469
	470	extern PyTypeObject *gmac_pytype;
	471	#define GMAC_PYCHECK(o) PyObject_TypeCheck((o), gmac_pytype)
	472	#define GMAC_M(o) (((gmac_pyobj *)(o))->m)
b6a86d2e	473	#define GMAC_F(o) (((gmac_pyobj *)(o))->f)
	474	extern PyObject gmac_pywrap(PyObject , gmac *, unsigned);
	475	extern int convgmac(PyObject , void );
	476
b6a86d2e	477	/----- Key generation ----------------------------------------------------/
	478
	479	typedef struct pfilt_pyobj {
	480	PyObject_HEAD
	481	pfilt f;
	482	int st;
	483	} pfilt_pyobj;
	484
	485	extern PyTypeObject *pfilt_pytype;
	486	#define PFILT_PYCHECK(o) PyObject_TypeCheck(o, pfilt_pytype)
	487	#define PFILT_F(o) (&((pfilt_pyobj *)(o))->f)
	488	#define PFILT_ST(o) (((pfilt_pyobj *)(o))->st)
	489
	490	typedef struct { pgen_proc proc; void ctx; } pgev;
	491	#define PGEV_HEAD PyObject_HEAD pgev pg;
	492
	493	typedef struct pgev_pyobj {
	494	PGEV_HEAD
	495	} pgev_pyobj;
	496
	497	extern PyTypeObject *pgev_pytype;
	498	#define PGEV_PYCHECK(o) PyObject_TypeCheck(o, pgev_pytype)
	499	#define PGEV_PG(o) (&((pgev_pyobj *)(o))->pg)
	500
	501	extern int convpgev(PyObject , void );
	502	extern void droppgev(pgev *);
	503	extern void pgenerr(void);
	504
	505	/----- Core utility functions --------------------------------------------/
	506
	507	extern PyObject mexp_common(PyObject , PyObject *, size_t,
	508	PyObject (id)(PyObject *),
	509	int (fill)(void , PyObject *,
	510	PyObject , PyObject ),
	511	PyObject (exp)(PyObject , void , int),
	512	void (drop)(void ));
	513
	514	extern int convulong(PyObject , void );
73a712fa	515	#define DECL_CONVU_(n) extern int convu##n(PyObject , void );
73a712fa	516	DOUINTSZ(DECL_CONVU_)
b6a86d2e	517	extern int convmpw(PyObject , void );
	518	extern int convuint(PyObject , void );
	519	extern int convszt(PyObject , void );
	520	extern int convbool(PyObject , void );
	521	extern PyObject abstract_pynew(PyTypeObject , PyObject , PyObject );
	522	extern PyObject *getbool(int);
73a712fa	523	#define DECL_GETU_(n) extern PyObject *getu##n(uint##n);
	524	DOUINTSZ(DECL_GETU_)
	525	extern PyObject * mkexc(PyObject , PyObject , const char , PyMethodDef );
f0526039	526	extern void newtype(PyTypeObject , const PyTypeObject , const char );
b6a86d2e	527	extern PyTypeObject inittype(PyTypeObject );
	528	extern void addmethods(const PyMethodDef *);
	529
	530	/----- That's all, folks -------------------------------------------------/
	531
	532	#ifdef __cplusplus
	533	}
	534	#endif
	535
	536	#endif