X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~mdw/git/catacomb-python/blobdiff_plain/6d4db0bf4b67e65708637466d4f0d306ed1dfe53..0463996f64aa7cde2c3d033f81d43963f54c2bba:/algorithms.c

diff --git a/algorithms.c b/algorithms.c
index 2113cc5..bd094fd 100644
--- a/algorithms.c
+++ b/algorithms.c
@@ -777,9 +777,48 @@ static PyGetSetDef gchash_pygetset[] = {
   { 0 }
 };
 
+#define GHMETH_HASHU_(n, W, w)						\
+  static PyObject *ghmeth_hashu##w(PyObject *me, PyObject *arg)		\
+  {									\
+    uint##n x;								\
+    if (!PyArg_ParseTuple(arg, "O&:hashu" #w, convu##n, &x)) goto end;	\
+    GH_HASHU##W(GHASH_H(me), x);					\
+    RETURN_ME;								\
+  end:									\
+    return (0);								\
+  }
+DOUINTCONV(GHMETH_HASHU_)
+
+#define GHMETH_HASHBUF_(n, W, w)					\
+  static PyObject *ghmeth_hashbuf##w(PyObject *me, PyObject *arg)	\
+  {									\
+    char *p;								\
+    int sz;								\
+    if (!PyArg_ParseTuple(arg, "s#:hashbuf" #w, &p, &sz)) goto end;	\
+    if (sz > MASK##n) TYERR("string too long");				\
+    GH_HASHBUF##W(GHASH_H(me), p, sz);					\
+    RETURN_ME;								\
+  end:									\
+    return (0);								\
+  }
+DOUINTCONV(GHMETH_HASHBUF_)
+
+static PyObject *ghmeth_hashstrz(PyObject *me, PyObject *arg)
+{
+  char *p;
+  if (!PyArg_ParseTuple(arg, "s:hashstrz", &p)) return (0);
+  GH_HASHSTRZ(GHASH_H(me), p);
+  RETURN_ME;
+}
+
 static PyMethodDef ghash_pymethods[] = {
 #define METHNAME(name) ghmeth_##name
   METH	(hash,			"H.hash(M)")
+#define METHU_(n, W, w) METH(hashu##w, "H.hashu" #w "(WORD)")
+  DOUINTCONV(METHU_)
+#define METHBUF_(n, W, w) METH(hashbuf##w, "H.hashbuf" #w "(BYTES)")
+  DOUINTCONV(METHBUF_)
+  METH	(hashstrz,		"H.hashstrz(STRING)")
   METH	(done,			"H.done() -> HASH")
 #undef METHNAME
   { 0 }