diff options
Diffstat (limited to 'libps/user.c')
| -rw-r--r-- | libps/user.c | 162 |
1 files changed, 162 insertions, 0 deletions
diff --git a/libps/user.c b/libps/user.c new file mode 100644 index 00000000..4af5014f --- /dev/null +++ b/libps/user.c @@ -0,0 +1,162 @@ +/* The ps_user type, for per-user info. + + Copyright (C) 1995, 1996 Free Software Foundation, Inc. + + Written by Miles Bader <miles@gnu.ai.mit.edu> + + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License as + published by the Free Software Foundation; either version 2, or (at + your option) any later version. + + This program 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 + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ + +#include <hurd.h> +#include <stdio.h> +#include <stdlib.h> +#include <assert.h> + +#include "ps.h" +#include "common.h" + +static error_t +install_passwd (struct ps_user *u, struct passwd *pw) +{ + int needed = 0; + +#define COUNT(field) if (pw->field != NULL) (needed += strlen(pw->field) + 1) + COUNT (pw_name); + COUNT (pw_passwd); + COUNT (pw_gecos); + COUNT (pw_dir); + COUNT (pw_shell); + + u->storage = malloc (needed); + if (u->storage != NULL) + { + char *p = u->storage; + + /* Copy each string field into storage allocated in the u + structure and point the fields at that instead of the static + storage that pw currently points to. */ +#define COPY(field) \ +if (pw->field != NULL) \ +strcpy(p, pw->field), (pw->field = p), (p += strlen (p) + 1) + COPY (pw_name); + COPY (pw_passwd); + COPY (pw_gecos); + COPY (pw_dir); + COPY (pw_shell); + + u->passwd = *pw; + + return 0; + } + else + return ENOMEM; +} + +/* Create a ps_user for the user referred to by UID, returning it in U. + If a memory allocation error occurs, ENOMEM is returned, otherwise 0. */ +error_t +ps_user_create (uid_t uid, struct ps_user **u) +{ + *u = NEW (struct ps_user); + if (*u == NULL) + return ENOMEM; + + (*u)->uid = uid; + (*u)->passwd_state = PS_USER_PASSWD_PENDING; + + return 0; +} + +/* Create a ps_user for the user referred to by UNAME, returning it in U. + If a memory allocation error occurs, ENOMEM is returned. If no such user + is known, EINVAL is returned. */ +error_t +ps_user_uname_create (char *uname, struct ps_user **u) +{ + struct passwd *pw = getpwnam (uname); + if (pw) + return ps_user_passwd_create (pw, u); + else + return EINVAL; +} + +/* Makes makes a ps_user containing PW (which is copied). */ +error_t +ps_user_passwd_create (struct passwd *pw, struct ps_user **u) +{ + error_t err = 0; + + *u = NEW (struct ps_user); + if (*u == NULL) + err = ENOMEM; + else + { + err = install_passwd (*u, pw); + if (err) + FREE (*u); + else + { + (*u)->passwd_state = PS_USER_PASSWD_OK; + (*u)->uid = pw->pw_uid; + } + } + + return err; +} + +/* Free U and any resources it consumes. */ +void +ps_user_free (struct ps_user *u) +{ + if (u->passwd_state == PS_USER_PASSWD_OK) + free (u->storage); + free (u); +} + +/* ---------------------------------------------------------------- */ + +/* Returns the password file entry (struct passwd, from <pwd.h>) for the user + referred to by U, or NULL if it can't be gotten. */ +struct passwd *ps_user_passwd (struct ps_user *u) +{ + if (u->passwd_state == PS_USER_PASSWD_OK) + return &u->passwd; + else if (u->passwd_state == PS_USER_PASSWD_ERROR) + return NULL; + else + { + struct passwd *pw = getpwuid (u->uid); + if (pw != NULL && install_passwd (u, pw) == 0) + { + u->passwd_state = PS_USER_PASSWD_OK; + return &u->passwd; + } + else + { + u->passwd_state = PS_USER_PASSWD_ERROR; + return NULL; + } + } +} + +/* Returns the user name for the user referred to by U, or NULL if it can't + be gotten. */ +char *ps_user_name (struct ps_user *u) +{ + struct passwd *pw = ps_user_passwd (u); + if (pw) + return pw->pw_name; + else + return NULL; +} |