cpython/Modules/pwdmodule.c

125 lines
3.1 KiB
C
Raw Normal View History

1991-04-10 16:48:25 -03:00
/* UNIX password file access module */
1991-04-10 16:48:25 -03:00
1996-12-11 12:54:40 -04:00
#include "Python.h"
1991-04-10 16:48:25 -03:00
#include <sys/types.h>
#include <pwd.h>
1998-03-03 18:03:26 -04:00
static char pwd__doc__ [] = "\
This module provides access to the Unix password database.\n\
It is available on all Unix versions.\n\
\n\
Password database entries are reported as 7-tuples containing the following\n\
items from the password database (see `<pwd.h>'), in order:\n\
pw_name, pw_passwd, pw_uid, pw_gid, pw_gecos, pw_dir, pw_shell.\n\
The uid and gid items are integers, all others are strings. An\n\
exception is raised if the entry asked for cannot be found.";
1996-12-11 12:54:40 -04:00
static PyObject *
2000-07-10 10:12:27 -03:00
mkpwent(struct passwd *p)
1991-04-10 16:48:25 -03:00
{
1996-12-19 18:22:32 -04:00
return Py_BuildValue(
"(ssllsss)",
p->pw_name,
p->pw_passwd,
1995-02-07 11:38:56 -04:00
#if defined(NeXT) && defined(_POSIX_SOURCE) && defined(__LITTLE_ENDIAN__)
/* Correct a bug present on Intel machines in NextStep 3.2 and 3.3;
for later versions you may have to remove this */
1996-12-19 18:22:32 -04:00
(long)p->pw_short_pad1, /* ugh-NeXT broke the padding */
(long)p->pw_short_pad2,
1995-02-07 11:38:56 -04:00
#else
1996-12-19 18:22:32 -04:00
(long)p->pw_uid,
(long)p->pw_gid,
1995-02-07 11:38:56 -04:00
#endif
1996-12-19 18:22:32 -04:00
p->pw_gecos,
p->pw_dir,
p->pw_shell);
1991-04-10 16:48:25 -03:00
}
1998-03-03 18:03:26 -04:00
static char pwd_getpwuid__doc__[] = "\
getpwuid(uid) -> entry\n\
Return the password database entry for the given numeric user ID.\n\
See pwd.__doc__ for more on password database entries.";
1996-12-11 12:54:40 -04:00
static PyObject *
2000-07-10 10:12:27 -03:00
pwd_getpwuid(PyObject *self, PyObject *args)
1991-04-10 16:48:25 -03:00
{
int uid;
struct passwd *p;
1996-12-11 12:54:40 -04:00
if (!PyArg_Parse(args, "i", &uid))
1991-04-10 16:48:25 -03:00
return NULL;
if ((p = getpwuid(uid)) == NULL) {
1996-12-11 12:54:40 -04:00
PyErr_SetString(PyExc_KeyError, "getpwuid(): uid not found");
1991-04-10 16:48:25 -03:00
return NULL;
}
return mkpwent(p);
}
1998-03-03 18:03:26 -04:00
static char pwd_getpwnam__doc__[] = "\
getpwnam(name) -> entry\n\
Return the password database entry for the given user name.\n\
See pwd.__doc__ for more on password database entries.";
1996-12-11 12:54:40 -04:00
static PyObject *
2000-07-10 10:12:27 -03:00
pwd_getpwnam(PyObject *self, PyObject *args)
1991-04-10 16:48:25 -03:00
{
1992-01-27 12:51:30 -04:00
char *name;
1991-04-10 16:48:25 -03:00
struct passwd *p;
1996-12-11 12:54:40 -04:00
if (!PyArg_Parse(args, "s", &name))
1991-04-10 16:48:25 -03:00
return NULL;
1992-01-27 12:51:30 -04:00
if ((p = getpwnam(name)) == NULL) {
1996-12-11 12:54:40 -04:00
PyErr_SetString(PyExc_KeyError, "getpwnam(): name not found");
1991-04-10 16:48:25 -03:00
return NULL;
}
return mkpwent(p);
}
#ifdef HAVE_GETPWENT
1998-03-03 18:03:26 -04:00
static char pwd_getpwall__doc__[] = "\
getpwall() -> list_of_entries\n\
Return a list of all available password database entries, \
in arbitrary order.\n\
See pwd.__doc__ for more on password database entries.";
1996-12-11 12:54:40 -04:00
static PyObject *
2000-07-10 10:12:27 -03:00
pwd_getpwall(PyObject *self, PyObject *args)
1991-04-10 16:48:25 -03:00
{
1996-12-11 12:54:40 -04:00
PyObject *d;
1991-04-10 16:48:25 -03:00
struct passwd *p;
1996-12-11 12:54:40 -04:00
if (!PyArg_NoArgs(args))
1991-04-10 16:48:25 -03:00
return NULL;
1996-12-11 12:54:40 -04:00
if ((d = PyList_New(0)) == NULL)
1991-04-10 16:48:25 -03:00
return NULL;
setpwent();
while ((p = getpwent()) != NULL) {
1996-12-11 12:54:40 -04:00
PyObject *v = mkpwent(p);
if (v == NULL || PyList_Append(d, v) != 0) {
Py_XDECREF(v);
Py_DECREF(d);
1991-04-10 16:48:25 -03:00
return NULL;
}
Py_DECREF(v);
1991-04-10 16:48:25 -03:00
}
endpwent();
1991-04-10 16:48:25 -03:00
return d;
}
#endif
1991-04-10 16:48:25 -03:00
1996-12-11 12:54:40 -04:00
static PyMethodDef pwd_methods[] = {
{"getpwuid", pwd_getpwuid, METH_OLDARGS, pwd_getpwuid__doc__},
{"getpwnam", pwd_getpwnam, METH_OLDARGS, pwd_getpwnam__doc__},
#ifdef HAVE_GETPWENT
{"getpwall", pwd_getpwall, METH_OLDARGS, pwd_getpwall__doc__},
#endif
1991-04-10 16:48:25 -03:00
{NULL, NULL} /* sentinel */
};
DL_EXPORT(void)
initpwd(void)
1991-04-10 16:48:25 -03:00
{
1998-03-03 18:03:26 -04:00
Py_InitModule4("pwd", pwd_methods, pwd__doc__,
(PyObject *)NULL, PYTHON_API_VERSION);
1991-04-10 16:48:25 -03:00
}