blob: 05439f8939df96d7eeacbca12782384d4139dbb4 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#include "user.h"
#include "env.h"
#include <util/generic/yexception.h>
#ifdef _win_
#include "winint.h"
#else
#include <cerrno>
#include <pwd.h>
#include <unistd.h>
#endif
TString GetUsername() {
for (const auto& var : {"LOGNAME", "USER", "LNAME", "USERNAME"}) {
TString val = GetEnv(var);
if (val) {
return val;
}
}
TTempBuf nameBuf;
for (;;) {
#if defined(_win_)
DWORD len = (DWORD)Min(nameBuf.Size(), size_t(32767));
if (!GetUserNameA(nameBuf.Data(), &len)) {
DWORD err = GetLastError();
if ((err == ERROR_INSUFFICIENT_BUFFER) && (nameBuf.Size() <= 32767))
nameBuf = TTempBuf((size_t)len);
else
ythrow TSystemError(err) << " GetUserName failed";
} else {
return TString(nameBuf.Data(), (size_t)(len - 1));
}
#elif defined(_bionic_)
const passwd* pwd = getpwuid(geteuid());
if (pwd) {
return TString(pwd->pw_name);
}
ythrow TSystemError() << TStringBuf(" getpwuid failed");
#else
passwd pwd;
passwd* tmpPwd;
int err = getpwuid_r(geteuid(), &pwd, nameBuf.Data(), nameBuf.Size(), &tmpPwd);
if (err == 0 && tmpPwd) {
return TString(pwd.pw_name);
} else if (err == ERANGE) {
nameBuf = TTempBuf(nameBuf.Size() * 2);
} else {
ythrow TSystemError(err) << " getpwuid_r failed";
}
#endif
}
}
|
