diff options
| -rw-r--r-- | CHANGELOG.md | 1 | ||||
| -rw-r--r-- | src/zencore/filesystem.cpp | 22 |
2 files changed, 19 insertions, 4 deletions
diff --git a/CHANGELOG.md b/CHANGELOG.md index 6e21d9941..8160a2be9 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,7 @@ - Bugfix: Prevent destructors in ProjectStore::Project, ScopedActivityBase and FileMapping from throwing exceptions to avoid abort termination - Bugfix: Zen CLI command help now include descriptions for positional arguments - Bugfix: Correctly prefix auth token when using a bare token in project oplog import/export +- Bugfix: Make sure GetEnvVariable can handle values that are longer than 1023 characters - Improvement: Throw exception with information on failed memory allocation instead of calling ZEN_ASSERT - Improvement: Added support for streaming decompression - Improvement: Added zenserver.exe and zen.exe/zen.pdb to Sentry debug information upload to populate unwind information diff --git a/src/zencore/filesystem.cpp b/src/zencore/filesystem.cpp index 9d55331eb..cde4c52ab 100644 --- a/src/zencore/filesystem.cpp +++ b/src/zencore/filesystem.cpp @@ -1144,11 +1144,25 @@ GetEnvVariable(std::string_view VariableName) ZEN_ASSERT(!VariableName.empty()); #if ZEN_PLATFORM_WINDOWS - CHAR EnvVariableBuffer[1023 + 1]; - DWORD RESULT = GetEnvironmentVariableA(std::string(VariableName).c_str(), EnvVariableBuffer, sizeof(EnvVariableBuffer)); - if (RESULT > 0 && RESULT < sizeof(EnvVariableBuffer)) + std::vector<CHAR> EnvVariableBuffer(1023 + 1); + DWORD RESULT = GetEnvironmentVariableA(std::string(VariableName).c_str(), EnvVariableBuffer.data(), (DWORD)EnvVariableBuffer.size()); + if (RESULT == 0) { - return std::string(EnvVariableBuffer); + return ""; + } + if (RESULT <= EnvVariableBuffer.size()) + { + return std::string(EnvVariableBuffer.data(), size_t(RESULT)); + } + EnvVariableBuffer.resize(size_t(RESULT)); + RESULT = GetEnvironmentVariableA(std::string(VariableName).c_str(), EnvVariableBuffer.data(), (DWORD)EnvVariableBuffer.size()); + if (RESULT == 0) + { + return ""; + } + if (RESULT <= EnvVariableBuffer.size()) + { + return std::string(EnvVariableBuffer.data(), size_t(RESULT)); } #endif #if ZEN_PLATFORM_LINUX || ZEN_PLATFORM_MAC |