cpython/Modules/grpmodule.c

112 lines
2.5 KiB
C

/***********************************************************
Copyright (c) 2000, BeOpen.com.
Copyright (c) 1995-2000, Corporation for National Research Initiatives.
Copyright (c) 1990-1995, Stichting Mathematisch Centrum.
All rights reserved.
See the file "Misc/COPYRIGHT" for information on usage and
redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
******************************************************************/
/* UNIX group file access module */
#include "Python.h"
#include <sys/types.h>
#include <grp.h>
static PyObject *mkgrent(p)
struct group *p;
{
PyObject *v, *w;
char **member;
if ((w = PyList_New(0)) == NULL) {
return NULL;
}
for (member = p->gr_mem; *member != NULL; member++) {
PyObject *x = PyString_FromString(*member);
if (x == NULL || PyList_Append(w, x) != 0) {
Py_XDECREF(x);
Py_DECREF(w);
return NULL;
}
Py_DECREF(x);
}
v = Py_BuildValue("(sslO)",
p->gr_name,
p->gr_passwd,
#if defined(NeXT) && defined(_POSIX_SOURCE) && defined(__LITTLE_ENDIAN__)
/* Correct a bug present on Intel machines in NextStep 3.2 and 3.3;
for later versions you may have to remove this */
(long)p->gr_short_pad, /* ugh-NeXT broke the padding */
#else
(long)p->gr_gid,
#endif
w);
Py_DECREF(w);
return v;
}
static PyObject *grp_getgrgid(self, args)
PyObject *self, *args;
{
int gid;
struct group *p;
if (!PyArg_Parse((args),"i",(&gid)))
return NULL;
if ((p = getgrgid(gid)) == NULL) {
PyErr_SetString(PyExc_KeyError, "getgrgid(): gid not found");
return NULL;
}
return mkgrent(p);
}
static PyObject *grp_getgrnam(self, args)
PyObject *self, *args;
{
char *name;
struct group *p;
if (!PyArg_Parse((args),"s",(&name)))
return NULL;
if ((p = getgrnam(name)) == NULL) {
PyErr_SetString(PyExc_KeyError, "getgrnam(): name not found");
return NULL;
}
return mkgrent(p);
}
static PyObject *grp_getgrall(self, args)
PyObject *self, *args;
{
PyObject *d;
struct group *p;
if (!PyArg_NoArgs(args))
return NULL;
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);
return NULL;
}
Py_DECREF(v);
}
return d;
}
static PyMethodDef grp_methods[] = {
{"getgrgid", grp_getgrgid},
{"getgrnam", grp_getgrnam},
{"getgrall", grp_getgrall},
{NULL, NULL} /* sentinel */
};
DL_EXPORT(void)
initgrp()
{
Py_InitModule("grp", grp_methods);
}