anotherim/anotherim-desktop
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

make GetEnv more robust and not limit length of variables

Browse source
This commit is contained in:
mjk 2021-03-05 23:54:35 +00:00 committed by LAGonauta
parent fb754b0d43
commit 34519e96bd
1 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
plugins/windows-notification/api/src/win32.cpp
View file

@ -9,6 +9,8 @@ win32_error::win32_error() noexcept
: win32_error{::GetLastError()}
{}
constexpr auto noncharacter = L'\uFFFF';
std::wstring GetExePath()
{
std::wstring exePath(MAX_PATH, 0);
@ -23,14 +25,18 @@ std::wstring GetExePath()
std::wstring GetEnv(const wchar_t *const variable_name)
{
std::wstring shortcutPath(MAX_PATH, 0);
auto charWritten = GetEnvironmentVariable(variable_name, shortcutPath.data(), shortcutPath.size());
if (charWritten > 0)
{
shortcutPath.resize(charWritten);
return shortcutPath;
}
const auto bufsize = ::GetEnvironmentVariableW(variable_name, nullptr, 0);
if (not bufsize)
throw win32_error{};
std::wstring buf(bufsize, noncharacter);
const auto res =
::GetEnvironmentVariableW(variable_name, buf.data(), bufsize);
if (const auto e = ::GetLastError())
throw win32_error{e};
if (not res or res >= bufsize) // not entirely sure this isn't just paranoia
throw std::runtime_error{"GetEnvironmentVariableW misbehaved"};
buf.resize(res);
return buf;
}
bool ImplSetProcessAumid(const char *const aumid)