📄 pwd_grp.c
字号:
/* Copyright (C) 2003 Manuel Novoa III * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the Free * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. *//* Nov 6, 2003 Initial version. * * NOTE: This implementation is quite strict about requiring all * field seperators. It also does not allow leading whitespace * except when processing the numeric fields. glibc is more * lenient. See the various glibc difference comments below. * * TODO: * Move to dynamic allocation of (currently staticly allocated) * buffers; especially for the group-related functions since * large group member lists will cause error returns. * */#define _GNU_SOURCE#include <features.h>#include <stdio.h>#include <stdlib.h>#include <stdint.h>#include <string.h>#include <stddef.h>#include <errno.h>#include <assert.h>#include <ctype.h>#include <pwd.h>#include <grp.h>#include <shadow.h>#ifdef __UCLIBC_HAS_THREADS__#include <pthread.h>#endif/**********************************************************************//* Prototypes for internal functions. */extern int __parsepwent(void *pw, char *line);extern int __parsegrent(void *gr, char *line);extern int __parsespent(void *sp, char *line);extern int __pgsreader(int (*__parserfunc)(void *d, char *line), void *data, char *__restrict line_buff, size_t buflen, FILE *f);/**********************************************************************//* For the various fget??ent_r funcs, return * * 0: success * ENOENT: end-of-file encountered * ERANGE: buflen too small * other error values possible. See __pgsreader. * * Also, *result == resultbuf on success and NULL on failure. * * NOTE: glibc difference - For the ENOENT case, glibc also sets errno. * We do not, as it really isn't an error if we reach the end-of-file. * Doing so is analogous to having fgetc() set errno on EOF. *//**********************************************************************/#ifdef L_fgetpwent_rint fgetpwent_r(FILE *__restrict stream, struct passwd *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct passwd **__restrict result){ int rv; *result = NULL; if (!(rv = __pgsreader(__parsepwent, resultbuf, buffer, buflen, stream))) { *result = resultbuf; } return rv;}#endif/**********************************************************************/#ifdef L_fgetgrent_rint fgetgrent_r(FILE *__restrict stream, struct group *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct group **__restrict result){ int rv; *result = NULL; if (!(rv = __pgsreader(__parsegrent, resultbuf, buffer, buflen, stream))) { *result = resultbuf; } return rv;}#endif/**********************************************************************/#ifdef L_fgetspent_rint fgetspent_r(FILE *__restrict stream, struct spwd *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct spwd **__restrict result){ int rv; *result = NULL; if (!(rv = __pgsreader(__parsespent, resultbuf, buffer, buflen, stream))) { *result = resultbuf; } return rv;}#endif/**********************************************************************//* For the various fget??ent funcs, return NULL on failure and a * pointer to the appropriate struct (staticly allocated) on success. *//**********************************************************************/#ifdef L_fgetpwentstruct passwd *fgetpwent(FILE *stream){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct passwd resultbuf; struct passwd *result; fgetpwent_r(stream, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_fgetgrentstruct group *fgetgrent(FILE *stream){ static char buffer[__UCLIBC_GRP_BUFFER_SIZE__]; static struct group resultbuf; struct group *result; fgetgrent_r(stream, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_fgetspentstruct spwd *fgetspent(FILE *stream){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct spwd resultbuf; struct spwd *result; fgetspent_r(stream, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_sgetspent_rint sgetspent_r(const char *string, struct spwd *result_buf, char *buffer, size_t buflen, struct spwd **result){ int rv = ERANGE; *result = NULL; if (buflen < __UCLIBC_PWD_BUFFER_SIZE__) { DO_ERANGE: __set_errno(rv); goto DONE; } if (string != buffer) { if (strlen(string) >= buflen) { goto DO_ERANGE; } strcpy(buffer, string); } if (!(rv = __parsespent(result_buf, buffer))) { *result = result_buf; } DONE: return rv;}#endif/**********************************************************************/#ifdef GETXXKEY_R_FUNC#error GETXXKEY_R_FUNC is already defined!#endif#ifdef L_getpwnam_r#define GETXXKEY_R_FUNC getpwnam_r#define GETXXKEY_R_PARSER __parsepwent#define GETXXKEY_R_ENTTYPE struct passwd#define GETXXKEY_R_TEST(ENT) (!strcmp((ENT)->pw_name, key))#define DO_GETXXKEY_R_KEYTYPE const char *__restrict#define DO_GETXXKEY_R_PATHNAME _PATH_PASSWD#endif#ifdef L_getgrnam_r#define GETXXKEY_R_FUNC getgrnam_r#define GETXXKEY_R_PARSER __parsegrent#define GETXXKEY_R_ENTTYPE struct group#define GETXXKEY_R_TEST(ENT) (!strcmp((ENT)->gr_name, key))#define DO_GETXXKEY_R_KEYTYPE const char *__restrict#define DO_GETXXKEY_R_PATHNAME _PATH_GROUP#endif#ifdef L_getspnam_r#define GETXXKEY_R_FUNC getspnam_r#define GETXXKEY_R_PARSER __parsespent#define GETXXKEY_R_ENTTYPE struct spwd#define GETXXKEY_R_TEST(ENT) (!strcmp((ENT)->sp_namp, key))#define DO_GETXXKEY_R_KEYTYPE const char *__restrict#define DO_GETXXKEY_R_PATHNAME _PATH_SHADOW#endif#ifdef L_getpwuid_r#define GETXXKEY_R_FUNC getpwuid_r#define GETXXKEY_R_PARSER __parsepwent#define GETXXKEY_R_ENTTYPE struct passwd#define GETXXKEY_R_TEST(ENT) ((ENT)->pw_uid == key)#define DO_GETXXKEY_R_KEYTYPE uid_t#define DO_GETXXKEY_R_PATHNAME _PATH_PASSWD#endif#ifdef L_getgrgid_r#define GETXXKEY_R_FUNC getgrgid_r#define GETXXKEY_R_PARSER __parsegrent#define GETXXKEY_R_ENTTYPE struct group#define GETXXKEY_R_TEST(ENT) ((ENT)->gr_gid == key)#define DO_GETXXKEY_R_KEYTYPE gid_t#define DO_GETXXKEY_R_PATHNAME _PATH_GROUP#endif/**********************************************************************/#ifdef GETXXKEY_R_FUNCint GETXXKEY_R_FUNC(DO_GETXXKEY_R_KEYTYPE key, GETXXKEY_R_ENTTYPE *__restrict resultbuf, char *__restrict buffer, size_t buflen, GETXXKEY_R_ENTTYPE **__restrict result){ FILE *stream; int rv; *result = NULL; if (!(stream = fopen(DO_GETXXKEY_R_PATHNAME, "r"))) { rv = errno; } else { __STDIO_SET_USER_LOCKING(stream); do { if (!(rv = __pgsreader(GETXXKEY_R_PARSER, resultbuf, buffer, buflen, stream)) ) { if (GETXXKEY_R_TEST(resultbuf)) { /* Found key? */ *result = resultbuf; break; } } else { if (rv == ENOENT) { /* end-of-file encountered. */ rv = 0; } break; } } while (1); fclose(stream); } return rv;}#endif/**********************************************************************/#ifdef L_getpwuidstruct passwd *getpwuid(uid_t uid){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct passwd resultbuf; struct passwd *result; getpwuid_r(uid, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getgrgidstruct group *getgrgid(gid_t gid){ static char buffer[__UCLIBC_GRP_BUFFER_SIZE__]; static struct group resultbuf; struct group *result; getgrgid_r(gid, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getspuid_r/* This function is non-standard and is currently not built. It seems * to have been created as a reentrant version of the non-standard * functions getspuid. Why getspuid was added, I do not know. */int getspuid_r(uid_t uid, struct spwd *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct spwd **__restrict result){ int rv; struct passwd *pp; struct passwd password; char pwd_buff[__UCLIBC_PWD_BUFFER_SIZE__]; *result = NULL; if (!(rv = getpwuid_r(uid, &password, pwd_buff, sizeof(pwd_buff), &pp))) { rv = getspnam_r(password.pw_name, resultbuf, buffer, buflen, result); } return rv;}#endif/**********************************************************************/#ifdef L_getspuid/* This function is non-standard and is currently not built. * Why it was added, I do not know. */struct spwd *getspuid(uid_t uid){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct spwd resultbuf; struct spwd *result; getspuid_r(uid, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getpwnamstruct passwd *getpwnam(const char *name){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct passwd resultbuf; struct passwd *result; getpwnam_r(name, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getgrnamstruct group *getgrnam(const char *name){ static char buffer[__UCLIBC_GRP_BUFFER_SIZE__]; static struct group resultbuf; struct group *result; getgrnam_r(name, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getspnamstruct spwd *getspnam(const char *name){ static char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; static struct spwd resultbuf; struct spwd *result; getspnam_r(name, &resultbuf, buffer, sizeof(buffer), &result); return result;}#endif/**********************************************************************/#ifdef L_getpwint getpw(uid_t uid, char *buf){ struct passwd resultbuf; struct passwd *result; char buffer[__UCLIBC_PWD_BUFFER_SIZE__]; if (!buf) { __set_errno(EINVAL); } else if (!getpwuid_r(uid, &resultbuf, buffer, sizeof(buffer), &result)) { if (sprintf(buf, "%s:%s:%lu:%lu:%s:%s:%s\n", resultbuf.pw_name, resultbuf.pw_passwd, (unsigned long)(resultbuf.pw_uid), (unsigned long)(resultbuf.pw_gid), resultbuf.pw_gecos, resultbuf.pw_dir, resultbuf.pw_shell) >= 0 ) { return 0; } } return -1;}#endif/**********************************************************************/#ifdef L_getpwent_r#ifdef __UCLIBC_HAS_THREADS__static pthread_mutex_t mylock = PTHREAD_MUTEX_INITIALIZER;# define LOCK __pthread_mutex_lock(&mylock)# define UNLOCK __pthread_mutex_unlock(&mylock);#else # define LOCK ((void) 0)# define UNLOCK ((void) 0)#endif static FILE *pwf /*= NULL*/;void setpwent(void){ LOCK; if (pwf) { rewind(pwf); } UNLOCK;}void endpwent(void){ LOCK; if (pwf) { fclose(pwf); pwf = NULL; } UNLOCK;}int getpwent_r(struct passwd *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct passwd **__restrict result){ int rv; LOCK; *result = NULL; /* In case of error... */ if (!pwf) { if (!(pwf = fopen(_PATH_PASSWD, "r"))) { rv = errno; goto ERR; } __STDIO_SET_USER_LOCKING(pwf); } if (!(rv = __pgsreader(__parsepwent, resultbuf, buffer, buflen, pwf))) { *result = resultbuf; } ERR: UNLOCK; return rv;}#endif/**********************************************************************/#ifdef L_getgrent_r#ifdef __UCLIBC_HAS_THREADS__static pthread_mutex_t mylock = PTHREAD_MUTEX_INITIALIZER;# define LOCK __pthread_mutex_lock(&mylock)# define UNLOCK __pthread_mutex_unlock(&mylock);#else # define LOCK ((void) 0)# define UNLOCK ((void) 0)#endif static FILE *grf /*= NULL*/;void setgrent(void){ LOCK; if (grf) { rewind(grf); } UNLOCK;}void endgrent(void){ LOCK; if (grf) { fclose(grf); grf = NULL; } UNLOCK;}int getgrent_r(struct group *__restrict resultbuf, char *__restrict buffer, size_t buflen, struct group **__restrict result){ int rv; LOCK; *result = NULL; /* In case of error... */ if (!grf) { if (!(grf = fopen(_PATH_GROUP, "r"))) { rv = errno; goto ERR; } __STDIO_SET_USER_LOCKING(grf); } if (!(rv = __pgsreader(__parsegrent, resultbuf, buffer, buflen, grf))) { *result = resultbuf; } ERR: UNLOCK; return rv;}#endif/**********************************************************************/#ifdef L_getspent_r#ifdef __UCLIBC_HAS_THREADS__static pthread_mutex_t mylock = PTHREAD_MUTEX_INITIALIZER;# define LOCK __pthread_mutex_lock(&mylock)# define UNLOCK __pthread_mutex_unlock(&mylock);#else # define LOCK ((void) 0)# define UNLOCK ((void) 0)#endif static FILE *spf /*= NULL*/;void setspent(void){ LOCK; if (spf) { rewind(spf); } UNLOCK;}void endspent(void)
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -