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 /*----- Main code ---------------------------------------------------------*/
33 static const struct nameval consts[] = {
34 #define CF(f, x) { #x, f, x }
35 #define C(x) { #x, (x) >= 0 ? 0 : CF_SIGNED, x }
36 C(FTY_PRIME), C(FTY_BINARY),
37 C(PGEN_PASS), C(PGEN_FAIL), C(PGEN_BEGIN), C(PGEN_TRY), C(PGEN_DONE),
41 C(PMODE_READ), C(PMODE_VERIFY),
42 C(KOPEN_READ), C(KOPEN_WRITE), C(KOPEN_NOFILE),
43 CF(0, KEXP_FOREVER), CF(0, KEXP_EXPIRE),
44 C(KF_ENCMASK), C(KENC_BINARY), C(KENC_MP), C(KENC_STRUCT),
45 C(KENC_ENCRYPT), C(KENC_STRING), C(KENC_EC),
46 C(KF_CATMASK), C(KCAT_SYMM), C(KCAT_PRIV), C(KCAT_PUB), C(KCAT_SHARE),
48 C(KF_BURN), C(KF_OPT),
49 C(EC_XONLY), C(EC_YBIT), C(EC_LSB), C(EC_CMPR), C(EC_EXPLY), C(EC_SORT),
50 C(X25519_KEYSZ), C(X25519_PUBSZ), C(X25519_OUTSZ),
51 C(X448_KEYSZ), C(X448_PUBSZ), C(X448_OUTSZ),
52 C(ED25519_KEYSZ), C(ED25519_PUBSZ), C(ED25519_SIGSZ),
53 C(ED25519_MAXPERSOSZ),
54 C(ED448_KEYSZ), C(ED448_PUBSZ), C(ED448_SIGSZ), C(ED448_MAXPERSOSZ),
55 C(AEADF_PCHSZ), C(AEADF_PCMSZ), C(AEADF_PCTSZ),
56 C(AEADF_AADNDEP), C(AEADF_AADFIRST), C(AEADF_NOAAD),
57 #define ENTRY(tag, val, str) C(KERR_##tag),
65 PyObject *mexp_common(PyObject *me, PyObject *arg,
67 PyObject *(*id)(PyObject *),
68 int (*fill)(void *, PyObject *,
69 PyObject *, PyObject *),
70 PyObject *(*exp)(PyObject *, void *, int),
73 int i = 0, j, n, flat;
74 PyObject *qq, *x, *y, *z = 0;
77 if (PyTuple_GET_SIZE(arg) == 1)
78 arg = PyTuple_GET_ITEM(arg, 0);
80 if (!PySequence_Check(arg)) TYERR("not a sequence");
81 n = PySequence_Size(arg); if (n < 0) goto end;
82 if (!n) { z = id(me); goto end; }
83 x = PySequence_GetItem(arg, 0);
84 if (PySequence_Check(x))
87 if (n % 2) VALERR("must have even number of arguments");
93 v = xmalloc(n * efsz);
94 for (i = j = 0, vv = v; i < n; i++, vv += efsz) {
96 x = PySequence_GetItem(arg, j++);
97 y = PySequence_GetItem(arg, j++);
99 qq = PySequence_GetItem(arg, j++);
101 if (!PySequence_Check(qq) || PySequence_Size(qq) != 2) {
103 TYERR("want a sequence of pairs");
105 x = PySequence_GetItem(qq, 0);
106 y = PySequence_GetItem(qq, 1);
109 if (!x || !y) goto end;
110 if (fill(vv, me, x, y)) {
113 if (!PyErr_Occurred())
114 PyErr_SetString(PyExc_TypeError, "type mismatch");
124 for (j = 0, vv = v; j < i; j++, vv += efsz)
132 static PyObject *smallprimes(void)
134 PyObject *v = PyList_New(NPRIME);
137 for (i = 0; i < NPRIME; i++)
138 PyList_SET_ITEM(v, i, PyInt_FromLong(primetab[i]));
142 static PyObject *meth__ego(PyObject *me, PyObject *arg)
145 if (!PyArg_ParseTuple(arg, "s:_ego", &argv0))
147 if (STRCMP(QUIS, ==, "<UNNAMED>"))
152 static PyMethodDef methods[] = {
153 #define METHNAME(func) meth_##func
154 METH (_ego, "_ego(ARGV0)")
159 static void init_random(void)
161 #if PY_MAJOR_VERSION >= 3 || (PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION >= 6)
165 if (!Py_HashRandomizationFlag) return;
166 seed = getenv("PYTHONHASHSEED");
167 if (!seed || STRCMP(seed, ==, "random")) r = GR_WORD(&rand_global);
168 else r = strtoul(seed, 0, 0);
169 if (!r) r = 0xe011f220; /* zero doesn't work well */
170 unihash_setkey(&unihash_global, r);
174 EXPORT void init_base(void)
180 mod = Py_InitModule("catacomb._base", donemethods());
182 INSERT("smallprimes", smallprimes());
183 setconstants(mod, consts);
186 /*----- That's all, folks -------------------------------------------------*/