Checkin, Debianized and more or less complete.

[pyke] / catacomb.c

diff --git a/catacomb.c b/catacomb.c
index 0e1c711..47a94dd 100644 (file)
--- a/catacomb.c
+++ b/catacomb.c
@@ -41,6 +41,16 @@ static void setconstants(PyObject *mod)
       C(PGEN_ABORT),
     C(MPW_MAX),
     C(PMODE_READ), C(PMODE_VERIFY),
+    C(KOPEN_READ), C(KOPEN_WRITE), C(KOPEN_NOFILE),
+    C(KEXP_FOREVER), C(KEXP_EXPIRE),
+    C(KF_ENCMASK), C(KENC_BINARY), C(KENC_MP), C(KENC_STRUCT),
+      C(KENC_ENCRYPT), C(KENC_STRING), C(KENC_EC),
+    C(KF_CATMASK), C(KCAT_SYMM), C(KCAT_PRIV), C(KCAT_PUB), C(KCAT_SHARE),
+    C(KF_NONSECRET),
+    C(KF_BURN), C(KF_OPT),
+#define ENTRY(tag, val, str) C(KERR_##tag),
+    KEY_ERRORS(ENTRY)
+#undef ENTRY
 #undef C
     { 0 }
   };
@@ -56,13 +66,15 @@ static void setconstants(PyObject *mod)
   }
 }
 
-PyObject *getu32(uint32 w)
-{
-  if (w <= 0x7fffffff)
-    return (PyInt_FromLong(w));
-  else
-    return (PyLong_FromUnsignedLong(w));
-}
+#define GETU_(n)                                                       \
+  PyObject *getu##n(uint##n w)                                         \
+  {                                                                    \
+    if (w <= MASK##n)                                                  \
+      return (PyInt_FromLong(w));                                      \
+    else                                                               \
+      return (PyLong_FromUnsignedLong(w));                             \
+  }
+DOUINTSZ(GETU_)
 
 PyObject *getbool(int b)
 {
@@ -97,18 +109,20 @@ end:
   return (0);
 }
 
-int convu32(PyObject *o, void *pp)
-{
-  unsigned long u;
-  uint32 *p = pp;
-
-  if (!convulong(o, &u)) goto end;
-  if (u > 0xffffffff) TYERR("out of range");
-  *p = u;
-  return (1);
-end:
-  return (0);
-}
+#define CONVU_(n)                                                      \
+  int convu##n(PyObject *o, void *pp)                                  \
+  {                                                                    \
+    unsigned long u;                                                   \
+    uint##n *p = pp;                                                   \
+                                                                       \
+    if (!convulong(o, &u)) goto end;                                   \
+    if (u > MASK##n) TYERR("out of range");                            \
+    *p = u;                                                            \
+    return (1);                                                                \
+  end:                                                                 \
+    return (0);                                                                \
+  }
+DOUINTSZ(CONVU_)
 
 int convuint(PyObject *o, void *pp)
 {
@@ -221,6 +235,48 @@ end:
   return (z);
 }
 
+PyObject * mkexc(PyObject *mod, PyObject *base,
+                const char *name, PyMethodDef *mm)
+{
+  PyObject *nameobj = 0;
+  PyObject *dict = 0;
+  PyObject *exc = 0;
+  PyObject *func = 0;
+  PyObject *meth = 0;
+
+  if ((nameobj = PyString_FromFormat("%s.%s",
+                                    PyModule_GetName(mod),
+                                    name)) == 0 ||
+      (dict = PyDict_New()) == 0 ||
+      (exc = PyErr_NewException(PyString_AS_STRING(nameobj),
+                               base, dict)) == 0)
+    goto fail;
+
+  if (mm) {
+    while (mm->ml_name) {
+      if ((func = PyCFunction_NewEx(mm, 0, mod)) == 0 ||
+         (meth = PyMethod_New(func, 0, exc)) == 0 ||
+         PyDict_SetItemString(dict, mm->ml_name, meth))
+       goto fail;
+      Py_DECREF(func); func = 0;
+      Py_DECREF(meth); meth = 0;
+      mm++;
+    }
+  }
+
+done:
+  Py_XDECREF(nameobj);
+  Py_XDECREF(dict);
+  return (exc);
+
+fail:
+  Py_XDECREF(exc);
+  Py_XDECREF(func);
+  Py_XDECREF(meth);
+  exc = 0;
+  goto done;
+}
+
 DA_DECL(method_v, PyMethodDef);
 static method_v global_pymethods = DA_INIT;
 void addmethods(const PyMethodDef *m)
@@ -287,9 +343,27 @@ PyTypeObject *inittype(PyTypeObject *tyskel)
   return (ty);
 }
 
+static PyObject *meth__ego(PyObject *me, PyObject *arg)
+{
+  char *argv0;
+  if (!PyArg_ParseTuple(arg, "s:_ego", &argv0))
+    return (0);
+  if (strcmp(QUIS, "<UNNAMED>") == 0)
+    ego(argv0);
+  RETURN_NONE;
+}
+
+static PyMethodDef methods[] = {
+#define METHNAME(func) meth_##func
+  METH (_ego,                  "_ego(ARGV0)")
+#undef METHNAME
+  { 0 }
+};
+
 void init_base(void) {
   static const PyMethodDef mzero = { 0 };
   PyObject *mod;
+  addmethods(methods);
   INIT_MODULES;
   DA_PUSH(&global_pymethods, mzero);
   mod = Py_InitModule("catacomb._base", DA(&global_pymethods));