#include <Python.h>

#include <systemd/sd-messages.h>

#include "pyutil.h"

PyDoc_STRVAR(module__doc__,

"Python interface to the libsystemd-id128 library.\n\n"

"Provides SD_MESSAGE_* constants and functions to query and generate\n"

"128-bit unique identifiers."

);

PyDoc_STRVAR(randomize__doc__,

"randomize() -> UUID\n\n"

"Return a new random 128-bit unique identifier.\n"

"Wraps sd_id128_randomize(3)."

);

PyDoc_STRVAR(get_machine__doc__,

"get_machine() -> UUID\n\n"

"Return a 128-bit unique identifier for this machine.\n"

"Wraps sd_id128_get_machine(3)."

);

PyDoc_STRVAR(get_boot__doc__,

"get_boot() -> UUID\n\n"

"Return a 128-bit unique identifier for this boot.\n"

"Wraps sd_id128_get_boot(3)."

);

static PyObject* make_uuid(sd_id128_t id) {

_cleanup_Py_DECREF_ PyObject

*uuid = NULL, *UUID = NULL, *bytes = NULL,

*args = NULL, *kwargs = NULL;

uuid = PyImport_ImportModule("uuid");

if (!uuid)

return NULL;

UUID = PyObject_GetAttrString(uuid, "UUID");

bytes = PyBytes_FromStringAndSize((const char*) &id.bytes, sizeof(id.bytes));

args = Py_BuildValue("()");

kwargs = PyDict_New();

if (!UUID || !bytes || !args || !kwargs)

return NULL;

if (PyDict_SetItemString(kwargs, "bytes", bytes) < 0)

return NULL;

return PyObject_Call(UUID, args, kwargs);

}

#define helper(name) \

static PyObject *name(PyObject *self, PyObject *args) { \

sd_id128_t id; \

int r; \

\

assert(args == NULL); \

\

r = sd_id128_##name(&id); \

if (r < 0) { \

errno = -r; \

return PyErr_SetFromErrno(PyExc_IOError); \

} \

\

return make_uuid(id); \

}

helper(randomize)

helper(get_machine)

helper(get_boot)

static PyMethodDef methods[] = {

{ "randomize", randomize, METH_NOARGS, randomize__doc__},

{ "get_machine", get_machine, METH_NOARGS, get_machine__doc__},

{ "get_boot", get_boot, METH_NOARGS, get_boot__doc__},

{ NULL, NULL, 0, NULL } /* Sentinel */

};

static int add_id(PyObject *module, const char* name, sd_id128_t id) {

PyObject *obj;

obj = make_uuid(id);

if (!obj)

return -1;

return PyModule_AddObject(module, name, obj);

}

#pragma GCC diagnostic push

#pragma GCC diagnostic ignored "-Wmissing-prototypes"

#if PY_MAJOR_VERSION < 3

PyMODINIT_FUNC initid128(void) {

PyObject *m;

m = Py_InitModule3("id128", methods, module__doc__);

if (m == NULL)

return;

/* a series of lines like 'add_id() ;' follow */

#define JOINER ;

#include "id128-constants.h"

#undef JOINER

PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION);

}

#else

static struct PyModuleDef module = {

PyModuleDef_HEAD_INIT,

"id128", /* name of module */

module__doc__, /* module documentation, may be NULL */

-1, /* size of per-interpreter state of the module */

methods

};

PyMODINIT_FUNC PyInit_id128(void) {

PyObject *m;

m = PyModule_Create(&module);

if (m == NULL)

return NULL;

if ( /* a series of lines like 'add_id() ||' follow */

#define JOINER ||

#include "id128-constants.h"

#undef JOINER

PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION)) {

Py_DECREF(m);

return NULL;

}

return m;

}

#endif

#pragma GCC diagnostic pop