Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-03 11:11:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTUSER][USER32] Make NtUserBuildHwndList exclusive (#4485)

Browse source 
- Modify prototype of NtUserBuildHwndList function.
- Wrap the code by UserEnterExclusive(); and UserLeave();.
CORE-18173
This commit is contained in:
Katayama Hirofumi MZ 2022-05-05 08:49:00 +09:00 committed by GitHub
parent bbc9a877cc
commit c2a51c7249
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 37 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
win32ss/include/ntuser.h
View file

@ -1561,9 +1561,9 @@ NtUserBuildHwndList(
HWND hwndParent,
BOOLEAN bChildren,
ULONG dwThreadId,
ULONG lParam,
HWND *pWnd,
ULONG *pBufSize);
ULONG cHwnd,
HWND *phwndList,
ULONG *pcHwndNeeded);
NTSTATUS
NTAPI

51
win32ss/user/ntuser/window.c
View file

@ -1500,22 +1500,24 @@ VOID FASTCALL IntFreeHwndList(PWINDOWLIST pwlTarget)
* @implemented
*/
NTSTATUS
APIENTRY
NTAPI
NtUserBuildHwndList(
HDESK hDesktop,
HWND hwndParent,
BOOLEAN bChildren,
ULONG dwThreadId,
ULONG lParam,
HWND* pWnd,
ULONG* pBufSize)
ULONG cHwnd,
HWND* phwndList,
ULONG* pcHwndNeeded)
{
NTSTATUS Status;
ULONG dwCount = 0;
if (pBufSize == 0)
if (pcHwndNeeded == NULL)
return ERROR_INVALID_PARAMETER;
UserEnterExclusive();
if (hwndParent || !dwThreadId)
{
PDESKTOP Desktop;
@ -1525,7 +1527,8 @@ NtUserBuildHwndList(
{
if(hDesktop == NULL && !(Desktop = IntGetActiveDesktop()))
{
return ERROR_INVALID_HANDLE;
Status = ERROR_INVALID_HANDLE;
goto Quit;
}
if(hDesktop)
@ -1536,7 +1539,8 @@ NtUserBuildHwndList(
&Desktop);
if(!NT_SUCCESS(Status))
{
return ERROR_INVALID_HANDLE;
Status = ERROR_INVALID_HANDLE;
goto Quit;
}
}
hwndParent = Desktop->DesktopWindow;
@ -1556,13 +1560,13 @@ NtUserBuildHwndList(
{
if (bGoDown)
{
if(dwCount++ < *pBufSize && pWnd)
if (dwCount++ < cHwnd && phwndList)
{
_SEH2_TRY
{
ProbeForWrite(pWnd, sizeof(HWND), 1);
*pWnd = Window->head.h;
pWnd++;
ProbeForWrite(phwndList, sizeof(HWND), 1);
*phwndList = Window->head.h;
phwndList++;
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
@ -1571,7 +1575,6 @@ NtUserBuildHwndList(
_SEH2_END
if(!NT_SUCCESS(Status))
{
SetLastNtError(Status);
break;
}
}
@ -1612,13 +1615,15 @@ NtUserBuildHwndList(
if (!NT_SUCCESS(Status))
{
ERR("Thread Id is not valid!\n");
return ERROR_INVALID_PARAMETER;
Status = ERROR_INVALID_PARAMETER;
goto Quit;
}
if (!(W32Thread = (PTHREADINFO)Thread->Tcb.Win32Thread))
{
ObDereferenceObject(Thread);
TRACE("Tried to enumerate windows of a non gui thread\n");
return ERROR_INVALID_PARAMETER;
Status = ERROR_INVALID_PARAMETER;
goto Quit;
}
// Do not use Thread link list due to co_UserFreeWindow!!!
@ -1633,13 +1638,13 @@ NtUserBuildHwndList(
Window = ValidateHwndNoErr(List[i]);
if (Window && Window->head.pti == W32Thread)
{
if (dwCount < *pBufSize && pWnd)
if (dwCount < cHwnd && phwndList)
{
_SEH2_TRY
{
ProbeForWrite(pWnd, sizeof(HWND), 1);
*pWnd = Window->head.h;
pWnd++;
ProbeForWrite(phwndList, sizeof(HWND), 1);
*phwndList = Window->head.h;
phwndList++;
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
@ -1649,7 +1654,6 @@ NtUserBuildHwndList(
if (!NT_SUCCESS(Status))
{
ERR("Failure to build window list!\n");
SetLastNtError(Status);
break;
}
}
@ -1662,8 +1666,13 @@ NtUserBuildHwndList(
ObDereferenceObject(Thread);
}
*pBufSize = dwCount;
return STATUS_SUCCESS;
*pcHwndNeeded = dwCount;
Status = STATUS_SUCCESS;
Quit:
SetLastNtError(Status);
UserLeave();
return Status;
}
static void IntSendParentNotify( PWND pWindow, UINT msg )

4
win32ss/user/user32/windows/mdi.c
View file

@ -145,7 +145,7 @@ HWND* WIN_ListChildren (HWND hWndparent)
HANDLE hHeap;
NTSTATUS Status;
Status = NtUserBuildHwndList ( NULL, hWndparent, FALSE, 0, 0, NULL, &dwCount );
Status = NtUserBuildHwndList(NULL, hWndparent, FALSE, 0, dwCount, NULL, &dwCount);
if ( !NT_SUCCESS( Status ) )
return 0;
@ -161,7 +161,7 @@ HWND* WIN_ListChildren (HWND hWndparent)
}
/* now call kernel again to fill the buffer this time */
Status = NtUserBuildHwndList (NULL, hWndparent, FALSE, 0, 0, pHwnd, &dwCount );
Status = NtUserBuildHwndList(NULL, hWndparent, FALSE, 0, dwCount, pHwnd, &dwCount);
if ( !NT_SUCCESS( Status ) )
{

4
win32ss/user/user32/windows/window.c
View file

@ -701,7 +701,7 @@ User32EnumWindows(HDESK hDesktop,
hWndparent,
bChildren,
dwThreadId,
lParam,
dwCount,
NULL,
&dwCount);
if (!NT_SUCCESS(Status))
@ -729,7 +729,7 @@ User32EnumWindows(HDESK hDesktop,
hWndparent,
bChildren,
dwThreadId,
lParam,
dwCount,
pHwnd,
&dwCount);
if (!NT_SUCCESS(Status))