Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 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" |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 7 | |
| 8 | #include <sys/types.h> |
| 9 | #ifdef HAVE_SHADOW_H |
| 10 | #include <shadow.h> |
| 11 | #endif |
| 12 | |
Brett Cannon | 52d67ef | 2014-08-22 14:01:56 -0400 | [diff] [blame] | 13 | #include "clinic/spwdmodule.c.h" |
| 14 | |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 15 | /*[clinic input] |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 16 | module spwd |
| 17 | [clinic start generated code]*/ |
Serhiy Storchaka | 1009bf1 | 2015-04-03 23:53:51 +0300 | [diff] [blame] | 18 | /*[clinic end generated code: output=da39a3ee5e6b4b0d input=c0b841b90a6a07ce]*/ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 19 | |
| 20 | PyDoc_STRVAR(spwd__doc__, |
| 21 | "This module provides access to the Unix shadow password database.\n\ |
| 22 | It is available on various Unix versions.\n\ |
| 23 | \n\ |
| 24 | Shadow password database entries are reported as 9-tuples of type struct_spwd,\n\ |
| 25 | containing the following items from the password database (see `<shadow.h>'):\n\ |
| 26 | sp_namp, sp_pwdp, sp_lstchg, sp_min, sp_max, sp_warn, sp_inact, sp_expire, sp_flag.\n\ |
| 27 | The sp_namp and sp_pwdp are strings, the rest are integers.\n\ |
| 28 | An exception is raised if the entry asked for cannot be found.\n\ |
| 29 | You have to be root to be able to use this module."); |
| 30 | |
| 31 | |
| 32 | #if defined(HAVE_GETSPNAM) || defined(HAVE_GETSPENT) |
| 33 | |
| 34 | static PyStructSequence_Field struct_spwd_type_fields[] = { |
R David Murray | bd90d09 | 2013-11-03 19:54:05 -0500 | [diff] [blame] | 35 | {"sp_namp", "login name"}, |
| 36 | {"sp_pwdp", "encrypted password"}, |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 37 | {"sp_lstchg", "date of last change"}, |
| 38 | {"sp_min", "min #days between changes"}, |
| 39 | {"sp_max", "max #days between changes"}, |
| 40 | {"sp_warn", "#days before pw expires to warn user about it"}, |
R David Murray | bd90d09 | 2013-11-03 19:54:05 -0500 | [diff] [blame] | 41 | {"sp_inact", "#days after pw expires until account is disabled"}, |
| 42 | {"sp_expire", "#days since 1970-01-01 when account expires"}, |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 43 | {"sp_flag", "reserved"}, |
R David Murray | bd90d09 | 2013-11-03 19:54:05 -0500 | [diff] [blame] | 44 | {"sp_nam", "login name; deprecated"}, /* Backward compatibility */ |
| 45 | {"sp_pwd", "encrypted password; deprecated"}, /* Backward compatibility */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 46 | {0} |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 47 | }; |
| 48 | |
| 49 | PyDoc_STRVAR(struct_spwd__doc__, |
| 50 | "spwd.struct_spwd: Results from getsp*() routines.\n\n\ |
| 51 | This object may be accessed either as a 9-tuple of\n\ |
R David Murray | bd90d09 | 2013-11-03 19:54:05 -0500 | [diff] [blame] | 52 | (sp_namp,sp_pwdp,sp_lstchg,sp_min,sp_max,sp_warn,sp_inact,sp_expire,sp_flag)\n\ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 53 | or via the object attributes as named in the above tuple."); |
| 54 | |
| 55 | static PyStructSequence_Desc struct_spwd_type_desc = { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 56 | "spwd.struct_spwd", |
| 57 | struct_spwd__doc__, |
| 58 | struct_spwd_type_fields, |
| 59 | 9, |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 60 | }; |
| 61 | |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 62 | static int initialized; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 63 | static PyTypeObject StructSpwdType; |
| 64 | |
| 65 | |
| 66 | static void |
Neal Norwitz | eb8b3a6 | 2007-08-24 23:26:23 +0000 | [diff] [blame] | 67 | sets(PyObject *v, int i, const char* val) |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 68 | { |
Martin v. Löwis | b6a748b | 2009-05-29 15:23:17 +0000 | [diff] [blame] | 69 | if (val) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 70 | PyObject *o = PyUnicode_DecodeFSDefault(val); |
| 71 | PyStructSequence_SET_ITEM(v, i, o); |
Martin v. Löwis | b6a748b | 2009-05-29 15:23:17 +0000 | [diff] [blame] | 72 | } else { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 73 | PyStructSequence_SET_ITEM(v, i, Py_None); |
| 74 | Py_INCREF(Py_None); |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 75 | } |
| 76 | } |
| 77 | |
| 78 | static PyObject *mkspent(struct spwd *p) |
| 79 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 80 | int setIndex = 0; |
| 81 | PyObject *v = PyStructSequence_New(&StructSpwdType); |
| 82 | if (v == NULL) |
| 83 | return NULL; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 84 | |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 85 | #define SETI(i,val) PyStructSequence_SET_ITEM(v, i, PyLong_FromLong((long) val)) |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 86 | #define SETS(i,val) sets(v, i, val) |
| 87 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 88 | SETS(setIndex++, p->sp_namp); |
| 89 | SETS(setIndex++, p->sp_pwdp); |
| 90 | SETI(setIndex++, p->sp_lstchg); |
| 91 | SETI(setIndex++, p->sp_min); |
| 92 | SETI(setIndex++, p->sp_max); |
| 93 | SETI(setIndex++, p->sp_warn); |
| 94 | SETI(setIndex++, p->sp_inact); |
| 95 | SETI(setIndex++, p->sp_expire); |
| 96 | SETI(setIndex++, p->sp_flag); |
R David Murray | bd90d09 | 2013-11-03 19:54:05 -0500 | [diff] [blame] | 97 | SETS(setIndex++, p->sp_namp); /* Backward compatibility for sp_nam */ |
| 98 | SETS(setIndex++, p->sp_pwdp); /* Backward compatibility for sp_pwd */ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 99 | |
| 100 | #undef SETS |
| 101 | #undef SETI |
| 102 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 103 | if (PyErr_Occurred()) { |
| 104 | Py_DECREF(v); |
| 105 | return NULL; |
| 106 | } |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 107 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 108 | return v; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 109 | } |
| 110 | |
| 111 | #endif /* HAVE_GETSPNAM || HAVE_GETSPENT */ |
| 112 | |
| 113 | |
| 114 | #ifdef HAVE_GETSPNAM |
| 115 | |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 116 | /*[clinic input] |
| 117 | spwd.getspnam |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 118 | |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 119 | arg: unicode |
| 120 | / |
| 121 | |
| 122 | Return the shadow password database entry for the given user name. |
| 123 | |
| 124 | See `help(spwd)` for more on shadow password database entries. |
| 125 | [clinic start generated code]*/ |
| 126 | |
| 127 | static PyObject * |
Serhiy Storchaka | 1a2b24f | 2016-07-07 17:35:15 +0300 | [diff] [blame] | 128 | spwd_getspnam_impl(PyObject *module, PyObject *arg) |
| 129 | /*[clinic end generated code: output=701250cf57dc6ebe input=dd89429e6167a00f]*/ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 130 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 131 | char *name; |
| 132 | struct spwd *p; |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 133 | PyObject *bytes, *retval = NULL; |
Martin v. Löwis | b6a748b | 2009-05-29 15:23:17 +0000 | [diff] [blame] | 134 | |
Victor Stinner | ae6265f | 2010-05-15 16:27:27 +0000 | [diff] [blame] | 135 | if ((bytes = PyUnicode_EncodeFSDefault(arg)) == NULL) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 136 | return NULL; |
Serhiy Storchaka | f7eae0a | 2017-06-28 08:30:06 +0300 | [diff] [blame] | 137 | /* check for embedded null bytes */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 138 | if (PyBytes_AsStringAndSize(bytes, &name, NULL) == -1) |
| 139 | goto out; |
| 140 | if ((p = getspnam(name)) == NULL) { |
Berker Peksag | 3c3d7f4 | 2016-03-19 11:44:17 +0200 | [diff] [blame] | 141 | if (errno != 0) |
| 142 | PyErr_SetFromErrno(PyExc_OSError); |
| 143 | else |
| 144 | PyErr_SetString(PyExc_KeyError, "getspnam(): name not found"); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 145 | goto out; |
| 146 | } |
| 147 | retval = mkspent(p); |
Martin v. Löwis | b6a748b | 2009-05-29 15:23:17 +0000 | [diff] [blame] | 148 | out: |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 149 | Py_DECREF(bytes); |
| 150 | return retval; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 151 | } |
| 152 | |
Neal Norwitz | 6267996 | 2005-01-24 23:33:50 +0000 | [diff] [blame] | 153 | #endif /* HAVE_GETSPNAM */ |
| 154 | |
| 155 | #ifdef HAVE_GETSPENT |
| 156 | |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 157 | /*[clinic input] |
| 158 | spwd.getspall |
| 159 | |
| 160 | Return a list of all available shadow password database entries, in arbitrary order. |
| 161 | |
| 162 | See `help(spwd)` for more on shadow password database entries. |
| 163 | [clinic start generated code]*/ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 164 | |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 165 | static PyObject * |
Serhiy Storchaka | 1a2b24f | 2016-07-07 17:35:15 +0300 | [diff] [blame] | 166 | spwd_getspall_impl(PyObject *module) |
| 167 | /*[clinic end generated code: output=4fda298d6bf6d057 input=b2c84b7857d622bd]*/ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 168 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 169 | PyObject *d; |
| 170 | struct spwd *p; |
| 171 | if ((d = PyList_New(0)) == NULL) |
| 172 | return NULL; |
| 173 | setspent(); |
| 174 | while ((p = getspent()) != NULL) { |
| 175 | PyObject *v = mkspent(p); |
| 176 | if (v == NULL || PyList_Append(d, v) != 0) { |
| 177 | Py_XDECREF(v); |
| 178 | Py_DECREF(d); |
| 179 | endspent(); |
| 180 | return NULL; |
| 181 | } |
| 182 | Py_DECREF(v); |
| 183 | } |
| 184 | endspent(); |
| 185 | return d; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 186 | } |
| 187 | |
| 188 | #endif /* HAVE_GETSPENT */ |
| 189 | |
| 190 | static PyMethodDef spwd_methods[] = { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 191 | #ifdef HAVE_GETSPNAM |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 192 | SPWD_GETSPNAM_METHODDEF |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 193 | #endif |
| 194 | #ifdef HAVE_GETSPENT |
Brett Cannon | 20cf6dd | 2014-08-22 13:59:24 -0400 | [diff] [blame] | 195 | SPWD_GETSPALL_METHODDEF |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 196 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 197 | {NULL, NULL} /* sentinel */ |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 198 | }; |
| 199 | |
| 200 | |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 201 | |
| 202 | static struct PyModuleDef spwdmodule = { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 203 | PyModuleDef_HEAD_INIT, |
| 204 | "spwd", |
| 205 | spwd__doc__, |
| 206 | -1, |
| 207 | spwd_methods, |
| 208 | NULL, |
| 209 | NULL, |
| 210 | NULL, |
| 211 | NULL |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 212 | }; |
| 213 | |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 214 | PyMODINIT_FUNC |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 215 | PyInit_spwd(void) |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 216 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 217 | PyObject *m; |
| 218 | m=PyModule_Create(&spwdmodule); |
| 219 | if (m == NULL) |
| 220 | return NULL; |
Victor Stinner | 1c8f059 | 2013-07-22 22:24:54 +0200 | [diff] [blame] | 221 | if (!initialized) { |
| 222 | if (PyStructSequence_InitType2(&StructSpwdType, |
| 223 | &struct_spwd_type_desc) < 0) |
| 224 | return NULL; |
| 225 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 226 | Py_INCREF((PyObject *) &StructSpwdType); |
| 227 | PyModule_AddObject(m, "struct_spwd", (PyObject *) &StructSpwdType); |
| 228 | initialized = 1; |
| 229 | return m; |
Martin v. Löwis | c300175 | 2005-01-23 09:27:24 +0000 | [diff] [blame] | 230 | } |