1

#include <Python.h>

     2

     3

#include <utmp.h>

     4

     5

#include <stdio.h>

     6

     7

PyObject *parse (PyObject *self, PyObject *args) {

     8

    char *bytes;

     9

    size_t size;

    10

    11

    struct utmp *item;

    12

    13

    /* not unicode */

    14

    if (!PyArg_ParseTuple(args, "t#", &bytes, &size))

    15

        return NULL;

    16

    17

    if (size != sizeof(struct utmp)) {

    18

        PyErr_SetString(PyExc_ValueError, "given buffer is of the wrong length");

    19

        return NULL;

    20

    }

    21

    22

    item = (struct utmp *) bytes;

    23

    24

    /* parse utmp from bytes to result */

    25

    return Py_BuildValue("hIs#s#s#s#(hh)i(ii)s#",

    26

        item->ut_type, item->ut_pid,

    27

        item->ut_line, sizeof(item->ut_line),

    28

        item->ut_id, sizeof(item->ut_id),

    29

        item->ut_user, sizeof(item->ut_user),

    30

        item->ut_host, sizeof(item->ut_host),

    31

        item->ut_exit.e_termination, item->ut_exit.e_exit,

    32

        item->ut_session,

    33

        item->ut_tv.tv_sec, item->ut_tv.tv_usec,

    34

        item->ut_addr_v6, sizeof(item->ut_addr_v6)

    35

    );

    36

}

    37

    38

PyObject *size (PyObject *self, PyObject *args) {

    39

    /* return the size of an UTMP struct */

    40

    41

    if (!PyArg_ParseTuple(args, ""))

    42

        return NULL;

    43

    44

    return PyInt_FromSsize_t(sizeof(struct utmp));

    45

}

    46

    47

static PyMethodDef module_methods[] = {

    48

    {"parse",       parse,      METH_VARARGS,   "parse a utmp struct from a byte string"},

    49

    {"size",        size,       METH_VARARGS,   "return the size of an utmp record in bytes"},

    50

    {NULL}

    51

};

    52

    53

PyMODINIT_FUNC init_utmp(void) {

    54

    PyObject *m;

    55

    56

    m = Py_InitModule3("_utmp", module_methods, "utmp struct parsing");

    57

}