#include "Python.h"

#include "structseq.h"

#include <sys/types.h>

#include <pwd.h>

static PyStructSequence_Field struct_pwd_type_fields[] = {

{"pw_name", "user name"},

{"pw_passwd", "password"},

{"pw_uid", "user id"},

{"pw_gid", "group id"},

{"pw_gecos", "real name"},

{"pw_dir", "home directory"},

{"pw_shell", "shell program"},

{0}

};

PyDoc_STRVAR(struct_passwd__doc__,

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

static PyStructSequence_Desc struct_pwd_type_desc = {

"pwd.struct_passwd",

struct_passwd__doc__,

struct_pwd_type_fields,

7,

};

PyDoc_STRVAR(pwd__doc__,

exception is raised if the entry asked for cannot be found.");

static int initialized;

static PyTypeObject StructPwdType;

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 *

mkpwent(struct passwd *p)

{

int setIndex = 0;

PyObject *v = PyStructSequence_New(&StructPwdType);

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->pw_name);

#ifdef __VMS

SETS(setIndex++, "");

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

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

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

#ifdef __VMS

SETS(setIndex++, "");

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

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

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

#undef SETS

#undef SETI

if (PyErr_Occurred()) {

Py_XDECREF(v);

return NULL;

}

return v;

}

PyDoc_STRVAR(pwd_getpwuid__doc__,

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

static PyObject *

pwd_getpwuid(PyObject *self, PyObject *args)

{

unsigned int uid;

struct passwd *p;

if (!PyArg_ParseTuple(args, "I:getpwuid", &uid))

return NULL;

if ((p = getpwuid(uid)) == NULL) {

PyErr_Format(PyExc_KeyError,

"getpwuid(): uid not found: %d", uid);

return NULL;

}

return mkpwent(p);

}

PyDoc_STRVAR(pwd_getpwnam__doc__,

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

static PyObject *

pwd_getpwnam(PyObject *self, PyObject *args)

{

char *name;

struct passwd *p;

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

return NULL;

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

PyErr_Format(PyExc_KeyError,

"getpwnam(): name not found: %s", name);

return NULL;

}

return mkpwent(p);

}

#ifdef HAVE_GETPWENT

PyDoc_STRVAR(pwd_getpwall__doc__,

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

static PyObject *

pwd_getpwall(PyObject *self)

{

PyObject *d;

struct passwd *p;

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

return NULL;

#if defined(PYOS_OS2) && defined(PYCC_GCC)

if ((p = getpwuid(0)) != NULL) {

setpwent();

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

PyObject *v = mkpwent(p);

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

Py_XDECREF(v);

Py_DECREF(d);

return NULL;

}

Py_DECREF(v);

}

endpwent();

return d;

}

static PyMethodDef pwd_methods[] = {

{"getpwuid", pwd_getpwuid, METH_VARARGS, pwd_getpwuid__doc__},

{"getpwnam", pwd_getpwnam, METH_VARARGS, pwd_getpwnam__doc__},

#ifdef HAVE_GETPWENT

{"getpwall", (PyCFunction)pwd_getpwall,

METH_NOARGS, pwd_getpwall__doc__},

{NULL, NULL} /* sentinel */

};

initpwd(void)

{

PyObject *m;

m = Py_InitModule3("pwd", pwd_methods, pwd__doc__);

if (m == NULL)

return;

if (!initialized)

PyStructSequence_InitType(&StructPwdType,

&struct_pwd_type_desc);

Py_INCREF((PyObject *) &StructPwdType);

PyModule_AddObject(m, "struct_passwd", (PyObject *) &StructPwdType);

/* And for b/w compatibility (this was defined by mistake): */

PyModule_AddObject(m, "struct_pwent", (PyObject *) &StructPwdType);

initialized = 1;

}