williamjcm/wxMASSManager
1
0
Fork 0
Code Issues Pull requests Releases 16 Wiki Activity

Use a scope guard for more safety.

Browse source
This commit is contained in:
Guillaume Jacquemin 2021-04-20 15:55:06 +02:00
parent 97eb69922f
commit d56b2cc4c2
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
MassBuilderManager/MassBuilderManager.cpp
View file

@ -23,6 +23,7 @@
#include <shlobj.h> #include <shlobj.h>
#include <wtsapi32.h> #include <wtsapi32.h>
#include <Corrade/Containers/ScopeGuard.h>
#include <Corrade/Utility/Directory.h> #include <Corrade/Utility/Directory.h>
#include <Corrade/Utility/Unicode.h> #include <Corrade/Utility/Unicode.h>
@ -52,6 +53,8 @@ auto MassBuilderManager::checkGameState() -> GameState {
_gameState = GameState::Unknown; _gameState = GameState::Unknown;
if(WTSEnumerateProcessesW(WTS_CURRENT_SERVER_HANDLE, 0, 1, &process_infos, &process_count)) { if(WTSEnumerateProcessesW(WTS_CURRENT_SERVER_HANDLE, 0, 1, &process_infos, &process_count)) {
Containers::ScopeGuard guard{process_infos, WTSFreeMemory};
for(unsigned long i = 0; i < process_count; ++i) { for(unsigned long i = 0; i < process_count; ++i) {
if(std::wcscmp(process_infos[i].pProcessName, L"MASS_Builder-Win64-Shipping.exe") == 0) { if(std::wcscmp(process_infos[i].pProcessName, L"MASS_Builder-Win64-Shipping.exe") == 0) {
_gameState = GameState::Running; _gameState = GameState::Running;
@ -63,11 +66,6 @@ auto MassBuilderManager::checkGameState() -> GameState {
} }
} }
if(process_infos != nullptr) {
WTSFreeMemory(process_infos);
process_infos = nullptr;
}
return _gameState; return _gameState;
} }