Browse Source

refactor group file access code

this allows getgrnam and getgrgid to share code with the _r versions
in preparation for alternate backend support.
master
Josiah Worcester 11 years ago
committed by Rich Felker
parent
commit
7c5f0a5212
  1. 4
      src/passwd/fgetgrent.c
  2. 32
      src/passwd/getgr_a.c
  3. 20
      src/passwd/getgr_r.c
  4. 32
      src/passwd/getgrent.c
  5. 9
      src/passwd/getgrent_a.c
  6. 3
      src/passwd/pwf.h

4
src/passwd/fgetgrent.c

@ -5,6 +5,8 @@ struct group *fgetgrent(FILE *f)
{ {
static char *line, **mem; static char *line, **mem;
static struct group gr; static struct group gr;
struct group *res;
size_t size=0, nmem=0; size_t size=0, nmem=0;
return __getgrent_a(f, &gr, &line, &size, &mem, &nmem); __getgrent_a(f, &gr, &line, &size, &mem, &nmem, &res);
return res;
} }

32
src/passwd/getgr_a.c

@ -0,0 +1,32 @@
#include <pthread.h>
#include "pwf.h"
int __getgr_a(const char *name, gid_t gid, struct group *gr, char **buf, size_t *size, char ***mem, size_t *nmem, struct group **res)
{
FILE *f;
int rv = 0;
int cs;
*res = 0;
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
f = fopen("/etc/group", "rbe");
if (!f) {
rv = errno;
goto done;
}
while (!(rv = __getgrent_a(f, gr, buf, size, mem, nmem, res)) && *res) {
if (name && !strcmp(name, (*res)->gr_name)
|| !name && (*res)->gr_gid == gid) {
break;
}
}
fclose(f);
done:
pthread_setcancelstate(cs, 0);
if (rv) errno = rv;
return rv;
}

20
src/passwd/getgr_r.c

@ -5,7 +5,6 @@
static int getgr_r(const char *name, gid_t gid, struct group *gr, char *buf, size_t size, struct group **res) static int getgr_r(const char *name, gid_t gid, struct group *gr, char *buf, size_t size, struct group **res)
{ {
FILE *f;
char *line = 0; char *line = 0;
size_t len = 0; size_t len = 0;
char **mem = 0; char **mem = 0;
@ -16,21 +15,12 @@ static int getgr_r(const char *name, gid_t gid, struct group *gr, char *buf, siz
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs); pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
f = fopen("/etc/group", "rbe"); rv = __getgr_a(name, gid, gr, &line, &len, &mem, &nmem, res);
if (!f) { if (!rv && size < len + (nmem+1)*sizeof(char *) + 32) {
rv = errno;
goto done;
}
*res = 0; *res = 0;
while (__getgrent_a(f, gr, &line, &len, &mem, &nmem)) {
if (name && !strcmp(name, gr->gr_name)
|| !name && gr->gr_gid == gid) {
if (size < len + (nmem+1)*sizeof(char *) + 32) {
rv = ERANGE; rv = ERANGE;
break;
} }
*res = gr; if (!rv) {
buf += (16-(uintptr_t)buf)%16; buf += (16-(uintptr_t)buf)%16;
gr->gr_mem = (void *)buf; gr->gr_mem = (void *)buf;
buf += (nmem+1)*sizeof(char *); buf += (nmem+1)*sizeof(char *);
@ -40,13 +30,9 @@ static int getgr_r(const char *name, gid_t gid, struct group *gr, char *buf, siz
for (i=0; mem[i]; i++) for (i=0; mem[i]; i++)
gr->gr_mem[i] = mem[i]-line+buf; gr->gr_mem[i] = mem[i]-line+buf;
gr->gr_mem[i] = 0; gr->gr_mem[i] = 0;
break;
}
} }
free(mem); free(mem);
free(line); free(line);
fclose(f);
done:
pthread_setcancelstate(cs, 0); pthread_setcancelstate(cs, 0);
return rv; return rv;
} }

32
src/passwd/getgrent.c

@ -1,6 +1,8 @@
#include "pwf.h" #include "pwf.h"
static FILE *f; static FILE *f;
static char *line, **mem;
static struct group gr;
void setgrent() void setgrent()
{ {
@ -12,34 +14,26 @@ weak_alias(setgrent, endgrent);
struct group *getgrent() struct group *getgrent()
{ {
static char *line, **mem; struct group *res;
static struct group gr;
size_t size=0, nmem=0; size_t size=0, nmem=0;
if (!f) f = fopen("/etc/group", "rbe"); if (!f) f = fopen("/etc/group", "rbe");
if (!f) return 0; if (!f) return 0;
return __getgrent_a(f, &gr, &line, &size, &mem, &nmem); __getgrent_a(f, &gr, &line, &size, &mem, &nmem, &res);
return res;
} }
struct group *getgrgid(gid_t gid) struct group *getgrgid(gid_t gid)
{ {
struct group *gr; struct group *res;
int errno_saved; size_t size=0, nmem=0;
setgrent(); __getgr_a(0, gid, &gr, &line, &size, &mem, &nmem, &res);
while ((gr=getgrent()) && gr->gr_gid != gid); return res;
errno_saved = errno;
endgrent();
errno = errno_saved;
return gr;
} }
struct group *getgrnam(const char *name) struct group *getgrnam(const char *name)
{ {
struct group *gr; struct group *res;
int errno_saved; size_t size=0, nmem=0;
setgrent(); __getgr_a(name, 0, &gr, &line, &size, &mem, &nmem, &res);
while ((gr=getgrent()) && strcmp(gr->gr_name, name)); return res;
errno_saved = errno;
endgrent();
errno = errno_saved;
return gr;
} }

9
src/passwd/getgrent_a.c

@ -8,15 +8,17 @@ static unsigned atou(char **s)
return x; return x;
} }
struct group *__getgrent_a(FILE *f, struct group *gr, char **line, size_t *size, char ***mem, size_t *nmem) int __getgrent_a(FILE *f, struct group *gr, char **line, size_t *size, char ***mem, size_t *nmem, struct group **res)
{ {
ssize_t l; ssize_t l;
char *s, *mems; char *s, *mems;
size_t i; size_t i;
int rv = 0;
int cs; int cs;
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs); pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
for (;;) { for (;;) {
if ((l=getline(line, size, f)) < 0) { if ((l=getline(line, size, f)) < 0) {
rv = errno;
free(*line); free(*line);
*line = 0; *line = 0;
gr = 0; gr = 0;
@ -43,6 +45,7 @@ struct group *__getgrent_a(FILE *f, struct group *gr, char **line, size_t *size,
free(*mem); free(*mem);
*mem = calloc(sizeof(char *), *nmem+1); *mem = calloc(sizeof(char *), *nmem+1);
if (!*mem) { if (!*mem) {
rv = errno;
free(*line); free(*line);
*line = 0; *line = 0;
return 0; return 0;
@ -58,5 +61,7 @@ struct group *__getgrent_a(FILE *f, struct group *gr, char **line, size_t *size,
gr->gr_mem = *mem; gr->gr_mem = *mem;
end: end:
pthread_setcancelstate(cs, 0); pthread_setcancelstate(cs, 0);
return gr; *res = gr;
if(rv) errno = rv;
return rv;
} }

3
src/passwd/pwf.h

@ -10,5 +10,6 @@
int __getpwent_a(FILE *f, struct passwd *pw, char **line, size_t *size, struct passwd **res); int __getpwent_a(FILE *f, struct passwd *pw, char **line, size_t *size, struct passwd **res);
int __getpw_a(const char *name, uid_t uid, struct passwd *pw, char **buf, size_t *size, struct passwd **res); int __getpw_a(const char *name, uid_t uid, struct passwd *pw, char **buf, size_t *size, struct passwd **res);
struct group *__getgrent_a(FILE *f, struct group *gr, char **line, size_t *size, char ***mem, size_t *nmem); int __getgrent_a(FILE *f, struct group *gr, char **line, size_t *size, char ***mem, size_t *nmem, struct group **res);
int __getgr_a(const char *name, gid_t gid, struct group *gr, char **buf, size_t *size, char ***mem, size_t *nmem, struct group **res);
int __parsespent(char *s, struct spwd *sp); int __parsespent(char *s, struct spwd *sp);

Loading…
Cancel
Save