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

Add winstd::system_impersonator

Browse Source 
Signed-off-by: Simon Rozman <simon@rozman.si>
This commit is contained in:
Simon Rozman
2022-10-14 13:21:36 +02:00
parent 539828dd2c
commit 28b70eedb8
2 changed files with 117 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
UnitTests/Win.cpp
View File

@@ -37,5 +37,21 @@ namespace UnitTests
if (!lib_shell32)
Assert::Fail(L"LoadLibraryEx failed");
}
TEST_METHOD(system_impersonator)
{
winstd::win_handle<NULL> processToken;
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY | TOKEN_DUPLICATE, processToken))
Assert::Fail(L"OpenProcessToken failed");
DWORD isElevated, dwLength;
if (!GetTokenInformation(processToken, TokenElevation, &isElevated, sizeof(isElevated), &dwLength))
Assert::Fail(L"GetTokenInformation failed");
winstd::system_impersonator system_impersonator;
// SYSTEM impersonation works in elevated processes only.
if (dwLength == sizeof(isElevated) && isElevated)
Assert::IsTrue(system_impersonator);
else
Assert::IsTrue(!system_impersonator && GetLastError() == ERROR_ACCESS_DENIED);
}
};
}