1
2 /* UNIX shadow password file access module */
3 /* A lot of code has been taken from pwdmodule.c */
4 /* For info also see http://www.unixpapa.com/incnote/passwd.html */
5
6 #include "Python.h"
7 #include "structseq.h"
8
9 #include <sys/types.h>
10 #ifdef HAVE_SHADOW_H
11 #include <shadow.h>
12 #endif
13
14
15 PyDoc_STRVAR(spwd__doc__,
16 "This module provides access to the Unix shadow password database.\n\
17 It is available on various Unix versions.\n\
18 \n\
19 Shadow password database entries are reported as 9-tuples of type struct_spwd,\n\
20 containing the following items from the password database (see `<shadow.h>'):\n\
21 sp_namp, sp_pwdp, sp_lstchg, sp_min, sp_max, sp_warn, sp_inact, sp_expire, sp_flag.\n\
22 The sp_namp and sp_pwdp are strings, the rest are integers.\n\
23 An exception is raised if the entry asked for cannot be found.\n\
24 You have to be root to be able to use this module.");
25
26
27 #if defined(HAVE_GETSPNAM) || defined(HAVE_GETSPENT)
28
29 static PyStructSequence_Field struct_spwd_type_fields[] = {
30 {"sp_nam", "login name"},
31 {"sp_pwd", "encrypted password"},
32 {"sp_lstchg", "date of last change"},
33 {"sp_min", "min #days between changes"},
34 {"sp_max", "max #days between changes"},
35 {"sp_warn", "#days before pw expires to warn user about it"},
36 {"sp_inact", "#days after pw expires until account is blocked"},
37 {"sp_expire", "#days since 1970-01-01 until account is disabled"},
38 {"sp_flag", "reserved"},
39 {0}
40 };
41
42 PyDoc_STRVAR(struct_spwd__doc__,
43 "spwd.struct_spwd: Results from getsp*() routines.\n\n\
44 This object may be accessed either as a 9-tuple of\n\
45 (sp_nam,sp_pwd,sp_lstchg,sp_min,sp_max,sp_warn,sp_inact,sp_expire,sp_flag)\n\
46 or via the object attributes as named in the above tuple.");
47
48 static PyStructSequence_Desc struct_spwd_type_desc = {
49 "spwd.struct_spwd",
50 struct_spwd__doc__,
51 struct_spwd_type_fields,
52 9,
53 };
54
55 static int initialized;
56 static PyTypeObject StructSpwdType;
57
58
59 static void
sets(PyObject * v,int i,char * val)60 sets(PyObject *v, int i, char* val)
61 {
62 if (val)
63 PyStructSequence_SET_ITEM(v, i, PyString_FromString(val));
64 else {
65 PyStructSequence_SET_ITEM(v, i, Py_None);
66 Py_INCREF(Py_None);
67 }
68 }
69
mkspent(struct spwd * p)70 static PyObject *mkspent(struct spwd *p)
71 {
72 int setIndex = 0;
73 PyObject *v = PyStructSequence_New(&StructSpwdType);
74 if (v == NULL)
75 return NULL;
76
77 #define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyInt_FromLong((long) val))
78 #define SETS(i,val) sets(v, i, val)
79
80 SETS(setIndex++, p->sp_namp);
81 SETS(setIndex++, p->sp_pwdp);
82 SETI(setIndex++, p->sp_lstchg);
83 SETI(setIndex++, p->sp_min);
84 SETI(setIndex++, p->sp_max);
85 SETI(setIndex++, p->sp_warn);
86 SETI(setIndex++, p->sp_inact);
87 SETI(setIndex++, p->sp_expire);
88 SETI(setIndex++, p->sp_flag);
89
90 #undef SETS
91 #undef SETI
92
93 if (PyErr_Occurred()) {
94 Py_DECREF(v);
95 return NULL;
96 }
97
98 return v;
99 }
100
101 #endif /* HAVE_GETSPNAM || HAVE_GETSPENT */
102
103
104 #ifdef HAVE_GETSPNAM
105
106 PyDoc_STRVAR(spwd_getspnam__doc__,
107 "getspnam(name) -> (sp_namp, sp_pwdp, sp_lstchg, sp_min, sp_max,\n\
108 sp_warn, sp_inact, sp_expire, sp_flag)\n\
109 Return the shadow password database entry for the given user name.\n\
110 See spwd.__doc__ for more on shadow password database entries.");
111
spwd_getspnam(PyObject * self,PyObject * args)112 static PyObject* spwd_getspnam(PyObject *self, PyObject *args)
113 {
114 char *name;
115 struct spwd *p;
116 if (!PyArg_ParseTuple(args, "s:getspnam", &name))
117 return NULL;
118 if ((p = getspnam(name)) == NULL) {
119 PyErr_SetString(PyExc_KeyError, "getspnam(): name not found");
120 return NULL;
121 }
122 return mkspent(p);
123 }
124
125 #endif /* HAVE_GETSPNAM */
126
127 #ifdef HAVE_GETSPENT
128
129 PyDoc_STRVAR(spwd_getspall__doc__,
130 "getspall() -> list_of_entries\n\
131 Return a list of all available shadow password database entries, \
132 in arbitrary order.\n\
133 See spwd.__doc__ for more on shadow password database entries.");
134
135 static PyObject *
spwd_getspall(PyObject * self,PyObject * args)136 spwd_getspall(PyObject *self, PyObject *args)
137 {
138 PyObject *d;
139 struct spwd *p;
140 if ((d = PyList_New(0)) == NULL)
141 return NULL;
142 setspent();
143 while ((p = getspent()) != NULL) {
144 PyObject *v = mkspent(p);
145 if (v == NULL || PyList_Append(d, v) != 0) {
146 Py_XDECREF(v);
147 Py_DECREF(d);
148 endspent();
149 return NULL;
150 }
151 Py_DECREF(v);
152 }
153 endspent();
154 return d;
155 }
156
157 #endif /* HAVE_GETSPENT */
158
159 static PyMethodDef spwd_methods[] = {
160 #ifdef HAVE_GETSPNAM
161 {"getspnam", spwd_getspnam, METH_VARARGS, spwd_getspnam__doc__},
162 #endif
163 #ifdef HAVE_GETSPENT
164 {"getspall", spwd_getspall, METH_NOARGS, spwd_getspall__doc__},
165 #endif
166 {NULL, NULL} /* sentinel */
167 };
168
169
170 PyMODINIT_FUNC
initspwd(void)171 initspwd(void)
172 {
173 PyObject *m;
174 m=Py_InitModule3("spwd", spwd_methods, spwd__doc__);
175 if (m == NULL)
176 return;
177 if (!initialized)
178 PyStructSequence_InitType(&StructSpwdType,
179 &struct_spwd_type_desc);
180 Py_INCREF((PyObject *) &StructSpwdType);
181 PyModule_AddObject(m, "struct_spwd", (PyObject *) &StructSpwdType);
182 initialized = 1;
183 }
184