Added section about adding contextual information to log output.
[python.git] / Modules / pwdmodule.c
blob9e01f4875768fd957e7045a8d2c895875b1d8c76
2 /* UNIX password file access module */
4 #include "Python.h"
5 #include "structseq.h"
7 #include <sys/types.h>
8 #include <pwd.h>
10 static PyStructSequence_Field struct_pwd_type_fields[] = {
11 {"pw_name", "user name"},
12 {"pw_passwd", "password"},
13 {"pw_uid", "user id"},
14 {"pw_gid", "group id"},
15 {"pw_gecos", "real name"},
16 {"pw_dir", "home directory"},
17 {"pw_shell", "shell program"},
18 {0}
21 PyDoc_STRVAR(struct_passwd__doc__,
22 "pwd.struct_passwd: Results from getpw*() routines.\n\n\
23 This object may be accessed either as a tuple of\n\
24 (pw_name,pw_passwd,pw_uid,pw_gid,pw_gecos,pw_dir,pw_shell)\n\
25 or via the object attributes as named in the above tuple.");
27 static PyStructSequence_Desc struct_pwd_type_desc = {
28 "pwd.struct_passwd",
29 struct_passwd__doc__,
30 struct_pwd_type_fields,
34 PyDoc_STRVAR(pwd__doc__,
35 "This module provides access to the Unix password database.\n\
36 It is available on all Unix versions.\n\
37 \n\
38 Password database entries are reported as 7-tuples containing the following\n\
39 items from the password database (see `<pwd.h>'), in order:\n\
40 pw_name, pw_passwd, pw_uid, pw_gid, pw_gecos, pw_dir, pw_shell.\n\
41 The uid and gid items are integers, all others are strings. An\n\
42 exception is raised if the entry asked for cannot be found.");
45 static int initialized;
46 static PyTypeObject StructPwdType;
48 static void
49 sets(PyObject *v, int i, char* val)
51 if (val)
52 PyStructSequence_SET_ITEM(v, i, PyString_FromString(val));
53 else {
54 PyStructSequence_SET_ITEM(v, i, Py_None);
55 Py_INCREF(Py_None);
59 static PyObject *
60 mkpwent(struct passwd *p)
62 int setIndex = 0;
63 PyObject *v = PyStructSequence_New(&StructPwdType);
64 if (v == NULL)
65 return NULL;
67 #define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyInt_FromLong((long) val))
68 #define SETS(i,val) sets(v, i, val)
70 SETS(setIndex++, p->pw_name);
71 #ifdef __VMS
72 SETS(setIndex++, "");
73 #else
74 SETS(setIndex++, p->pw_passwd);
75 #endif
76 SETI(setIndex++, p->pw_uid);
77 SETI(setIndex++, p->pw_gid);
78 #ifdef __VMS
79 SETS(setIndex++, "");
80 #else
81 SETS(setIndex++, p->pw_gecos);
82 #endif
83 SETS(setIndex++, p->pw_dir);
84 SETS(setIndex++, p->pw_shell);
86 #undef SETS
87 #undef SETI
89 if (PyErr_Occurred()) {
90 Py_XDECREF(v);
91 return NULL;
94 return v;
97 PyDoc_STRVAR(pwd_getpwuid__doc__,
98 "getpwuid(uid) -> (pw_name,pw_passwd,pw_uid,\n\
99 pw_gid,pw_gecos,pw_dir,pw_shell)\n\
100 Return the password database entry for the given numeric user ID.\n\
101 See pwd.__doc__ for more on password database entries.");
103 static PyObject *
104 pwd_getpwuid(PyObject *self, PyObject *args)
106 unsigned int uid;
107 struct passwd *p;
108 if (!PyArg_ParseTuple(args, "I:getpwuid", &uid))
109 return NULL;
110 if ((p = getpwuid(uid)) == NULL) {
111 PyErr_Format(PyExc_KeyError,
112 "getpwuid(): uid not found: %d", uid);
113 return NULL;
115 return mkpwent(p);
118 PyDoc_STRVAR(pwd_getpwnam__doc__,
119 "getpwnam(name) -> (pw_name,pw_passwd,pw_uid,\n\
120 pw_gid,pw_gecos,pw_dir,pw_shell)\n\
121 Return the password database entry for the given user name.\n\
122 See pwd.__doc__ for more on password database entries.");
124 static PyObject *
125 pwd_getpwnam(PyObject *self, PyObject *args)
127 char *name;
128 struct passwd *p;
129 if (!PyArg_ParseTuple(args, "s:getpwnam", &name))
130 return NULL;
131 if ((p = getpwnam(name)) == NULL) {
132 PyErr_Format(PyExc_KeyError,
133 "getpwnam(): name not found: %s", name);
134 return NULL;
136 return mkpwent(p);
139 #ifdef HAVE_GETPWENT
140 PyDoc_STRVAR(pwd_getpwall__doc__,
141 "getpwall() -> list_of_entries\n\
142 Return a list of all available password database entries, \
143 in arbitrary order.\n\
144 See pwd.__doc__ for more on password database entries.");
146 static PyObject *
147 pwd_getpwall(PyObject *self)
149 PyObject *d;
150 struct passwd *p;
151 if ((d = PyList_New(0)) == NULL)
152 return NULL;
153 #if defined(PYOS_OS2) && defined(PYCC_GCC)
154 if ((p = getpwuid(0)) != NULL) {
155 #else
156 setpwent();
157 while ((p = getpwent()) != NULL) {
158 #endif
159 PyObject *v = mkpwent(p);
160 if (v == NULL || PyList_Append(d, v) != 0) {
161 Py_XDECREF(v);
162 Py_DECREF(d);
163 return NULL;
165 Py_DECREF(v);
167 endpwent();
168 return d;
170 #endif
172 static PyMethodDef pwd_methods[] = {
173 {"getpwuid", pwd_getpwuid, METH_VARARGS, pwd_getpwuid__doc__},
174 {"getpwnam", pwd_getpwnam, METH_VARARGS, pwd_getpwnam__doc__},
175 #ifdef HAVE_GETPWENT
176 {"getpwall", (PyCFunction)pwd_getpwall,
177 METH_NOARGS, pwd_getpwall__doc__},
178 #endif
179 {NULL, NULL} /* sentinel */
182 PyMODINIT_FUNC
183 initpwd(void)
185 PyObject *m;
186 m = Py_InitModule3("pwd", pwd_methods, pwd__doc__);
187 if (m == NULL)
188 return;
190 if (!initialized)
191 PyStructSequence_InitType(&StructPwdType,
192 &struct_pwd_type_desc);
193 Py_INCREF((PyObject *) &StructPwdType);
194 PyModule_AddObject(m, "struct_passwd", (PyObject *) &StructPwdType);
195 /* And for b/w compatibility (this was defined by mistake): */
196 PyModule_AddObject(m, "struct_pwent", (PyObject *) &StructPwdType);
197 initialized = 1;