1991-04-11 03:48:25 +08:00
|
|
|
|
1994-08-01 19:34:53 +08:00
|
|
|
/* UNIX password file access module */
|
1991-04-11 03:48:25 +08:00
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
#include "Python.h"
|
2002-03-01 18:38:44 +08:00
|
|
|
#include "structseq.h"
|
1991-04-11 03:48:25 +08:00
|
|
|
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <pwd.h>
|
|
|
|
|
2002-03-01 18:38:44 +08:00
|
|
|
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}
|
|
|
|
};
|
|
|
|
|
2002-06-14 04:33:02 +08:00
|
|
|
PyDoc_STRVAR(struct_passwd__doc__,
|
2002-03-01 18:38:44 +08:00
|
|
|
"pwd.struct_passwd: Results from getpw*() routines.\n\n\
|
|
|
|
This object may be accessed either as a tuple of\n\
|
|
|
|
(pw_name,pw_passwd,pw_uid,pw_gid,pw_gecos,pw_dir,pw_shell)\n\
|
2002-06-14 04:33:02 +08:00
|
|
|
or via the object attributes as named in the above tuple.");
|
2002-03-01 18:38:44 +08:00
|
|
|
|
|
|
|
static PyStructSequence_Desc struct_pwd_type_desc = {
|
|
|
|
"pwd.struct_passwd",
|
|
|
|
struct_passwd__doc__,
|
|
|
|
struct_pwd_type_fields,
|
|
|
|
7,
|
|
|
|
};
|
|
|
|
|
2002-06-14 04:33:02 +08:00
|
|
|
PyDoc_STRVAR(pwd__doc__,
|
|
|
|
"This module provides access to the Unix password database.\n\
|
1998-03-04 06:03:26 +08:00
|
|
|
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\
|
2002-06-14 04:33:02 +08:00
|
|
|
exception is raised if the entry asked for cannot be found.");
|
1998-03-04 06:03:26 +08:00
|
|
|
|
|
|
|
|
2006-04-21 18:40:58 +08:00
|
|
|
static int initialized;
|
2002-03-01 18:38:44 +08:00
|
|
|
static PyTypeObject StructPwdType;
|
|
|
|
|
2002-09-17 17:34:06 +08:00
|
|
|
static void
|
2007-08-25 07:26:23 +08:00
|
|
|
sets(PyObject *v, int i, const char* val)
|
2002-09-17 17:34:06 +08:00
|
|
|
{
|
2007-10-27 13:40:06 +08:00
|
|
|
if (val) {
|
|
|
|
PyObject *o =
|
|
|
|
PyUnicode_DecodeUnicodeEscape(val, strlen(val), "strict");
|
|
|
|
PyStructSequence_SET_ITEM(v, i, o);
|
|
|
|
}
|
2002-09-17 17:34:06 +08:00
|
|
|
else {
|
|
|
|
PyStructSequence_SET_ITEM(v, i, Py_None);
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
static PyObject *
|
2000-07-10 21:12:27 +08:00
|
|
|
mkpwent(struct passwd *p)
|
1991-04-11 03:48:25 +08:00
|
|
|
{
|
2002-03-01 18:38:44 +08:00
|
|
|
int setIndex = 0;
|
|
|
|
PyObject *v = PyStructSequence_New(&StructPwdType);
|
|
|
|
if (v == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
2007-12-02 22:31:20 +08:00
|
|
|
#define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyLong_FromLong((long) val))
|
2002-09-17 17:34:06 +08:00
|
|
|
#define SETS(i,val) sets(v, i, val)
|
2002-03-01 18:38:44 +08:00
|
|
|
|
|
|
|
SETS(setIndex++, p->pw_name);
|
2002-12-06 20:48:53 +08:00
|
|
|
#ifdef __VMS
|
|
|
|
SETS(setIndex++, "");
|
|
|
|
#else
|
2002-03-01 18:38:44 +08:00
|
|
|
SETS(setIndex++, p->pw_passwd);
|
2002-12-06 20:48:53 +08:00
|
|
|
#endif
|
2002-03-01 18:38:44 +08:00
|
|
|
SETI(setIndex++, p->pw_uid);
|
|
|
|
SETI(setIndex++, p->pw_gid);
|
2002-12-06 20:48:53 +08:00
|
|
|
#ifdef __VMS
|
|
|
|
SETS(setIndex++, "");
|
|
|
|
#else
|
2002-03-01 18:38:44 +08:00
|
|
|
SETS(setIndex++, p->pw_gecos);
|
2002-12-06 20:48:53 +08:00
|
|
|
#endif
|
2002-03-01 18:38:44 +08:00
|
|
|
SETS(setIndex++, p->pw_dir);
|
|
|
|
SETS(setIndex++, p->pw_shell);
|
|
|
|
|
|
|
|
#undef SETS
|
|
|
|
#undef SETI
|
|
|
|
|
|
|
|
if (PyErr_Occurred()) {
|
|
|
|
Py_XDECREF(v);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return v;
|
1991-04-11 03:48:25 +08:00
|
|
|
}
|
|
|
|
|
2002-06-14 04:33:02 +08:00
|
|
|
PyDoc_STRVAR(pwd_getpwuid__doc__,
|
|
|
|
"getpwuid(uid) -> (pw_name,pw_passwd,pw_uid,\n\
|
|
|
|
pw_gid,pw_gecos,pw_dir,pw_shell)\n\
|
1998-03-04 06:03:26 +08:00
|
|
|
Return the password database entry for the given numeric user ID.\n\
|
2002-06-14 04:33:02 +08:00
|
|
|
See pwd.__doc__ for more on password database entries.");
|
1998-03-04 06:03:26 +08:00
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
static PyObject *
|
2000-07-10 21:12:27 +08:00
|
|
|
pwd_getpwuid(PyObject *self, PyObject *args)
|
1991-04-11 03:48:25 +08:00
|
|
|
{
|
2005-09-15 02:09:42 +08:00
|
|
|
unsigned int uid;
|
1991-04-11 03:48:25 +08:00
|
|
|
struct passwd *p;
|
2005-09-15 02:09:42 +08:00
|
|
|
if (!PyArg_ParseTuple(args, "I:getpwuid", &uid))
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
|
|
|
if ((p = getpwuid(uid)) == NULL) {
|
2004-01-21 05:07:23 +08:00
|
|
|
PyErr_Format(PyExc_KeyError,
|
|
|
|
"getpwuid(): uid not found: %d", uid);
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return mkpwent(p);
|
|
|
|
}
|
|
|
|
|
2002-06-14 04:33:02 +08:00
|
|
|
PyDoc_STRVAR(pwd_getpwnam__doc__,
|
|
|
|
"getpwnam(name) -> (pw_name,pw_passwd,pw_uid,\n\
|
|
|
|
pw_gid,pw_gecos,pw_dir,pw_shell)\n\
|
1998-03-04 06:03:26 +08:00
|
|
|
Return the password database entry for the given user name.\n\
|
2002-06-14 04:33:02 +08:00
|
|
|
See pwd.__doc__ for more on password database entries.");
|
1998-03-04 06:03:26 +08:00
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
static PyObject *
|
2000-07-10 21:12:27 +08:00
|
|
|
pwd_getpwnam(PyObject *self, PyObject *args)
|
1991-04-11 03:48:25 +08:00
|
|
|
{
|
1992-01-28 00:51:30 +08:00
|
|
|
char *name;
|
1991-04-11 03:48:25 +08:00
|
|
|
struct passwd *p;
|
2002-03-31 23:27:00 +08:00
|
|
|
if (!PyArg_ParseTuple(args, "s:getpwnam", &name))
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
1992-01-28 00:51:30 +08:00
|
|
|
if ((p = getpwnam(name)) == NULL) {
|
2004-01-21 05:07:23 +08:00
|
|
|
PyErr_Format(PyExc_KeyError,
|
|
|
|
"getpwnam(): name not found: %s", name);
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return mkpwent(p);
|
|
|
|
}
|
|
|
|
|
1997-08-23 04:42:00 +08:00
|
|
|
#ifdef HAVE_GETPWENT
|
2002-06-14 04:33:02 +08:00
|
|
|
PyDoc_STRVAR(pwd_getpwall__doc__,
|
|
|
|
"getpwall() -> list_of_entries\n\
|
1998-03-04 06:03:26 +08:00
|
|
|
Return a list of all available password database entries, \
|
|
|
|
in arbitrary order.\n\
|
2002-06-14 04:33:02 +08:00
|
|
|
See pwd.__doc__ for more on password database entries.");
|
1998-03-04 06:03:26 +08:00
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
static PyObject *
|
2002-03-26 04:46:46 +08:00
|
|
|
pwd_getpwall(PyObject *self)
|
1991-04-11 03:48:25 +08:00
|
|
|
{
|
1996-12-12 00:54:40 +08:00
|
|
|
PyObject *d;
|
1991-04-11 03:48:25 +08:00
|
|
|
struct passwd *p;
|
1996-12-12 00:54:40 +08:00
|
|
|
if ((d = PyList_New(0)) == NULL)
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
2002-03-03 10:59:16 +08:00
|
|
|
#if defined(PYOS_OS2) && defined(PYCC_GCC)
|
|
|
|
if ((p = getpwuid(0)) != NULL) {
|
|
|
|
#else
|
1991-04-11 03:48:25 +08:00
|
|
|
setpwent();
|
|
|
|
while ((p = getpwent()) != NULL) {
|
2002-03-03 10:59:16 +08:00
|
|
|
#endif
|
1996-12-12 00:54:40 +08:00
|
|
|
PyObject *v = mkpwent(p);
|
|
|
|
if (v == NULL || PyList_Append(d, v) != 0) {
|
|
|
|
Py_XDECREF(v);
|
|
|
|
Py_DECREF(d);
|
1991-04-11 03:48:25 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-01-10 06:22:05 +08:00
|
|
|
Py_DECREF(v);
|
1991-04-11 03:48:25 +08:00
|
|
|
}
|
2001-03-11 11:03:07 +08:00
|
|
|
endpwent();
|
1991-04-11 03:48:25 +08:00
|
|
|
return d;
|
|
|
|
}
|
1997-08-23 04:42:00 +08:00
|
|
|
#endif
|
1991-04-11 03:48:25 +08:00
|
|
|
|
1996-12-12 00:54:40 +08:00
|
|
|
static PyMethodDef pwd_methods[] = {
|
2002-03-31 23:27:00 +08:00
|
|
|
{"getpwuid", pwd_getpwuid, METH_VARARGS, pwd_getpwuid__doc__},
|
|
|
|
{"getpwnam", pwd_getpwnam, METH_VARARGS, pwd_getpwnam__doc__},
|
1997-08-23 04:42:00 +08:00
|
|
|
#ifdef HAVE_GETPWENT
|
2002-03-30 03:58:25 +08:00
|
|
|
{"getpwall", (PyCFunction)pwd_getpwall,
|
|
|
|
METH_NOARGS, pwd_getpwall__doc__},
|
1997-08-23 04:42:00 +08:00
|
|
|
#endif
|
1991-04-11 03:48:25 +08:00
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
};
|
|
|
|
|
2008-06-11 13:26:20 +08:00
|
|
|
static struct PyModuleDef pwdmodule = {
|
|
|
|
PyModuleDef_HEAD_INIT,
|
|
|
|
"pwd",
|
|
|
|
pwd__doc__,
|
|
|
|
-1,
|
|
|
|
pwd_methods,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2002-08-02 10:27:13 +08:00
|
|
|
PyMODINIT_FUNC
|
2008-06-11 13:26:20 +08:00
|
|
|
PyInit_pwd(void)
|
1991-04-11 03:48:25 +08:00
|
|
|
{
|
2002-04-16 00:29:00 +08:00
|
|
|
PyObject *m;
|
2008-06-11 13:26:20 +08:00
|
|
|
m = PyModule_Create(&pwdmodule);
|
2006-01-19 14:09:39 +08:00
|
|
|
if (m == NULL)
|
2008-06-11 13:26:20 +08:00
|
|
|
return NULL;
|
2002-04-14 05:07:45 +08:00
|
|
|
|
2006-04-21 18:40:58 +08:00
|
|
|
if (!initialized)
|
|
|
|
PyStructSequence_InitType(&StructPwdType,
|
|
|
|
&struct_pwd_type_desc);
|
2002-04-14 05:07:45 +08:00
|
|
|
Py_INCREF((PyObject *) &StructPwdType);
|
2003-10-24 07:54:57 +08:00
|
|
|
PyModule_AddObject(m, "struct_passwd", (PyObject *) &StructPwdType);
|
|
|
|
/* And for b/w compatibility (this was defined by mistake): */
|
2002-04-14 05:07:45 +08:00
|
|
|
PyModule_AddObject(m, "struct_pwent", (PyObject *) &StructPwdType);
|
2006-04-21 18:40:58 +08:00
|
|
|
initialized = 1;
|
2008-06-11 13:26:20 +08:00
|
|
|
return m;
|
1991-04-11 03:48:25 +08:00
|
|
|
}
|