|
1 |
|
2 /* UNIX password file access module */ |
|
3 |
|
4 #include "Python.h" |
|
5 #include "structseq.h" |
|
6 |
|
7 #include <sys/types.h> |
|
8 #include <pwd.h> |
|
9 |
|
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} |
|
19 }; |
|
20 |
|
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."); |
|
26 |
|
27 static PyStructSequence_Desc struct_pwd_type_desc = { |
|
28 "pwd.struct_passwd", |
|
29 struct_passwd__doc__, |
|
30 struct_pwd_type_fields, |
|
31 7, |
|
32 }; |
|
33 |
|
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."); |
|
43 |
|
44 |
|
45 static int initialized; |
|
46 static PyTypeObject StructPwdType; |
|
47 |
|
48 static void |
|
49 sets(PyObject *v, int i, char* val) |
|
50 { |
|
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); |
|
56 } |
|
57 } |
|
58 |
|
59 static PyObject * |
|
60 mkpwent(struct passwd *p) |
|
61 { |
|
62 int setIndex = 0; |
|
63 PyObject *v = PyStructSequence_New(&StructPwdType); |
|
64 if (v == NULL) |
|
65 return NULL; |
|
66 |
|
67 #define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyInt_FromLong((long) val)) |
|
68 #define SETS(i,val) sets(v, i, val) |
|
69 |
|
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); |
|
85 |
|
86 #undef SETS |
|
87 #undef SETI |
|
88 |
|
89 if (PyErr_Occurred()) { |
|
90 Py_XDECREF(v); |
|
91 return NULL; |
|
92 } |
|
93 |
|
94 return v; |
|
95 } |
|
96 |
|
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."); |
|
102 |
|
103 static PyObject * |
|
104 pwd_getpwuid(PyObject *self, PyObject *args) |
|
105 { |
|
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; |
|
114 } |
|
115 return mkpwent(p); |
|
116 } |
|
117 |
|
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."); |
|
123 |
|
124 static PyObject * |
|
125 pwd_getpwnam(PyObject *self, PyObject *args) |
|
126 { |
|
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; |
|
135 } |
|
136 return mkpwent(p); |
|
137 } |
|
138 |
|
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."); |
|
145 |
|
146 static PyObject * |
|
147 pwd_getpwall(PyObject *self) |
|
148 { |
|
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; |
|
164 } |
|
165 Py_DECREF(v); |
|
166 } |
|
167 endpwent(); |
|
168 return d; |
|
169 } |
|
170 #endif |
|
171 |
|
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 */ |
|
180 }; |
|
181 |
|
182 PyMODINIT_FUNC |
|
183 initpwd(void) |
|
184 { |
|
185 PyObject *m; |
|
186 m = Py_InitModule3("pwd", pwd_methods, pwd__doc__); |
|
187 if (m == NULL) |
|
188 return; |
|
189 |
|
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; |
|
198 } |