diff --git a/libc/unistd/getlogin_r.cpp b/libc/unistd/getlogin_r.cpp
index d4ed5224..8bc1f647 100644
--- a/libc/unistd/getlogin_r.cpp
+++ b/libc/unistd/getlogin_r.cpp
@@ -18,22 +18,42 @@
along with the Sortix C Library. If not, see .
unistd/getlogin_r.cpp
- Get name of user logged ina t the controlling terminal.
+ Get name of user logged in at the controlling terminal.
*******************************************************************************/
#include
+#include
#include
-#include
#include
+#include
extern "C" int getlogin_r(char* buf, size_t size)
{
- const char* env_username = getenv("USERNAME");
- const char* username = env_username ? env_username : "root";
+ struct passwd passwd_object;
+ struct passwd* passwd;
+ int errnum = 0;
+
+ char* pwdbuf = NULL;
+ size_t pwdbuflen = 0;
+ do
+ {
+ size_t new_pwdbuflen = pwdbuflen ? 2 * pwdbuflen : 64;
+ char* new_pwdbuf = pwdbuf = (char*) realloc(pwdbuf, new_pwdbuflen);
+ if ( !new_pwdbuf )
+ return free(pwdbuf), -1;
+ pwdbuf = new_pwdbuf;
+ pwdbuflen = new_pwdbuflen;
+ } while ( (errnum = getpwuid_r(getuid(), &passwd_object, pwdbuf, pwdbuflen,
+ &passwd)) == ERANGE );
+ if ( errnum )
+ return free(pwdbuf), errno = errnum, -1;
+
+ const char* username = passwd->pw_name;
size_t username_len = strlen(username);
if ( size < (username_len + 1) * sizeof(char) )
- return errno = ERANGE, -1;
+ return free(pwdbuf), errno = ERANGE, -1;
strcpy(buf, username);
+
return 0;
}