#include "Python.h"

#include "posixmodule.h"

#include <grp.h>

#include "clinic/grpmodule.c.h"

static PyStructSequence_Field struct_group_type_fields[] = {

{"gr_name", "group name"},

{"gr_passwd", "password"},

{"gr_gid", "group id"},

{"gr_mem", "group members"},

{0}

};

PyDoc_STRVAR(struct_group__doc__,

or via the object attributes as named in the above tuple.\n");

static PyStructSequence_Desc struct_group_type_desc = {

"grp.struct_group",

struct_group__doc__,

struct_group_type_fields,

4,

};

static int initialized;

static PyTypeObject StructGrpType;

static PyObject *

mkgrent(struct group *p)

{

int setIndex = 0;

PyObject *v = PyStructSequence_New(&StructGrpType), *w;

char **member;

if (v == NULL)

return NULL;

if ((w = PyList_New(0)) == NULL) {

Py_DECREF(v);

return NULL;

}

for (member = p->gr_mem; *member != NULL; member++) {

PyObject *x = PyUnicode_DecodeFSDefault(*member);

if (x == NULL || PyList_Append(w, x) != 0) {

Py_XDECREF(x);

Py_DECREF(w);

Py_DECREF(v);

return NULL;

}

Py_DECREF(x);

}

#define SET(i,val) PyStructSequence_SET_ITEM(v, i, val)

SET(setIndex++, PyUnicode_DecodeFSDefault(p->gr_name));

if (p->gr_passwd)

SET(setIndex++, PyUnicode_DecodeFSDefault(p->gr_passwd));

else {

SET(setIndex++, Py_None);

Py_INCREF(Py_None);

}

SET(setIndex++, _PyLong_FromGid(p->gr_gid));

SET(setIndex++, w);

#undef SET

if (PyErr_Occurred()) {

Py_DECREF(v);

return NULL;

}

return v;

}

static PyObject *

grp_getgrgid_impl(PyObject *module, PyObject *id)

{

PyObject *py_int_id;

gid_t gid;

struct group *p;

if (!_Py_Gid_Converter(id, &gid)) {

if (!PyErr_ExceptionMatches(PyExc_TypeError)) {

return NULL;

}

PyErr_Clear();

if (PyErr_WarnFormat(PyExc_DeprecationWarning, 1,

"group id must be int, not %.200",

id->ob_type->tp_name) < 0) {

return NULL;

}

py_int_id = PyNumber_Long(id);

if (!py_int_id)

return NULL;

if (!_Py_Gid_Converter(py_int_id, &gid)) {

Py_DECREF(py_int_id);

return NULL;

}

Py_DECREF(py_int_id);

}

if ((p = getgrgid(gid)) == NULL) {

PyObject *gid_obj = _PyLong_FromGid(gid);

if (gid_obj == NULL)

return NULL;

PyErr_Format(PyExc_KeyError, "getgrgid(): gid not found: %S", gid_obj);

Py_DECREF(gid_obj);

return NULL;

}

return mkgrent(p);

}

static PyObject *

grp_getgrnam_impl(PyObject *module, PyObject *name)

{

char *name_chars;

struct group *p;

PyObject *bytes, *retval = NULL;

if ((bytes = PyUnicode_EncodeFSDefault(name)) == NULL)

return NULL;

if (PyBytes_AsStringAndSize(bytes, &name_chars, NULL) == -1)

goto out;

if ((p = getgrnam(name_chars)) == NULL) {

PyErr_Format(PyExc_KeyError, "getgrnam(): name not found: %s", name_chars);

goto out;

}

retval = mkgrent(p);

out:

Py_DECREF(bytes);

return retval;

}

static PyObject *

grp_getgrall_impl(PyObject *module)

{

PyObject *d;

struct group *p;

if ((d = PyList_New(0)) == NULL)

return NULL;

setgrent();

while ((p = getgrent()) != NULL) {

PyObject *v = mkgrent(p);

if (v == NULL || PyList_Append(d, v) != 0) {

Py_XDECREF(v);

Py_DECREF(d);

endgrent();

return NULL;

}

Py_DECREF(v);

}

endgrent();

return d;

}

static PyMethodDef grp_methods[] = {

GRP_GETGRGID_METHODDEF

GRP_GETGRNAM_METHODDEF

GRP_GETGRALL_METHODDEF

{NULL, NULL}

};

PyDoc_STRVAR(grp__doc__,

complete membership information.)");

static struct PyModuleDef grpmodule = {

PyModuleDef_HEAD_INIT,

"grp",

grp__doc__,

-1,

grp_methods,

NULL,

NULL,

NULL,

NULL

};

PyInit_grp(void)

{

PyObject *m, *d;

m = PyModule_Create(&grpmodule);

if (m == NULL)

return NULL;

d = PyModule_GetDict(m);

if (!initialized) {

if (PyStructSequence_InitType2(&StructGrpType,

&struct_group_type_desc) < 0)

return NULL;

}

if (PyDict_SetItemString(d, "struct_group",

(PyObject *)&StructGrpType) < 0)

return NULL;

initialized = 1;

return m;

}