Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-27 09:34:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

[REACTOS] Fix GetTokenInformation() usage (#2997)

Browse source 
The first call to GetTokenInformation is used to determine the size of a TokenInformation buffer.
It should fail and return ERROR_INSUFFICIENT_BUFFER
This commit is contained in:
Serge Gautherie 2020-07-21 14:39:11 +02:00 committed by GitHub
parent b338fb7beb
commit 98bbe8358c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 11 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
base/services/rpcss/setup.c
View file

@ -65,7 +65,7 @@ RunningAsSYSTEM(VOID)
return FALSE;
/* Retrieve token's information */
if (!GetTokenInformation(hToken, TokenUser, NULL, 0, &cbTokenBuffer) &&
if (GetTokenInformation(hToken, TokenUser, NULL, 0, &cbTokenBuffer) ||
GetLastError() != ERROR_INSUFFICIENT_BUFFER)
{
goto Quit;

4
base/services/svchost/security.cxx
View file

@ -53,14 +53,14 @@ DwInitializeSdFromThreadToken (
}
/* Get the size of the token's user */
if ((GetTokenInformation(hToken, TokenUser, NULL, 0, &dwUserLength) == FALSE) ||
if ((GetTokenInformation(hToken, TokenUser, NULL, 0, &dwUserLength) != FALSE) ||
(GetLastError() != ERROR_INSUFFICIENT_BUFFER))
{
return GetLastError();
}
/* Get the size of the token's primary group */
if ((GetTokenInformation(hToken, TokenPrimaryGroup, NULL, 0, &dwGroupLength) == FALSE) ||
if ((GetTokenInformation(hToken, TokenPrimaryGroup, NULL, 0, &dwGroupLength) != FALSE) ||
(GetLastError() != ERROR_INSUFFICIENT_BUFFER))
{
return GetLastError();

6
dll/cpl/sysdm/userprofile.c
View file

@ -695,9 +695,11 @@ AddUserProfiles(
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
return;
GetTokenInformation(hToken, TokenUser, NULL, 0, &dwSize);
if (dwSize == 0)
if (GetTokenInformation(hToken, TokenUser, NULL, 0, &dwSize) ||
GetLastError() != ERROR_INSUFFICIENT_BUFFER)
{
goto done;
}
pTokenUser = HeapAlloc(GetProcessHeap(), 0, dwSize);
if (pTokenUser == NULL)

11
dll/win32/netid/netid.c
View file

@ -179,14 +179,11 @@ IsUserAdmin(VOID)
&hToken))
goto done;
dwSize = 0;
GetTokenInformation(hToken,
TokenGroups,
NULL,
0,
&dwSize);
if (dwSize == 0)
if (GetTokenInformation(hToken, TokenGroups, NULL, 0, &dwSize) ||
GetLastError() != ERROR_INSUFFICIENT_BUFFER)
{
goto done;
}
pGroups = HeapAlloc(GetProcessHeap(), 0, dwSize);
if (pGroups == NULL)