/* UNIX password file access module */ #include "Python.h" #include "structseq.h" #include #include 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__, "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\ 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__, "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 `'), 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."); static int initialized; static PyTypeObject StructPwdType; static void sets(PyObject *v, int i, const char* val) { if (val) { PyObject *o = PyUnicode_DecodeUnicodeEscape(val, strlen(val), "strict"); PyStructSequence_SET_ITEM(v, i, o); } 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++, ""); #else SETS(setIndex++, p->pw_passwd); #endif SETI(setIndex++, p->pw_uid); SETI(setIndex++, p->pw_gid); #ifdef __VMS SETS(setIndex++, ""); #else SETS(setIndex++, p->pw_gecos); #endif 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__, "getpwuid(uid) -> (pw_name,pw_passwd,pw_uid,\n\ pw_gid,pw_gecos,pw_dir,pw_shell)\n\ Return the password database entry for the given numeric user ID.\n\ 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__, "getpwnam(name) -> (pw_name,pw_passwd,pw_uid,\n\ pw_gid,pw_gecos,pw_dir,pw_shell)\n\ Return the password database entry for the given user name.\n\ 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__, "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."); 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) { #else setpwent(); while ((p = getpwent()) != NULL) { #endif 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; } #endif 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__}, #endif {NULL, NULL} /* sentinel */ }; PyMODINIT_FUNC 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; }