#include "Python.h"

#include "structseq.h"

#include <sys/types.h>

#ifdef HAVE_SHADOW_H

#include <shadow.h>

PyDoc_STRVAR(spwd__doc__,

You have to be root to be able to use this module.");

#if defined(HAVE_GETSPNAM) || defined(HAVE_GETSPENT)

static PyStructSequence_Field struct_spwd_type_fields[] = {

{"sp_nam", "login name"},

{"sp_pwd", "encrypted password"},

{"sp_lstchg", "date of last change"},

{"sp_min", "min #days between changes"},

{"sp_max", "max #days between changes"},

{"sp_warn", "#days before pw expires to warn user about it"},

{"sp_inact", "#days after pw expires until account is blocked"},

{"sp_expire", "#days since 1970-01-01 until account is disabled"},

{"sp_flag", "reserved"},

{0}

};

PyDoc_STRVAR(struct_spwd__doc__,

or via the object attributes as named in the above tuple.");

static PyStructSequence_Desc struct_spwd_type_desc = {

"spwd.struct_spwd",

struct_spwd__doc__,

struct_spwd_type_fields,

9,

};

static int initialized;

static PyTypeObject StructSpwdType;

static void

sets(PyObject *v, int i, char* val)

{

if (val)

PyStructSequence_SET_ITEM(v, i, PyString_FromString(val));

else {

PyStructSequence_SET_ITEM(v, i, Py_None);

Py_INCREF(Py_None);

}

}

static PyObject *mkspent(struct spwd *p)

{

int setIndex = 0;

PyObject *v = PyStructSequence_New(&StructSpwdType);

if (v == NULL)

return NULL;

#define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyInt_FromLong((long) val))

#define SETS(i,val) sets(v, i, val)

SETS(setIndex++, p->sp_namp);

SETS(setIndex++, p->sp_pwdp);

SETI(setIndex++, p->sp_lstchg);

SETI(setIndex++, p->sp_min);

SETI(setIndex++, p->sp_max);

SETI(setIndex++, p->sp_warn);

SETI(setIndex++, p->sp_inact);

SETI(setIndex++, p->sp_expire);

SETI(setIndex++, p->sp_flag);

#undef SETS

#undef SETI

if (PyErr_Occurred()) {

Py_DECREF(v);

return NULL;

}

return v;

}

#endif /* HAVE_GETSPNAM || HAVE_GETSPENT */

#ifdef HAVE_GETSPNAM

PyDoc_STRVAR(spwd_getspnam__doc__,

See spwd.__doc__ for more on shadow password database entries.");

static PyObject* spwd_getspnam(PyObject *self, PyObject *args)

{

char *name;

struct spwd *p;

if (!PyArg_ParseTuple(args, "s:getspnam", &name))

return NULL;

if ((p = getspnam(name)) == NULL) {

PyErr_SetString(PyExc_KeyError, "getspnam(): name not found");

return NULL;

}

return mkspent(p);

}

#endif /* HAVE_GETSPNAM */

#ifdef HAVE_GETSPENT

PyDoc_STRVAR(spwd_getspall__doc__,

See spwd.__doc__ for more on shadow password database entries.");

static PyObject *

spwd_getspall(PyObject *self, PyObject *args)

{

PyObject *d;

struct spwd *p;

if ((d = PyList_New(0)) == NULL)

return NULL;

setspent();

while ((p = getspent()) != NULL) {

PyObject *v = mkspent(p);

if (v == NULL || PyList_Append(d, v) != 0) {

Py_XDECREF(v);

Py_DECREF(d);

endspent();

return NULL;

}

Py_DECREF(v);

}

endspent();

return d;

}

#endif /* HAVE_GETSPENT */

static PyMethodDef spwd_methods[] = {

#ifdef HAVE_GETSPNAM

{"getspnam", spwd_getspnam, METH_VARARGS, spwd_getspnam__doc__},

#ifdef HAVE_GETSPENT

{"getspall", spwd_getspall, METH_NOARGS, spwd_getspall__doc__},

{NULL, NULL} /* sentinel */

};

initspwd(void)

{

PyObject *m;

m=Py_InitModule3("spwd", spwd_methods, spwd__doc__);

if (m == NULL)

return;

if (!initialized)

PyStructSequence_InitType(&StructSpwdType,

&struct_spwd_type_desc);

Py_INCREF((PyObject *) &StructSpwdType);

PyModule_AddObject(m, "struct_spwd", (PyObject *) &StructSpwdType);

initialized = 1;

}