luo980/glog
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix username lookup in case of missing USER environment variable

Browse Source
This commit is contained in:
Dariusz Ostolski 2017-09-19 22:23:53 +02:00
parent 2a6df66252
commit 2d3cf2681d
2 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configure.ac
View File

@ -31,6 +31,7 @@ AC_HEADER_STDC
AC_CHECK_HEADER(stdint.h, ac_cv_have_stdint_h=1, ac_cv_have_stdint_h=0) AC_CHECK_HEADER(stdint.h, ac_cv_have_stdint_h=1, ac_cv_have_stdint_h=0)
AC_CHECK_HEADER(sys/types.h, ac_cv_have_systypes_h=1, ac_cv_have_systypes_h=0) AC_CHECK_HEADER(sys/types.h, ac_cv_have_systypes_h=1, ac_cv_have_systypes_h=0)
AC_CHECK_HEADER(inttypes.h, ac_cv_have_inttypes_h=1, ac_cv_have_inttypes_h=0) AC_CHECK_HEADER(inttypes.h, ac_cv_have_inttypes_h=1, ac_cv_have_inttypes_h=0)
AC_CHECK_HEADER(pwd.h, ac_cv_have_pwd_h=1, ac_cv_have_pwd_h=0)
AC_CHECK_HEADERS(unistd.h, ac_cv_have_unistd_h=1, ac_cv_have_unistd_h=0) AC_CHECK_HEADERS(unistd.h, ac_cv_have_unistd_h=1, ac_cv_have_unistd_h=0)
AC_CHECK_HEADERS(syscall.h) AC_CHECK_HEADERS(syscall.h)
AC_CHECK_HEADERS(sys/syscall.h) AC_CHECK_HEADERS(sys/syscall.h)

26
src/utilities.cc
View File

@ -47,6 +47,12 @@
#ifdef HAVE_SYSLOG_H #ifdef HAVE_SYSLOG_H
# include <syslog.h> # include <syslog.h>
#endif #endif
#ifdef HAVE_UNISTD_H
# include <unistd.h> // For geteuid.
#endif
#ifdef HAVE_PWD_H
# include <pwd.h>
#endif
#include "base/googleinit.h" #include "base/googleinit.h"
@ -299,8 +305,26 @@ static void MyUserNameInitializer() {
if (user != NULL) { if (user != NULL) {
g_my_user_name = user; g_my_user_name = user;
} else { } else {
g_my_user_name = "invalid-user"; #if defined(HAVE_PWD_H) && defined(HAVE_UNISTD_H)
uid_t uid;
struct passwd pwd;
struct passwd*result = NULL;
char buffer[1024] = {'\0'};
uid = geteuid();
int pwuid_res = getpwuid_r(uid, &pwd, buffer, sizeof (buffer), &result);
if(pwuid_res == 0) {
g_my_user_name = pwd.pw_name;
}
else {
snprintf(buffer, sizeof(buffer), "uid%d", uid);
g_my_user_name = buffer;
}
#endif
if(g_my_user_name.empty()) {
g_my_user_name = "invalid-user";
}
} }
} }
REGISTER_MODULE_INITIALIZER(utilities, MyUserNameInitializer()); REGISTER_MODULE_INITIALIZER(utilities, MyUserNameInitializer());