--- src/tspi/ps/tspps.c.orig 2016-11-19 03:09:49 UTC +++ src/tspi/ps/tspps.c @@ -35,6 +35,11 @@ #define LE_16 htole16 #define LE_32 htole32 #define LE_64 htole64 +#elif defined(HAVE_SYS_ENDIAN_H) +#include +#define LE_16 htole16 +#define LE_32 htole32 +#define LE_64 htole64 #else #define LE_16(x) (x) #define LE_32(x) (x) @@ -49,9 +54,13 @@ #include "tspps.h" #include "tsplog.h" + static int user_ps_fd = -1; static MUTEX_DECLARE_INIT(user_ps_lock); #if (defined (__FreeBSD__) || defined (__OpenBSD__)) +#include +#define LE_16 htole16 +#define LE_32 htole32 static MUTEX_DECLARE_INIT(user_ps_path); #endif static struct flock fl; @@ -66,9 +75,7 @@ get_user_ps_path(char **file) TSS_RESULT result; char *file_name = NULL, *home_dir = NULL; struct passwd *pwp; -#if (defined (__linux) || defined (linux) || defined(__GLIBC__)) struct passwd pw; -#endif struct stat stat_buf; char buf[PASSWD_BUFSIZE]; uid_t euid; @@ -78,10 +85,6 @@ get_user_ps_path(char **file) *file = strdup(file_name); return (*file) ? TSS_SUCCESS : TSPERR(TSS_E_OUTOFMEMORY); } -#if (defined (__FreeBSD__) || defined (__OpenBSD__)) - MUTEX_LOCK(user_ps_path); -#endif - euid = geteuid(); #if defined (SOLARIS) @@ -94,32 +97,14 @@ get_user_ps_path(char **file) */ rc = snprintf(buf, sizeof (buf), "%s/%d", TSS_USER_PS_DIR, euid); #else - setpwent(); - while (1) { -#if (defined (__linux) || defined (linux) || defined(__GLIBC__)) - rc = getpwent_r(&pw, buf, PASSWD_BUFSIZE, &pwp); - if (rc) { - LogDebugFn("USER PS: Error getting path to home directory: getpwent_r: %s", - strerror(rc)); - endpwent(); - return TSPERR(TSS_E_INTERNAL_ERROR); - } + rc = getpwuid_r(euid, &pw, buf, PASSWD_BUFSIZE, &pwp); + if (rc) { + LogDebugFn("USER PS: Error getting path to home directory: " + "getpwent_r: %s", strerror(rc)); + return TSPERR(TSS_E_INTERNAL_ERROR); + } -#elif (defined (__FreeBSD__) || defined (__OpenBSD__)) - if ((pwp = getpwent()) == NULL) { - LogDebugFn("USER PS: Error getting path to home directory: getpwent: %s", - strerror(rc)); - endpwent(); - MUTEX_UNLOCK(user_ps_path); - return TSPERR(TSS_E_INTERNAL_ERROR); - } -#endif - if (euid == pwp->pw_uid) { - home_dir = strdup(pwp->pw_dir); - break; - } - } - endpwent(); + home_dir = strdup(pwp->pw_dir); if (!home_dir) return TSPERR(TSS_E_OUTOFMEMORY);