Amebis/stdex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sys_info: add username

Browse Source 
Signed-off-by: Simon Rozman <simon@rozman.si>
This commit is contained in:
Simon Rozman 2024-09-30 17:18:12 +02:00
parent 9fb899849d
commit 8844c4059d
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
include/stdex/sys_info.hpp
View File

@ -118,6 +118,11 @@ namespace stdex
/// ///
bool elevated; bool elevated;
///
/// Currently signed in user
///
sstring username;
sys_info_t() : sys_info_t() :
os_platform(platform_id::unknown), os_platform(platform_id::unknown),
#ifdef _WIN32 #ifdef _WIN32
@ -225,6 +230,27 @@ namespace stdex
// TODO: Set admin. // TODO: Set admin.
elevated = geteuid() == 0; elevated = geteuid() == 0;
#endif #endif
#ifdef _WIN32
{
TCHAR szStackBuffer[0x100];
ULONG ulSize = _countof(szStackBuffer);
if (GetUserNameEx(NameFormat, szStackBuffer, &ulSize))
username.assign(szStackBuffer, ulSize);
if (GetLastError() == ERROR_MORE_DATA) {
// Allocate buffer on heap and retry.
username.resize(ulSize - 1);
if (!GetUserNameEx(NameFormat, &username[0], &ulSize))
username.clear();
}
}
#else
{
struct passwd *pw = getpwuid(geteuid());
if (pw && pw->pw_name)
username = pw->pw_name;
}
#endif
} }
/// ///