[greatlrd] 21956: Bug 1507 patch from w3seek : This patch fixes more issues with NtUserFindWindowEx. (one example of bugs have been fixed with this patch is the tray icon)
Author: greatlrd Date: Sun May 21 02:09:42 2006 New Revision: 21956
URL: http://svn.reactos.ru/svn/reactos?rev=21956&view=rev Log: Bug 1507 patch from w3seek : This patch fixes more issues with NtUserFindWindowEx. (one example of bugs have been fixed with this patch is the tray icon)
Modified: trunk/reactos/subsystems/win32/win32k/include/class.h trunk/reactos/subsystems/win32/win32k/ntuser/class.c trunk/reactos/subsystems/win32/win32k/ntuser/window.c
Modified: trunk/reactos/subsystems/win32/win32k/include/class.h URL: http://svn.reactos.ru/svn/reactos/trunk/reactos/subsystems/win32/win32k/incl... ============================================================================== --- trunk/reactos/subsystems/win32/win32k/include/class.h (original) +++ trunk/reactos/subsystems/win32/win32k/include/class.h Sun May 21 02:09:42 2006 @@ -70,6 +70,10 @@ OUT PWINDOWCLASS **Link OPTIONAL);
BOOL +IntGetAtomFromStringOrAtom(IN PUNICODE_STRING ClassName, + OUT RTL_ATOM *Atom); + +BOOL IntCheckProcessDesktopClasses(IN PDESKTOP Desktop, IN BOOL FreeOnFailure);
Modified: trunk/reactos/subsystems/win32/win32k/ntuser/class.c URL: http://svn.reactos.ru/svn/reactos/trunk/reactos/subsystems/win32/win32k/ntus... ============================================================================== --- trunk/reactos/subsystems/win32/win32k/ntuser/class.c (original) +++ trunk/reactos/subsystems/win32/win32k/ntuser/class.c Sun May 21 02:09:42 2006 @@ -835,12 +835,11 @@ }
ClassSize = sizeof(WINDOWCLASS) + lpwcx->cbClsExtra; - ClassSize += ClassName->Length + sizeof(UNICODE_NULL); - ClassSize += MenuName->Length + sizeof(UNICODE_NULL); - if (ClassName->Length != 0) - ClassSize += RtlUnicodeStringToAnsiSize(ClassName); if (MenuName->Length != 0) + { + ClassSize += MenuName->Length + sizeof(UNICODE_NULL); ClassSize += RtlUnicodeStringToAnsiSize(MenuName); + }
if (Desktop != NULL) { @@ -996,14 +995,11 @@ return Class; }
-RTL_ATOM -IntGetClassAtom(IN PUNICODE_STRING ClassName, - IN HINSTANCE hInstance OPTIONAL, - IN PW32PROCESSINFO pi OPTIONAL, - OUT PWINDOWCLASS *BaseClass OPTIONAL, - OUT PWINDOWCLASS **Link OPTIONAL) -{ - RTL_ATOM Atom = (RTL_ATOM)0; +BOOL +IntGetAtomFromStringOrAtom(IN PUNICODE_STRING ClassName, + OUT RTL_ATOM *Atom) +{ + BOOL Ret = FALSE;
if (ClassName->Length != 0) { @@ -1038,8 +1034,12 @@ /* lookup the atom */ Status = RtlLookupAtomInAtomTable(gAtomTable, AtomName, - &Atom); - if (!NT_SUCCESS(Status)) + Atom); + if (NT_SUCCESS(Status)) + { + Ret = TRUE; + } + else { if (Status == STATUS_OBJECT_NAME_NOT_FOUND) { @@ -1054,10 +1054,25 @@ else { ASSERT(IS_ATOM(ClassName->Buffer)); - Atom = (RTL_ATOM)((ULONG_PTR)ClassName->Buffer); - } - - if (BaseClass != NULL && Atom != (RTL_ATOM)0) + *Atom = (RTL_ATOM)((ULONG_PTR)ClassName->Buffer); + Ret = TRUE; + } + + return Ret; +} + +RTL_ATOM +IntGetClassAtom(IN PUNICODE_STRING ClassName, + IN HINSTANCE hInstance OPTIONAL, + IN PW32PROCESSINFO pi OPTIONAL, + OUT PWINDOWCLASS *BaseClass OPTIONAL, + OUT PWINDOWCLASS **Link OPTIONAL) +{ + RTL_ATOM Atom = (RTL_ATOM)0; + + if (IntGetAtomFromStringOrAtom(ClassName, + &Atom) && + BaseClass != NULL && Atom != (RTL_ATOM)0) { PWINDOWCLASS Class;
Modified: trunk/reactos/subsystems/win32/win32k/ntuser/window.c URL: http://svn.reactos.ru/svn/reactos/trunk/reactos/subsystems/win32/win32k/ntus... ============================================================================== --- trunk/reactos/subsystems/win32/win32k/ntuser/window.c (original) +++ trunk/reactos/subsystems/win32/win32k/ntuser/window.c Sun May 21 02:09:42 2006 @@ -2344,7 +2344,7 @@ }
-HWND FASTCALL +static HWND FASTCALL IntFindWindow(PWINDOW_OBJECT Parent, PWINDOW_OBJECT ChildAfter, RTL_ATOM ClassAtom, @@ -2356,7 +2356,7 @@
ASSERT(Parent);
- CheckWindowName = (WindowName && (WindowName->Length > 0)); + CheckWindowName = WindowName->Length != 0;
if((List = IntWinListChildren(Parent))) { @@ -2425,14 +2425,66 @@ PUNICODE_STRING ucWindowName) { PWINDOW_OBJECT Parent, ChildAfter; - UNICODE_STRING ClassName, WindowName; - NTSTATUS Status; + UNICODE_STRING ClassName = {0}, WindowName = {0}; HWND Desktop, Ret = NULL; - RTL_ATOM ClassAtom; + RTL_ATOM ClassAtom = (RTL_ATOM)0; DECLARE_RETURN(HWND);
DPRINT("Enter NtUserFindWindowEx\n"); UserEnterShared(); + + _SEH_TRY + { + if (ucClassName != NULL) + { + ClassName = ProbeForReadUnicodeString(ucClassName); + if (ClassName.Length != 0) + { + ProbeForRead(ClassName.Buffer, + ClassName.Length, + sizeof(WCHAR)); + } + else if (IS_ATOM(ClassName.Buffer)) + { + if (!IntGetAtomFromStringOrAtom(&ClassName, + &ClassAtom)) + { + _SEH_LEAVE; + } + } + } + + if (ucWindowName != NULL) + { + WindowName = ProbeForReadUnicodeString(ucWindowName); + if (WindowName.Length != 0) + { + ProbeForRead(WindowName.Buffer, + WindowName.Length, + sizeof(WCHAR)); + } + } + } + _SEH_HANDLE + { + SetLastNtError(_SEH_GetExceptionCode()); + RETURN(NULL); + } + _SEH_END; + + if (ucClassName != NULL && ClassName.Length == 0) + { + if (!IS_ATOM(ClassName.Buffer)) + { + SetLastWin32Error(ERROR_INVALID_PARAMETER); + RETURN(NULL); + } + else if (ClassAtom == (RTL_ATOM)0) + { + /* LastError code was set by IntGetAtomFromStringOrAtom */ + RETURN(NULL); + } + }
Desktop = IntGetCurrentThreadDesktopWindow();
@@ -2456,158 +2508,91 @@ RETURN( NULL); }
- /* copy the window name */ - Status = IntSafeCopyUnicodeString(&WindowName, ucWindowName); - if(!NT_SUCCESS(Status)) - { - SetLastNtError(Status); - goto Cleanup3; - } - - /* safely copy the class name */ - Status = MmCopyFromCaller(&ClassName, ucClassName, sizeof(UNICODE_STRING)); - if(!NT_SUCCESS(Status)) - { - SetLastNtError(Status); - goto Cleanup2; - } - if(ClassName.Length > 0 && ClassName.Buffer) - { - WCHAR *buf; - /* safely copy the class name string (NULL terminated because class-lookup - depends on it... */ - buf = ExAllocatePoolWithTag(PagedPool, ClassName.Length + sizeof(WCHAR), TAG_STRING); - if(!buf) - { - SetLastWin32Error(STATUS_INSUFFICIENT_RESOURCES); - goto Cleanup2; - } - Status = MmCopyFromCaller(buf, ClassName.Buffer, ClassName.Length); - if(!NT_SUCCESS(Status)) - { - ExFreePool(buf); - SetLastNtError(Status); - goto Cleanup2; - } - ClassName.Buffer = buf; - /* make sure the string is null-terminated */ - buf += ClassName.Length / sizeof(WCHAR); - *buf = L'\0'; - } - - /* find the class object */ - if(ClassName.Buffer) - { - PWINSTATION_OBJECT WinStaObject; - - if (PsGetWin32Thread()->Desktop == NULL) - { - SetLastWin32Error(ERROR_INVALID_HANDLE); - goto Cleanup; - } - - WinStaObject = PsGetWin32Thread()->Desktop->WindowStation; - - Status = RtlLookupAtomInAtomTable( - WinStaObject->AtomTable, - ClassName.Buffer, - &ClassAtom); - - if (!NT_SUCCESS(Status)) - { - DPRINT1("Failed to lookup class atom!\n"); - SetLastWin32Error(ERROR_CLASS_DOES_NOT_EXIST); - goto Cleanup; - } - } - - if(Parent->hSelf == Desktop) - { - HWND *List, *phWnd; - PWINDOW_OBJECT TopLevelWindow; - BOOLEAN CheckWindowName; - BOOLEAN CheckClassName; - BOOLEAN WindowMatches; - BOOLEAN ClassMatches; - - /* windows searches through all top-level windows if the parent is the desktop - window */ - - if((List = IntWinListChildren(Parent))) - { - phWnd = List; - - if(ChildAfter) - { - /* skip handles before and including ChildAfter */ - while(*phWnd && (*(phWnd++) != ChildAfter->hSelf)) - ; - } - - CheckWindowName = WindowName.Length > 0; - CheckClassName = ClassName.Buffer != NULL; - - /* search children */ - while(*phWnd) - { - if(!(TopLevelWindow = UserGetWindowObject(*(phWnd++)))) - { - continue; - } - - /* Do not send WM_GETTEXT messages in the kernel mode version! - The user mode version however calls GetWindowText() which will - send WM_GETTEXT messages to windows belonging to its processes */ - WindowMatches = !CheckWindowName || !RtlCompareUnicodeString( - &WindowName, &TopLevelWindow->WindowName, TRUE); - ClassMatches = !CheckClassName || - ClassAtom == TopLevelWindow->Class->Atom; - - if (WindowMatches && ClassMatches) - { - Ret = TopLevelWindow->hSelf; - break; - } - - if (IntFindWindow(TopLevelWindow, NULL, ClassAtom, &WindowName)) - { - /* window returns the handle of the top-level window, in case it found - the child window */ - Ret = TopLevelWindow->hSelf; - break; - } - - } - ExFreePool(List); - } - } - else - Ret = IntFindWindow(Parent, ChildAfter, ClassAtom, &WindowName); + _SEH_TRY + { + if(Parent->hSelf == Desktop) + { + HWND *List, *phWnd; + PWINDOW_OBJECT TopLevelWindow; + BOOLEAN CheckWindowName; + BOOLEAN WindowMatches; + BOOLEAN ClassMatches; + + /* windows searches through all top-level windows if the parent is the desktop + window */ + + if((List = IntWinListChildren(Parent))) + { + phWnd = List; + + if(ChildAfter) + { + /* skip handles before and including ChildAfter */ + while(*phWnd && (*(phWnd++) != ChildAfter->hSelf)) + ; + } + + CheckWindowName = WindowName.Length != 0; + + /* search children */ + while(*phWnd) + { + if(!(TopLevelWindow = UserGetWindowObject(*(phWnd++)))) + { + continue; + } + + /* Do not send WM_GETTEXT messages in the kernel mode version! + The user mode version however calls GetWindowText() which will + send WM_GETTEXT messages to windows belonging to its processes */ + WindowMatches = !CheckWindowName || !RtlCompareUnicodeString( + &WindowName, &TopLevelWindow->WindowName, TRUE); + ClassMatches = (ClassAtom == (RTL_ATOM)0) || + ClassAtom == TopLevelWindow->Class->Atom; + + if (WindowMatches && ClassMatches) + { + Ret = TopLevelWindow->hSelf; + break; + } + + if (IntFindWindow(TopLevelWindow, NULL, ClassAtom, &WindowName)) + { + /* window returns the handle of the top-level window, in case it found + the child window */ + Ret = TopLevelWindow->hSelf; + break; + } + + } + ExFreePool(List); + } + } + else + Ret = IntFindWindow(Parent, ChildAfter, ClassAtom, &WindowName);
#if 0
- if(Ret == NULL && hwndParent == NULL && hwndChildAfter == NULL) - { - /* FIXME - if both hwndParent and hwndChildAfter are NULL, we also should - search the message-only windows. Should this also be done if - Parent is the desktop window??? */ - PWINDOW_OBJECT MsgWindows; - - if((MsgWindows = UserGetWindowObject(IntGetMessageWindow()))) - { - Ret = IntFindWindow(MsgWindows, ChildAfter, ClassAtom, &WindowName); - } - } + if(Ret == NULL && hwndParent == NULL && hwndChildAfter == NULL) + { + /* FIXME - if both hwndParent and hwndChildAfter are NULL, we also should + search the message-only windows. Should this also be done if + Parent is the desktop window??? */ + PWINDOW_OBJECT MsgWindows; + + if((MsgWindows = UserGetWindowObject(IntGetMessageWindow()))) + { + Ret = IntFindWindow(MsgWindows, ChildAfter, ClassAtom, &WindowName); + } + } #endif - -Cleanup: - if(ClassName.Length > 0 && ClassName.Buffer) - ExFreePool(ClassName.Buffer); - -Cleanup2: - RtlFreeUnicodeString(&WindowName); - -Cleanup3: + } + _SEH_HANDLE + { + SetLastNtError(_SEH_GetExceptionCode()); + Ret = NULL; + } + _SEH_END;
RETURN( Ret);
-
greatlrd＠svn.reactos.org