#define PY_SSIZE_T_CLEAN

#pragma GCC diagnostic push

#pragma GCC diagnostic ignored "-Wredundant-decls"

#include <Python.h>

#pragma GCC diagnostic pop

#include "systemd/sd-login.h"

#include "pyutil.h"

#include "util.h"

#include "strv.h"

PyDoc_STRVAR(module__doc__,

"Python interface to the libsystemd-login library."

);

#define helper(name) \

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

_cleanup_strv_free_ char **list = NULL; \

int r; \

PyObject *ans; \

\

assert(args == NULL); \

\

r = sd_get_##name(&list); \

if (r < 0) { \

errno = -r; \

return PyErr_SetFromErrno(PyExc_IOError); \

} \

\

ans = PyList_New(r); \

if (!ans) \

return NULL; \

\

for (r--; r >= 0; r--) { \

PyObject *s = unicode_FromString(list[r]); \

if (!s) { \

Py_DECREF(ans); \

return NULL; \

} \

\

PyList_SetItem(ans, r, s); \

} \

\

return ans; \

}

helper(seats)

helper(sessions)

helper(machine_names)

#undef helper

static PyObject* uids(PyObject *self, PyObject *args) {

_cleanup_free_ uid_t *list = NULL;

int r;

PyObject *ans;

assert(args == NULL);

r = sd_get_uids(&list);

if (r < 0) {

errno = -r;

return PyErr_SetFromErrno(PyExc_IOError);

}

ans = PyList_New(r);

if (!ans)

return NULL;

for (r--; r >= 0; r--) {

PyObject *s = long_FromLong(list[r]);

if (!s) {

Py_DECREF(ans);

return NULL;

}

PyList_SetItem(ans, r, s);

}

return ans;

}

PyDoc_STRVAR(seats__doc__,

"seats() -> list\n\n"

"Returns a list of currently available local seats.\n"

"Wraps sd_get_seats(3)."

);

PyDoc_STRVAR(sessions__doc__,

"sessions() -> list\n\n"

"Returns a list of current login sessions.\n"

"Wraps sd_get_sessions(3)."

);

PyDoc_STRVAR(machine_names__doc__,

"machine_names() -> list\n\n"

"Returns a list of currently running virtual machines\n"

"and containers on the system.\n"

"Wraps sd_get_machine_names(3)."

);

PyDoc_STRVAR(uids__doc__,

"uids() -> list\n\n"

"Returns a list of uids of users who currently have login sessions.\n"

"Wraps sd_get_uids(3)."

);

static PyMethodDef methods[] = {

{ "seats", seats, METH_NOARGS, seats__doc__},

{ "sessions", sessions, METH_NOARGS, sessions__doc__},

{ "machine_names", machine_names, METH_NOARGS, machine_names__doc__},

{ "uids", uids, METH_NOARGS, uids__doc__},

{} /* Sentinel */

};

typedef struct {

PyObject_HEAD

sd_login_monitor *monitor;

} Monitor;

static PyTypeObject MonitorType;

static void Monitor_dealloc(Monitor* self) {

sd_login_monitor_unref(self->monitor);

Py_TYPE(self)->tp_free((PyObject*)self);

}

PyDoc_STRVAR(Monitor__doc__,

"Monitor([category]) -> ...\n\n"

"Monitor may be used to monitor login sessions, users, seats,\n"

"and virtual machines/containers. Monitor provides a file\n"

"descriptor which can be integrated in an external event loop.\n"

"See man:sd_login_monitor_new(3) for the details about what\n"

"can be monitored.");

static int Monitor_init(Monitor *self, PyObject *args, PyObject *keywds) {

const char *category = NULL;

int r;

static const char* const kwlist[] = {"category", NULL};

if (!PyArg_ParseTupleAndKeywords(args, keywds, "|z:__init__", (char**) kwlist,

&category))

return -1;

Py_BEGIN_ALLOW_THREADS

r = sd_login_monitor_new(category, &self->monitor);

Py_END_ALLOW_THREADS

return set_error(r, NULL, "Invalid category");

}

PyDoc_STRVAR(Monitor_fileno__doc__,

"fileno() -> int\n\n"

"Get a file descriptor to poll for events.\n"

"This method wraps sd_login_monitor_get_fd(3).");

static PyObject* Monitor_fileno(Monitor *self, PyObject *args) {

int fd = sd_login_monitor_get_fd(self->monitor);

set_error(fd, NULL, NULL);

if (fd < 0)

return NULL;

return long_FromLong(fd);

}

PyDoc_STRVAR(Monitor_get_events__doc__,

"get_events() -> int\n\n"

"Returns a mask of poll() events to wait for on the file\n"

"descriptor returned by .fileno().\n\n"

"See man:sd_login_monitor_get_events(3) for further discussion.");

static PyObject* Monitor_get_events(Monitor *self, PyObject *args) {

int r = sd_login_monitor_get_events(self->monitor);

set_error(r, NULL, NULL);

if (r < 0)

return NULL;

return long_FromLong(r);

}

PyDoc_STRVAR(Monitor_get_timeout__doc__,

"get_timeout() -> int or None\n\n"

"Returns a timeout value for usage in poll(), the time since the\n"

"epoch of clock_gettime(2) in microseconds, or None if no timeout\n"

"is necessary.\n\n"

"The return value must be converted to a relative timeout in\n"

"milliseconds if it is to be used as an argument for poll().\n"

"See man:sd_login_monitor_get_timeout(3) for further discussion.");

static PyObject* Monitor_get_timeout(Monitor *self, PyObject *args) {

int r;

uint64_t t;

r = sd_login_monitor_get_timeout(self->monitor, &t);

set_error(r, NULL, NULL);

if (r < 0)

return NULL;

if (t == (uint64_t) -1)

Py_RETURN_NONE;

assert_cc(sizeof(unsigned long long) == sizeof(t));

return PyLong_FromUnsignedLongLong(t);

}

PyDoc_STRVAR(Monitor_get_timeout_ms__doc__,

"get_timeout_ms() -> int\n\n"

"Returns a timeout value suitable for usage in poll(), the value\n"

"returned by .get_timeout() converted to relative ms, or -1 if\n"

"no timeout is necessary.");

static PyObject* Monitor_get_timeout_ms(Monitor *self, PyObject *args) {

int r;

uint64_t t;

r = sd_login_monitor_get_timeout(self->monitor, &t);

set_error(r, NULL, NULL);

if (r < 0)

return NULL;

return absolute_timeout(t);

}

PyDoc_STRVAR(Monitor_close__doc__,

"close() -> None\n\n"

"Free resources allocated by this Monitor object.\n"

"This method invokes sd_login_monitor_unref().\n"

"See man:sd_login_monitor_unref(3).");

static PyObject* Monitor_close(Monitor *self, PyObject *args) {

assert(self);

assert(!args);

sd_login_monitor_unref(self->monitor);

self->monitor = NULL;

Py_RETURN_NONE;

}

PyDoc_STRVAR(Monitor_flush__doc__,

"flush() -> None\n\n"

"Reset the wakeup state of the monitor object.\n"

"This method invokes sd_login_monitor_flush().\n"

"See man:sd_login_monitor_flush(3).");

static PyObject* Monitor_flush(Monitor *self, PyObject *args) {

assert(self);

assert(!args);

Py_BEGIN_ALLOW_THREADS

sd_login_monitor_flush(self->monitor);

Py_END_ALLOW_THREADS

Py_RETURN_NONE;

}

PyDoc_STRVAR(Monitor___enter____doc__,

"__enter__() -> self\n\n"

"Part of the context manager protocol.\n"

"Returns self.\n");

static PyObject* Monitor___enter__(PyObject *self, PyObject *args) {

assert(self);

assert(!args);

Py_INCREF(self);

return self;

}

PyDoc_STRVAR(Monitor___exit____doc__,

"__exit__(type, value, traceback) -> None\n\n"

"Part of the context manager protocol.\n"

"Closes the monitor..\n");

static PyObject* Monitor___exit__(Monitor *self, PyObject *args) {

return Monitor_close(self, args);

}

static PyMethodDef Monitor_methods[] = {

{"fileno", (PyCFunction) Monitor_fileno, METH_NOARGS, Monitor_fileno__doc__},

{"get_events", (PyCFunction) Monitor_get_events, METH_NOARGS, Monitor_get_events__doc__},

{"get_timeout", (PyCFunction) Monitor_get_timeout, METH_NOARGS, Monitor_get_timeout__doc__},

{"get_timeout_ms", (PyCFunction) Monitor_get_timeout_ms, METH_NOARGS, Monitor_get_timeout_ms__doc__},

{"close", (PyCFunction) Monitor_close, METH_NOARGS, Monitor_close__doc__},

{"flush", (PyCFunction) Monitor_flush, METH_NOARGS, Monitor_flush__doc__},

{"__enter__", (PyCFunction) Monitor___enter__, METH_NOARGS, Monitor___enter____doc__},

{"__exit__", (PyCFunction) Monitor___exit__, METH_VARARGS, Monitor___exit____doc__},

{} /* Sentinel */

};

static PyTypeObject MonitorType = {

PyVarObject_HEAD_INIT(NULL, 0)

.tp_name = "login.Monitor",

.tp_basicsize = sizeof(Monitor),

.tp_dealloc = (destructor) Monitor_dealloc,

.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,

.tp_doc = Monitor__doc__,

.tp_methods = Monitor_methods,

.tp_init = (initproc) Monitor_init,

.tp_new = PyType_GenericNew,

};

#pragma GCC diagnostic push

#pragma GCC diagnostic ignored "-Wmissing-prototypes"

#if PY_MAJOR_VERSION < 3

PyMODINIT_FUNC initlogin(void) {

PyObject *m;

if (PyType_Ready(&MonitorType) < 0)

return;

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

if (m == NULL)

return;

PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION);

Py_INCREF(&MonitorType);

PyModule_AddObject(m, "Monitor", (PyObject *) &MonitorType);

}

#else

static struct PyModuleDef module = {

PyModuleDef_HEAD_INIT,

"login", /* name of module */

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

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

methods

};

PyMODINIT_FUNC PyInit_login(void) {

PyObject *m;

if (PyType_Ready(&MonitorType) < 0)

return NULL;

m = PyModule_Create(&module);

if (m == NULL)

return NULL;

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

Py_DECREF(m);

return NULL;

}

Py_INCREF(&MonitorType);

if (PyModule_AddObject(m, "Monitor", (PyObject *) &MonitorType)) {

Py_DECREF(&MonitorType);

Py_DECREF(m);

return NULL;

}

return m;

}

#endif

#pragma GCC diagnostic pop