Add bindings for rpm keyring and pubkey types

1 files changed, 204 insertions, 0 deletions

diff --git a/python/rpmkeyring-py.c b/python/rpmkeyring-py.c
new file mode 100644
index 000000000..9fa4f4880
--- /dev/null
+++ b/python/rpmkeyring-py.c
@@ -0,0 +1,204 @@
+#include "rpmsystem-py.h"
+#include <rpm/rpmkeyring.h>
+#include "rpmkeyring-py.h"
+
+struct rpmPubkeyObject_s {
+    PyObject_HEAD
+    PyObject *md_dict;
+    rpmPubkey pubkey;
+};
+
+static void rpmPubkey_dealloc(rpmPubkeyObject * s)
+{
+    s->pubkey = rpmPubkeyFree(s->pubkey);
+    s->ob_type->tp_free((PyObject *)s);
+}
+
+static PyObject *rpmPubkey_new(PyTypeObject *subtype, 
+			   PyObject *args, PyObject *kwds)
+{
+    PyObject *key;
+    char *kwlist[] = { "key", NULL };
+    rpmPubkey pubkey = NULL;
+    uint8_t *pkt;
+    size_t pktlen;
+
+    if (!PyArg_ParseTupleAndKeywords(args, kwds, "S", kwlist, &key))
+	return NULL;
+
+    if (pgpParsePkts(PyString_AsString(key), &pkt, &pktlen) <= 0) {
+	PyErr_SetString(PyExc_ValueError, "invalid pubkey");
+	return NULL;
+    }
+    pubkey = rpmPubkeyNew(pkt, pktlen);
+
+    return rpmPubkey_Wrap(subtype, pubkey);
+}
+
+static PyObject * rpmPubkey_Base64(rpmPubkeyObject *s)
+{
+    char *b64 = rpmPubkeyBase64(s->pubkey);
+    PyObject *res = Py_BuildValue("s", b64);
+    free(b64);
+    return res;
+}
+
+static struct PyMethodDef rpmPubkey_methods[] = {
+    { "base64", (PyCFunction) rpmPubkey_Base64, METH_NOARGS, NULL },
+    { NULL }
+};
+
+static char rpmPubkey_doc[] = "";
+
+PyTypeObject rpmPubkey_Type = {
+	PyObject_HEAD_INIT(&PyType_Type)
+	0,				/* ob_size */
+	"rpm.pubkey",			/* tp_name */
+	sizeof(rpmPubkeyObject),	/* tp_size */
+	0,				/* tp_itemsize */
+	(destructor) rpmPubkey_dealloc,/* tp_dealloc */
+	0,				/* tp_print */
+	(getattrfunc)0, 		/* tp_getattr */
+	0,				/* tp_setattr */
+	0,				/* tp_compare */
+	0,				/* tp_repr */
+	0,				/* tp_as_number */
+	0,				/* tp_as_sequence */
+	0,				/* tp_as_mapping */
+	0,				/* tp_hash */
+	0,				/* tp_call */
+	0,				/* tp_str */
+	PyObject_GenericGetAttr,	/* tp_getattro */
+	PyObject_GenericSetAttr,	/* tp_setattro */
+	0,				/* tp_as_buffer */
+	Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,	/* tp_flags */
+	rpmPubkey_doc,			/* tp_doc */
+	0,				/* tp_traverse */
+	0,				/* tp_clear */
+	0,				/* tp_richcompare */
+	0,				/* tp_weaklistoffset */
+	0,				/* tp_iter */
+	0,				/* tp_iternext */
+	rpmPubkey_methods,		/* tp_methods */
+	0,				/* tp_members */
+	0,				/* tp_getset */
+	0,				/* tp_base */
+	0,				/* tp_dict */
+	0,				/* tp_descr_get */
+	0,				/* tp_descr_set */
+	0,				/* tp_dictoffset */
+	0,				/* tp_init */
+	0,				/* tp_alloc */
+	rpmPubkey_new,			/* tp_new */
+	0,				/* tp_free */
+	0,				/* tp_is_gc */
+};
+
+struct rpmKeyringObject_s {
+    PyObject_HEAD
+    PyObject *md_dict;
+    rpmKeyring keyring;
+};
+
+static void rpmKeyring_dealloc(rpmKeyringObject * s)
+{
+    rpmKeyringFree(s->keyring);
+    s->ob_type->tp_free((PyObject *)s);
+}
+
+static PyObject *rpmKeyring_new(PyTypeObject *subtype, 
+			   PyObject *args, PyObject *kwds)
+{
+    rpmKeyring keyring = rpmKeyringNew();
+    return rpmKeyring_Wrap(subtype, keyring);
+}
+
+static PyObject *rpmKeyring_addKey(rpmKeyringObject *s, PyObject *arg)
+{
+    rpmPubkeyObject *pubkey = NULL;
+
+    if (!PyArg_Parse(arg, "O!", &rpmPubkey_Type, &pubkey))
+	return NULL;
+
+    return Py_BuildValue("i", rpmKeyringAddKey(s->keyring, pubkey->pubkey));
+};
+
+static struct PyMethodDef rpmKeyring_methods[] = {
+    { "addKey", (PyCFunction) rpmKeyring_addKey, METH_O,
+        NULL },
+    {NULL,		NULL}		/* sentinel */
+};
+
+static char rpmKeyring_doc[] =
+"";
+
+PyTypeObject rpmKeyring_Type = {
+	PyObject_HEAD_INIT(&PyType_Type)
+	0,				/* ob_size */
+	"rpm.keyring",			/* tp_name */
+	sizeof(rpmKeyringObject),	/* tp_size */
+	0,				/* tp_itemsize */
+	(destructor) rpmKeyring_dealloc,/* tp_dealloc */
+	0,				/* tp_print */
+	0,		 		/* tp_getattr */
+	0,				/* tp_setattr */
+	0,				/* tp_compare */
+	0,				/* tp_repr */
+	0,				/* tp_as_number */
+	0,				/* tp_as_sequence */
+	0,				/* tp_as_mapping */
+	0,				/* tp_hash */
+	0,				/* tp_call */
+	0,				/* tp_str */
+	PyObject_GenericGetAttr,	/* tp_getattro */
+	PyObject_GenericSetAttr,	/* tp_setattro */
+	0,				/* tp_as_buffer */
+	Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,	/* tp_flags */
+	rpmKeyring_doc,			/* tp_doc */
+	0,				/* tp_traverse */
+	0,				/* tp_clear */
+	0,				/* tp_richcompare */
+	0,				/* tp_weaklistoffset */
+	0,				/* tp_iter */
+	0,				/* tp_iternext */
+	rpmKeyring_methods,		/* tp_methods */
+	0,				/* tp_members */
+	0,				/* tp_getset */
+	0,				/* tp_base */
+	0,				/* tp_dict */
+	0,				/* tp_descr_get */
+	0,				/* tp_descr_set */
+	0,				/* tp_dictoffset */
+	0,				/* tp_init */
+	0,				/* tp_alloc */
+	rpmKeyring_new,			/* tp_new */
+	0,				/* tp_free */
+	0,				/* tp_is_gc */
+};
+
+PyObject * rpmPubkey_Wrap(PyTypeObject *subtype, rpmPubkey pubkey)
+{
+    rpmPubkeyObject *s = (rpmPubkeyObject *)subtype->tp_alloc(subtype, 0);
+    if (s == NULL) return PyErr_NoMemory();
+
+    s->pubkey = pubkey;
+    return (PyObject*) s;
+}
+
+PyObject * rpmKeyring_Wrap(PyTypeObject *subtype, rpmKeyring keyring)
+{
+    rpmKeyringObject *s = (rpmKeyringObject *)subtype->tp_alloc(subtype, 0);
+    if (s == NULL) return PyErr_NoMemory();
+
+    s->keyring = keyring;
+    return (PyObject*) s;
+}
+
+int rpmKeyringFromPyObject(PyObject *item, rpmKeyring *keyring)
+{
+    rpmKeyringObject *kro;
+    if (!PyArg_Parse(item, "O!", &rpmKeyring_Type, &kro))
+	return 0;
+    *keyring = kro->keyring;
+    return 1;
+}