Author: tfaber Date: Fri Apr 15 19:49:33 2016 New Revision: 71163
URL: http://svn.reactos.org/svn/reactos?rev=71163&view=rev Log: [WIN32K:NTUSER] - Fix indentation. No code changes.
Modified: trunk/reactos/win32ss/user/ntuser/winsta.c
Modified: trunk/reactos/win32ss/user/ntuser/winsta.c URL: http://svn.reactos.org/svn/reactos/trunk/reactos/win32ss/user/ntuser/winsta.... ============================================================================== --- trunk/reactos/win32ss/user/ntuser/winsta.c [iso-8859-1] (original) +++ trunk/reactos/win32ss/user/ntuser/winsta.c [iso-8859-1] Fri Apr 15 19:49:33 2016 @@ -30,17 +30,17 @@ NTAPI InitWindowStationImpl(VOID) { - GENERIC_MAPPING IntWindowStationMapping = { WINSTA_READ, - WINSTA_WRITE, - WINSTA_EXECUTE, - WINSTA_ACCESS_ALL}; - - /* Set Winsta Object Attributes */ - ExWindowStationObjectType->TypeInfo.DefaultNonPagedPoolCharge = sizeof(WINSTATION_OBJECT); - ExWindowStationObjectType->TypeInfo.GenericMapping = IntWindowStationMapping; - ExWindowStationObjectType->TypeInfo.ValidAccessMask = WINSTA_ACCESS_ALL; - - return STATUS_SUCCESS; + GENERIC_MAPPING IntWindowStationMapping = { WINSTA_READ, + WINSTA_WRITE, + WINSTA_EXECUTE, + WINSTA_ACCESS_ALL}; + + /* Set Winsta Object Attributes */ + ExWindowStationObjectType->TypeInfo.DefaultNonPagedPoolCharge = sizeof(WINSTATION_OBJECT); + ExWindowStationObjectType->TypeInfo.GenericMapping = IntWindowStationMapping; + ExWindowStationObjectType->TypeInfo.ValidAccessMask = WINSTA_ACCESS_ALL; + + return STATUS_SUCCESS; }
NTSTATUS @@ -76,21 +76,21 @@ } }
- InitializeObjectAttributes(&ObjectAttributes, - &gustrWindowStationsDir, - 0, - NULL, - NULL); - Status = ZwCreateDirectoryObject(&hWinstaDir, 0, &ObjectAttributes); - if (!NT_SUCCESS(Status)) - { - ERR("Could not create %wZ directory (Status 0x%X)\n", &gustrWindowStationsDir, Status); - return Status; - } - - TRACE("Created directory %wZ for session %lu\n", &gustrWindowStationsDir, Peb->SessionId); - - return Status; + InitializeObjectAttributes(&ObjectAttributes, + &gustrWindowStationsDir, + 0, + NULL, + NULL); + Status = ZwCreateDirectoryObject(&hWinstaDir, 0, &ObjectAttributes); + if (!NT_SUCCESS(Status)) + { + ERR("Could not create %wZ directory (Status 0x%X)\n", &gustrWindowStationsDir, Status); + return Status; + } + + TRACE("Created directory %wZ for session %lu\n", &gustrWindowStationsDir, Peb->SessionId); + + return Status; }
/* OBJECT CALLBACKS **********************************************************/ @@ -101,19 +101,19 @@ _In_ PVOID Parameters) { PWIN32_DELETEMETHOD_PARAMETERS DeleteParameters = Parameters; - PWINSTATION_OBJECT WinSta = (PWINSTATION_OBJECT)DeleteParameters->Object; - - TRACE("Deleting window station (0x%p)\n", WinSta); - - WinSta->Flags |= WSS_DYING; - - UserEmptyClipboardData(WinSta); - - RtlDestroyAtomTable(WinSta->AtomTable); - - RtlFreeUnicodeString(&WinSta->Name); - - return STATUS_SUCCESS; + PWINSTATION_OBJECT WinSta = (PWINSTATION_OBJECT)DeleteParameters->Object; + + TRACE("Deleting window station (0x%p)\n", WinSta); + + WinSta->Flags |= WSS_DYING; + + UserEmptyClipboardData(WinSta); + + RtlDestroyAtomTable(WinSta->AtomTable); + + RtlFreeUnicodeString(&WinSta->Name); + + return STATUS_SUCCESS; }
NTSTATUS @@ -213,140 +213,139 @@
NTSTATUS FASTCALL IntValidateWindowStationHandle( - HWINSTA WindowStation, - KPROCESSOR_MODE AccessMode, - ACCESS_MASK DesiredAccess, - PWINSTATION_OBJECT *Object, - POBJECT_HANDLE_INFORMATION pObjectHandleInfo) -{ - NTSTATUS Status; - - if (WindowStation == NULL) - { - ERR("Invalid window station handle\n"); - EngSetLastError(ERROR_INVALID_HANDLE); - return STATUS_INVALID_HANDLE; - } - - Status = ObReferenceObjectByHandle( - WindowStation, - DesiredAccess, - ExWindowStationObjectType, - AccessMode, - (PVOID*)Object, - pObjectHandleInfo); - - if (!NT_SUCCESS(Status)) - SetLastNtError(Status); - - return Status; + HWINSTA WindowStation, + KPROCESSOR_MODE AccessMode, + ACCESS_MASK DesiredAccess, + PWINSTATION_OBJECT *Object, + POBJECT_HANDLE_INFORMATION pObjectHandleInfo) +{ + NTSTATUS Status; + + if (WindowStation == NULL) + { + ERR("Invalid window station handle\n"); + EngSetLastError(ERROR_INVALID_HANDLE); + return STATUS_INVALID_HANDLE; + } + + Status = ObReferenceObjectByHandle(WindowStation, + DesiredAccess, + ExWindowStationObjectType, + AccessMode, + (PVOID*)Object, + pObjectHandleInfo); + + if (!NT_SUCCESS(Status)) + SetLastNtError(Status); + + return Status; }
BOOL FASTCALL co_IntInitializeDesktopGraphics(VOID) { - TEXTMETRICW tmw; - UNICODE_STRING DriverName = RTL_CONSTANT_STRING(L"DISPLAY"); - PDESKTOP pdesk; - - ScreenDeviceContext = IntGdiCreateDC(&DriverName, NULL, NULL, NULL, FALSE); - if (NULL == ScreenDeviceContext) - { - IntDestroyPrimarySurface(); - return FALSE; - } - GreSetDCOwner(ScreenDeviceContext, GDI_OBJ_HMGR_PUBLIC); - - if (! IntCreatePrimarySurface()) - { - return FALSE; - } - - hSystemBM = NtGdiCreateCompatibleDC(ScreenDeviceContext); - - NtGdiSelectFont(hSystemBM, NtGdiGetStockObject(SYSTEM_FONT)); - GreSetDCOwner(hSystemBM, GDI_OBJ_HMGR_PUBLIC); - - /* Update the SERVERINFO */ - gpsi->aiSysMet[SM_CXSCREEN] = gppdevPrimary->gdiinfo.ulHorzRes; - gpsi->aiSysMet[SM_CYSCREEN] = gppdevPrimary->gdiinfo.ulVertRes; - gpsi->Planes = NtGdiGetDeviceCaps(ScreenDeviceContext, PLANES); - gpsi->BitsPixel = NtGdiGetDeviceCaps(ScreenDeviceContext, BITSPIXEL); - gpsi->BitCount = gpsi->Planes * gpsi->BitsPixel; - gpsi->dmLogPixels = NtGdiGetDeviceCaps(ScreenDeviceContext, LOGPIXELSY); - if (NtGdiGetDeviceCaps(ScreenDeviceContext, RASTERCAPS) & RC_PALETTE) - { - gpsi->PUSIFlags |= PUSIF_PALETTEDISPLAY; - } - else - gpsi->PUSIFlags &= ~PUSIF_PALETTEDISPLAY; - // Font is realized and this dc was previously set to internal DC_ATTR. - gpsi->cxSysFontChar = IntGetCharDimensions(hSystemBM, &tmw, (DWORD*)&gpsi->cySysFontChar); - gpsi->tmSysFont = tmw; - - /* Put the pointer in the center of the screen */ - gpsi->ptCursor.x = gpsi->aiSysMet[SM_CXSCREEN] / 2; - gpsi->ptCursor.y = gpsi->aiSysMet[SM_CYSCREEN] / 2; - - /* Attach monitor */ - UserAttachMonitor((HDEV)gppdevPrimary); - - /* Setup the cursor */ - co_IntLoadDefaultCursors(); - - /* Setup the icons */ - co_IntSetWndIcons(); - - /* Setup Menu */ - MenuInit(); - - /* Show the desktop */ - pdesk = IntGetActiveDesktop(); - ASSERT(pdesk); - co_IntShowDesktop(pdesk, gpsi->aiSysMet[SM_CXSCREEN], gpsi->aiSysMet[SM_CYSCREEN], TRUE); - - return TRUE; + TEXTMETRICW tmw; + UNICODE_STRING DriverName = RTL_CONSTANT_STRING(L"DISPLAY"); + PDESKTOP pdesk; + + ScreenDeviceContext = IntGdiCreateDC(&DriverName, NULL, NULL, NULL, FALSE); + if (NULL == ScreenDeviceContext) + { + IntDestroyPrimarySurface(); + return FALSE; + } + GreSetDCOwner(ScreenDeviceContext, GDI_OBJ_HMGR_PUBLIC); + + if (! IntCreatePrimarySurface()) + { + return FALSE; + } + + hSystemBM = NtGdiCreateCompatibleDC(ScreenDeviceContext); + + NtGdiSelectFont(hSystemBM, NtGdiGetStockObject(SYSTEM_FONT)); + GreSetDCOwner(hSystemBM, GDI_OBJ_HMGR_PUBLIC); + + /* Update the SERVERINFO */ + gpsi->aiSysMet[SM_CXSCREEN] = gppdevPrimary->gdiinfo.ulHorzRes; + gpsi->aiSysMet[SM_CYSCREEN] = gppdevPrimary->gdiinfo.ulVertRes; + gpsi->Planes = NtGdiGetDeviceCaps(ScreenDeviceContext, PLANES); + gpsi->BitsPixel = NtGdiGetDeviceCaps(ScreenDeviceContext, BITSPIXEL); + gpsi->BitCount = gpsi->Planes * gpsi->BitsPixel; + gpsi->dmLogPixels = NtGdiGetDeviceCaps(ScreenDeviceContext, LOGPIXELSY); + if (NtGdiGetDeviceCaps(ScreenDeviceContext, RASTERCAPS) & RC_PALETTE) + { + gpsi->PUSIFlags |= PUSIF_PALETTEDISPLAY; + } + else + gpsi->PUSIFlags &= ~PUSIF_PALETTEDISPLAY; + // Font is realized and this dc was previously set to internal DC_ATTR. + gpsi->cxSysFontChar = IntGetCharDimensions(hSystemBM, &tmw, (DWORD*)&gpsi->cySysFontChar); + gpsi->tmSysFont = tmw; + + /* Put the pointer in the center of the screen */ + gpsi->ptCursor.x = gpsi->aiSysMet[SM_CXSCREEN] / 2; + gpsi->ptCursor.y = gpsi->aiSysMet[SM_CYSCREEN] / 2; + + /* Attach monitor */ + UserAttachMonitor((HDEV)gppdevPrimary); + + /* Setup the cursor */ + co_IntLoadDefaultCursors(); + + /* Setup the icons */ + co_IntSetWndIcons(); + + /* Setup Menu */ + MenuInit(); + + /* Show the desktop */ + pdesk = IntGetActiveDesktop(); + ASSERT(pdesk); + co_IntShowDesktop(pdesk, gpsi->aiSysMet[SM_CXSCREEN], gpsi->aiSysMet[SM_CYSCREEN], TRUE); + + return TRUE; }
VOID FASTCALL IntEndDesktopGraphics(VOID) { - if (NULL != ScreenDeviceContext) - { // No need to allocate a new dcattr. - GreSetDCOwner(ScreenDeviceContext, GDI_OBJ_HMGR_POWNED); - GreDeleteObject(ScreenDeviceContext); - ScreenDeviceContext = NULL; - } - IntHideDesktop(IntGetActiveDesktop()); - IntDestroyPrimarySurface(); + if (NULL != ScreenDeviceContext) + { // No need to allocate a new dcattr. + GreSetDCOwner(ScreenDeviceContext, GDI_OBJ_HMGR_POWNED); + GreDeleteObject(ScreenDeviceContext); + ScreenDeviceContext = NULL; + } + IntHideDesktop(IntGetActiveDesktop()); + IntDestroyPrimarySurface(); }
HDC FASTCALL IntGetScreenDC(VOID) { - return ScreenDeviceContext; + return ScreenDeviceContext; }
BOOL FASTCALL CheckWinstaAttributeAccess(ACCESS_MASK DesiredAccess) { - PPROCESSINFO ppi = PsGetCurrentProcessWin32Process(); - if ( gpidLogon != PsGetCurrentProcessId() ) - { - if (!(ppi->W32PF_flags & W32PF_IOWINSTA)) - { - ERR("Requires Interactive Window Station\n"); - EngSetLastError(ERROR_REQUIRES_INTERACTIVE_WINDOWSTATION); - return FALSE; - } - if (!RtlAreAllAccessesGranted(ppi->amwinsta, DesiredAccess)) - { - ERR("Access Denied\n"); - EngSetLastError(ERROR_ACCESS_DENIED); - return FALSE; - } - } - return TRUE; + PPROCESSINFO ppi = PsGetCurrentProcessWin32Process(); + if ( gpidLogon != PsGetCurrentProcessId() ) + { + if (!(ppi->W32PF_flags & W32PF_IOWINSTA)) + { + ERR("Requires Interactive Window Station\n"); + EngSetLastError(ERROR_REQUIRES_INTERACTIVE_WINDOWSTATION); + return FALSE; + } + if (!RtlAreAllAccessesGranted(ppi->amwinsta, DesiredAccess)) + { + ERR("Access Denied\n"); + EngSetLastError(ERROR_ACCESS_DENIED); + return FALSE; + } + } + return TRUE; }
@@ -390,121 +389,118 @@
HWINSTA APIENTRY NtUserCreateWindowStation( - POBJECT_ATTRIBUTES ObjectAttributes, - ACCESS_MASK dwDesiredAccess, - DWORD Unknown2, - DWORD Unknown3, - DWORD Unknown4, - DWORD Unknown5, - DWORD Unknown6) -{ - UNICODE_STRING WindowStationName; - PWINSTATION_OBJECT WindowStationObject; - HWINSTA WindowStation; - NTSTATUS Status; - - TRACE("NtUserCreateWindowStation called\n"); - - Status = ObOpenObjectByName( - ObjectAttributes, - ExWindowStationObjectType, - UserMode, - NULL, - dwDesiredAccess, - NULL, - (PVOID*)&WindowStation); - - if (NT_SUCCESS(Status)) - { - TRACE("NtUserCreateWindowStation opened window station %wZ\n", ObjectAttributes->ObjectName); - return (HWINSTA)WindowStation; - } - - /* - * No existing window station found, try to create new one - */ - - /* Capture window station name */ - _SEH2_TRY - { - ProbeForRead( ObjectAttributes, sizeof(OBJECT_ATTRIBUTES), 1); - Status = IntSafeCopyUnicodeStringTerminateNULL(&WindowStationName, ObjectAttributes->ObjectName); - } - _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) - { - Status =_SEH2_GetExceptionCode(); - } - _SEH2_END - - if (! NT_SUCCESS(Status)) - { - ERR("Failed reading capturing window station name\n"); - SetLastNtError(Status); - return NULL; - } - - /* Create the window station object */ - Status = ObCreateObject( - UserMode, - ExWindowStationObjectType, - ObjectAttributes, - UserMode, - NULL, - sizeof(WINSTATION_OBJECT), - 0, - 0, - (PVOID*)&WindowStationObject); - - if (!NT_SUCCESS(Status)) - { - ERR("ObCreateObject failed for window station %wZ\n", &WindowStationName); - ExFreePoolWithTag(WindowStationName.Buffer, TAG_STRING); - SetLastNtError(STATUS_INSUFFICIENT_RESOURCES); - return 0; - } - - Status = ObInsertObject( - (PVOID)WindowStationObject, - NULL, - dwDesiredAccess, - 0, - NULL, - (PVOID*)&WindowStation); - - if (!NT_SUCCESS(Status)) - { - ERR("ObInsertObject failed for window station %wZ\n", &WindowStationName); - ExFreePoolWithTag(WindowStationName.Buffer, TAG_STRING); - SetLastNtError(STATUS_INSUFFICIENT_RESOURCES); - ObDereferenceObject(WindowStationObject); - return 0; - } - - /* Initialize the window station */ - RtlZeroMemory(WindowStationObject, sizeof(WINSTATION_OBJECT)); - - InitializeListHead(&WindowStationObject->DesktopListHead); - Status = RtlCreateAtomTable(37, &WindowStationObject->AtomTable); - WindowStationObject->Name = WindowStationName; - WindowStationObject->dwSessionId = NtCurrentPeb()->SessionId; - - if (InputWindowStation == NULL) - { - ERR("Initializing input window station\n"); - InputWindowStation = WindowStationObject; - - WindowStationObject->Flags &= ~WSS_NOIO; - - InitCursorImpl(); - } - else - { - WindowStationObject->Flags |= WSS_NOIO; - } - - TRACE("NtUserCreateWindowStation created object %p with name %wZ handle %p\n", + POBJECT_ATTRIBUTES ObjectAttributes, + ACCESS_MASK dwDesiredAccess, + DWORD Unknown2, + DWORD Unknown3, + DWORD Unknown4, + DWORD Unknown5, + DWORD Unknown6) +{ + UNICODE_STRING WindowStationName; + PWINSTATION_OBJECT WindowStationObject; + HWINSTA WindowStation; + NTSTATUS Status; + + TRACE("NtUserCreateWindowStation called\n"); + + Status = ObOpenObjectByName(ObjectAttributes, + ExWindowStationObjectType, + UserMode, + NULL, + dwDesiredAccess, + NULL, + (PVOID*)&WindowStation); + + if (NT_SUCCESS(Status)) + { + TRACE("NtUserCreateWindowStation opened window station %wZ\n", ObjectAttributes->ObjectName); + return (HWINSTA)WindowStation; + } + + /* + * No existing window station found, try to create new one + */ + + /* Capture window station name */ + _SEH2_TRY + { + ProbeForRead( ObjectAttributes, sizeof(OBJECT_ATTRIBUTES), 1); + Status = IntSafeCopyUnicodeStringTerminateNULL(&WindowStationName, ObjectAttributes->ObjectName); + } + _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) + { + Status =_SEH2_GetExceptionCode(); + } + _SEH2_END + + if (! NT_SUCCESS(Status)) + { + ERR("Failed reading capturing window station name\n"); + SetLastNtError(Status); + return NULL; + } + + /* Create the window station object */ + Status = ObCreateObject(UserMode, + ExWindowStationObjectType, + ObjectAttributes, + UserMode, + NULL, + sizeof(WINSTATION_OBJECT), + 0, + 0, + (PVOID*)&WindowStationObject); + + if (!NT_SUCCESS(Status)) + { + ERR("ObCreateObject failed for window station %wZ\n", &WindowStationName); + ExFreePoolWithTag(WindowStationName.Buffer, TAG_STRING); + SetLastNtError(STATUS_INSUFFICIENT_RESOURCES); + return 0; + } + + Status = ObInsertObject((PVOID)WindowStationObject, + NULL, + dwDesiredAccess, + 0, + NULL, + (PVOID*)&WindowStation); + + if (!NT_SUCCESS(Status)) + { + ERR("ObInsertObject failed for window station %wZ\n", &WindowStationName); + ExFreePoolWithTag(WindowStationName.Buffer, TAG_STRING); + SetLastNtError(STATUS_INSUFFICIENT_RESOURCES); + ObDereferenceObject(WindowStationObject); + return 0; + } + + /* Initialize the window station */ + RtlZeroMemory(WindowStationObject, sizeof(WINSTATION_OBJECT)); + + InitializeListHead(&WindowStationObject->DesktopListHead); + Status = RtlCreateAtomTable(37, &WindowStationObject->AtomTable); + WindowStationObject->Name = WindowStationName; + WindowStationObject->dwSessionId = NtCurrentPeb()->SessionId; + + if (InputWindowStation == NULL) + { + ERR("Initializing input window station\n"); + InputWindowStation = WindowStationObject; + + WindowStationObject->Flags &= ~WSS_NOIO; + + InitCursorImpl(); + } + else + { + WindowStationObject->Flags |= WSS_NOIO; + } + + TRACE("NtUserCreateWindowStation created object %p with name %wZ handle %p\n", WindowStation, &WindowStationObject->Name, WindowStation); - return WindowStation; + return WindowStation; }
/* @@ -533,31 +529,30 @@
HWINSTA APIENTRY NtUserOpenWindowStation( - POBJECT_ATTRIBUTES ObjectAttributes, - ACCESS_MASK dwDesiredAccess) -{ - HWINSTA hwinsta; - NTSTATUS Status; - - Status = ObOpenObjectByName( - ObjectAttributes, - ExWindowStationObjectType, - UserMode, - NULL, - dwDesiredAccess, - NULL, - (PVOID*)&hwinsta); - - if (!NT_SUCCESS(Status)) - { - ERR("NtUserOpenWindowStation failed\n"); - SetLastNtError(Status); - return 0; - } - - TRACE("Opened window station %wZ with handle %p\n", ObjectAttributes->ObjectName, hwinsta); - - return hwinsta; + POBJECT_ATTRIBUTES ObjectAttributes, + ACCESS_MASK dwDesiredAccess) +{ + HWINSTA hwinsta; + NTSTATUS Status; + + Status = ObOpenObjectByName(ObjectAttributes, + ExWindowStationObjectType, + UserMode, + NULL, + dwDesiredAccess, + NULL, + (PVOID*)&hwinsta); + + if (!NT_SUCCESS(Status)) + { + ERR("NtUserOpenWindowStation failed\n"); + SetLastNtError(Status); + return 0; + } + + TRACE("Opened window station %wZ with handle %p\n", ObjectAttributes->ObjectName, hwinsta); + + return hwinsta; }
/* @@ -584,44 +579,43 @@ BOOL APIENTRY NtUserCloseWindowStation( - HWINSTA hWinSta) -{ - PWINSTATION_OBJECT Object; - NTSTATUS Status; - - TRACE("NtUserCloseWindowStation called (%p)\n", hWinSta); - - if (hWinSta == UserGetProcessWindowStation()) - { + HWINSTA hWinSta) +{ + PWINSTATION_OBJECT Object; + NTSTATUS Status; + + TRACE("NtUserCloseWindowStation called (%p)\n", hWinSta); + + if (hWinSta == UserGetProcessWindowStation()) + { ERR("Attempted to close process window station\n"); - return FALSE; - } - - Status = IntValidateWindowStationHandle( - hWinSta, - UserMode, - 0, - &Object, - 0); - - if (!NT_SUCCESS(Status)) - { - ERR("Validation of window station handle (%p) failed\n", hWinSta); - return FALSE; - } - - ObDereferenceObject(Object); - - TRACE("Closing window station handle (%p)\n", hWinSta); - - Status = ObCloseHandle(hWinSta, UserMode); - if (!NT_SUCCESS(Status)) - { - SetLastNtError(Status); - return FALSE; - } - - return TRUE; + return FALSE; + } + + Status = IntValidateWindowStationHandle(hWinSta, + UserMode, + 0, + &Object, + 0); + + if (!NT_SUCCESS(Status)) + { + ERR("Validation of window station handle (%p) failed\n", hWinSta); + return FALSE; + } + + ObDereferenceObject(Object); + + TRACE("Closing window station handle (%p)\n", hWinSta); + + Status = ObCloseHandle(hWinSta, UserMode); + if (!NT_SUCCESS(Status)) + { + SetLastNtError(Status); + return FALSE; + } + + return TRUE; }
/* @@ -666,148 +660,146 @@
BOOL APIENTRY NtUserGetObjectInformation( - HANDLE hObject, - DWORD nIndex, - PVOID pvInformation, - DWORD nLength, - PDWORD nLengthNeeded) -{ - PWINSTATION_OBJECT WinStaObject; - PDESKTOP DesktopObject = NULL; - NTSTATUS Status; - PVOID pvData = NULL; - DWORD nDataSize = 0; - - _SEH2_TRY - { - if (nLengthNeeded) - ProbeForWrite(nLengthNeeded, sizeof(*nLengthNeeded), 1); - ProbeForWrite(pvInformation, nLength, 1); - } - _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) - { - SetLastNtError(_SEH2_GetExceptionCode()); - return FALSE; - } - _SEH2_END; - - /* try windowstation */ - TRACE("Trying to open window station %p\n", hObject); - Status = ObReferenceObjectByHandle( - hObject, - 0, - ExWindowStationObjectType, - UserMode, - (PVOID*)&WinStaObject, - NULL); - - if (Status == STATUS_OBJECT_TYPE_MISMATCH) - { - /* try desktop */ - TRACE("Trying to open desktop %p\n", hObject); - WinStaObject = NULL; - Status = IntValidateDesktopHandle( - hObject, - UserMode, - 0, - &DesktopObject); - } - - if (!NT_SUCCESS(Status)) - { - ERR("Failed: 0x%x\n", Status); - goto Exit; - } - - TRACE("WinSta or Desktop opened!!\n"); - - /* get data */ - switch (nIndex) - { - case UOI_FLAGS: - Status = STATUS_NOT_IMPLEMENTED; - ERR("UOI_FLAGS unimplemented!\n"); - break; - - case UOI_NAME: - if (WinStaObject != NULL) - { - pvData = WinStaObject->Name.Buffer; - nDataSize = WinStaObject->Name.Length + sizeof(WCHAR); - Status = STATUS_SUCCESS; - } - else if (DesktopObject != NULL) - { - pvData = DesktopObject->pDeskInfo->szDesktopName; - nDataSize = (wcslen(DesktopObject->pDeskInfo->szDesktopName) + 1) * sizeof(WCHAR); - Status = STATUS_SUCCESS; - } - else + HANDLE hObject, + DWORD nIndex, + PVOID pvInformation, + DWORD nLength, + PDWORD nLengthNeeded) +{ + PWINSTATION_OBJECT WinStaObject; + PDESKTOP DesktopObject = NULL; + NTSTATUS Status; + PVOID pvData = NULL; + DWORD nDataSize = 0; + + _SEH2_TRY + { + if (nLengthNeeded) + ProbeForWrite(nLengthNeeded, sizeof(*nLengthNeeded), 1); + ProbeForWrite(pvInformation, nLength, 1); + } + _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) + { + SetLastNtError(_SEH2_GetExceptionCode()); + return FALSE; + } + _SEH2_END; + + /* try windowstation */ + TRACE("Trying to open window station %p\n", hObject); + Status = ObReferenceObjectByHandle(hObject, + 0, + ExWindowStationObjectType, + UserMode, + (PVOID*)&WinStaObject, + NULL); + + if (Status == STATUS_OBJECT_TYPE_MISMATCH) + { + /* try desktop */ + TRACE("Trying to open desktop %p\n", hObject); + WinStaObject = NULL; + Status = IntValidateDesktopHandle(hObject, + UserMode, + 0, + &DesktopObject); + } + + if (!NT_SUCCESS(Status)) + { + ERR("Failed: 0x%x\n", Status); + goto Exit; + } + + TRACE("WinSta or Desktop opened!!\n"); + + /* get data */ + switch (nIndex) + { + case UOI_FLAGS: + Status = STATUS_NOT_IMPLEMENTED; + ERR("UOI_FLAGS unimplemented!\n"); + break; + + case UOI_NAME: + if (WinStaObject != NULL) + { + pvData = WinStaObject->Name.Buffer; + nDataSize = WinStaObject->Name.Length + sizeof(WCHAR); + Status = STATUS_SUCCESS; + } + else if (DesktopObject != NULL) + { + pvData = DesktopObject->pDeskInfo->szDesktopName; + nDataSize = (wcslen(DesktopObject->pDeskInfo->szDesktopName) + 1) * sizeof(WCHAR); + Status = STATUS_SUCCESS; + } + else + Status = STATUS_INVALID_PARAMETER; + break; + + case UOI_TYPE: + if (WinStaObject != NULL) + { + pvData = L"WindowStation"; + nDataSize = sizeof(L"WindowStation"); + Status = STATUS_SUCCESS; + } + else if (DesktopObject != NULL) + { + pvData = L"Desktop"; + nDataSize = sizeof(L"Desktop"); + Status = STATUS_SUCCESS; + } + else + Status = STATUS_INVALID_PARAMETER; + break; + + case UOI_USER_SID: + Status = STATUS_NOT_IMPLEMENTED; + ERR("UOI_USER_SID unimplemented!\n"); + break; + + default: Status = STATUS_INVALID_PARAMETER; - break; - - case UOI_TYPE: - if (WinStaObject != NULL) - { - pvData = L"WindowStation"; - nDataSize = sizeof(L"WindowStation"); - Status = STATUS_SUCCESS; - } - else if (DesktopObject != NULL) - { - pvData = L"Desktop"; - nDataSize = sizeof(L"Desktop"); - Status = STATUS_SUCCESS; - } - else - Status = STATUS_INVALID_PARAMETER; - break; - - case UOI_USER_SID: - Status = STATUS_NOT_IMPLEMENTED; - ERR("UOI_USER_SID unimplemented!\n"); - break; - - default: - Status = STATUS_INVALID_PARAMETER; - break; - } + break; + }
Exit: - if (Status == STATUS_SUCCESS && nLength < nDataSize) - Status = STATUS_BUFFER_TOO_SMALL; - - _SEH2_TRY - { - if (nLengthNeeded) - *nLengthNeeded = nDataSize; - - /* try to copy data to caller */ - if (Status == STATUS_SUCCESS) - { - TRACE("Trying to copy data to caller (len = %lu, len needed = %lu)\n", nLength, nDataSize); - RtlCopyMemory(pvInformation, pvData, nDataSize); - } - } - _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) - { - Status = _SEH2_GetExceptionCode(); - } - _SEH2_END; - - /* release objects */ - if (WinStaObject != NULL) - ObDereferenceObject(WinStaObject); - if (DesktopObject != NULL) - ObDereferenceObject(DesktopObject); - - if (!NT_SUCCESS(Status)) - { - SetLastNtError(Status); - return FALSE; - } - - return TRUE; + if (Status == STATUS_SUCCESS && nLength < nDataSize) + Status = STATUS_BUFFER_TOO_SMALL; + + _SEH2_TRY + { + if (nLengthNeeded) + *nLengthNeeded = nDataSize; + + /* try to copy data to caller */ + if (Status == STATUS_SUCCESS) + { + TRACE("Trying to copy data to caller (len = %lu, len needed = %lu)\n", nLength, nDataSize); + RtlCopyMemory(pvInformation, pvData, nDataSize); + } + } + _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) + { + Status = _SEH2_GetExceptionCode(); + } + _SEH2_END; + + /* release objects */ + if (WinStaObject != NULL) + ObDereferenceObject(WinStaObject); + if (DesktopObject != NULL) + ObDereferenceObject(DesktopObject); + + if (!NT_SUCCESS(Status)) + { + SetLastNtError(Status); + return FALSE; + } + + return TRUE; }
/* @@ -843,15 +835,15 @@ BOOL APIENTRY NtUserSetObjectInformation( - HANDLE hObject, - DWORD nIndex, - PVOID pvInformation, - DWORD nLength) -{ - /* FIXME: ZwQueryObject */ - /* FIXME: ZwSetInformationObject */ - SetLastNtError(STATUS_UNSUCCESSFUL); - return FALSE; + HANDLE hObject, + DWORD nIndex, + PVOID pvInformation, + DWORD nLength) +{ + /* FIXME: ZwQueryObject */ + /* FIXME: ZwSetInformationObject */ + SetLastNtError(STATUS_UNSUCCESSFUL); + return FALSE; }
@@ -883,7 +875,7 @@ HWINSTA APIENTRY NtUserGetProcessWindowStation(VOID) { - return UserGetProcessWindowStation(); + return UserGetProcessWindowStation(); }
BOOL FASTCALL @@ -900,64 +892,64 @@ /* Reference the new window station */ if(hWindowStation !=NULL) { - Status = IntValidateWindowStationHandle( hWindowStation, - UserMode, - 0, - &NewWinSta, - &ObjectHandleInfo); - if (!NT_SUCCESS(Status)) - { - TRACE("Validation of window station handle (%p) failed\n", - hWindowStation); - SetLastNtError(Status); - return FALSE; - } - } - - OldWinSta = ppi->prpwinsta; - hwinstaOld = PsGetProcessWin32WindowStation(ppi->peProcess); - - /* Dereference the previous window station */ - if(OldWinSta != NULL) - { - ObDereferenceObject(OldWinSta); - } - - /* Check if we have a stale handle (it should happen for console apps) */ - if(hwinstaOld != ppi->hwinsta) - { - ObCloseHandle(hwinstaOld, UserMode); - } - - /* - * FIXME: Don't allow changing the window station if there are threads that are attached to desktops and own GUI objects. - */ - - PsSetProcessWindowStation(ppi->peProcess, hWindowStation); - - ppi->prpwinsta = NewWinSta; - ppi->hwinsta = hWindowStation; - ppi->amwinsta = hWindowStation != NULL ? ObjectHandleInfo.GrantedAccess : 0; - TRACE("WS : Granted Access 0x%08lx\n",ppi->amwinsta); - - if (RtlAreAllAccessesGranted(ppi->amwinsta, WINSTA_READSCREEN)) - { - ppi->W32PF_flags |= W32PF_READSCREENACCESSGRANTED; - } - else - { - ppi->W32PF_flags &= ~W32PF_READSCREENACCESSGRANTED; - } - - if (NewWinSta && !(NewWinSta->Flags & WSS_NOIO) ) - { - ppi->W32PF_flags |= W32PF_IOWINSTA; - } - else // Might be closed if the handle is null. - { - ppi->W32PF_flags &= ~W32PF_IOWINSTA; - } - return TRUE; + Status = IntValidateWindowStationHandle(hWindowStation, + UserMode, + 0, + &NewWinSta, + &ObjectHandleInfo); + if (!NT_SUCCESS(Status)) + { + TRACE("Validation of window station handle (%p) failed\n", + hWindowStation); + SetLastNtError(Status); + return FALSE; + } + } + + OldWinSta = ppi->prpwinsta; + hwinstaOld = PsGetProcessWin32WindowStation(ppi->peProcess); + + /* Dereference the previous window station */ + if(OldWinSta != NULL) + { + ObDereferenceObject(OldWinSta); + } + + /* Check if we have a stale handle (it should happen for console apps) */ + if(hwinstaOld != ppi->hwinsta) + { + ObCloseHandle(hwinstaOld, UserMode); + } + + /* + * FIXME: Don't allow changing the window station if there are threads that are attached to desktops and own GUI objects. + */ + + PsSetProcessWindowStation(ppi->peProcess, hWindowStation); + + ppi->prpwinsta = NewWinSta; + ppi->hwinsta = hWindowStation; + ppi->amwinsta = hWindowStation != NULL ? ObjectHandleInfo.GrantedAccess : 0; + TRACE("WS : Granted Access 0x%08lx\n",ppi->amwinsta); + + if (RtlAreAllAccessesGranted(ppi->amwinsta, WINSTA_READSCREEN)) + { + ppi->W32PF_flags |= W32PF_READSCREENACCESSGRANTED; + } + else + { + ppi->W32PF_flags &= ~W32PF_READSCREENACCESSGRANTED; + } + + if (NewWinSta && !(NewWinSta->Flags & WSS_NOIO) ) + { + ppi->W32PF_flags |= W32PF_IOWINSTA; + } + else // Might be closed if the handle is null. + { + ppi->W32PF_flags &= ~W32PF_IOWINSTA; + } + return TRUE; }
/* @@ -1002,37 +994,36 @@ BOOL APIENTRY NtUserLockWindowStation(HWINSTA hWindowStation) { - PWINSTATION_OBJECT Object; - NTSTATUS Status; - - TRACE("About to set process window station with handle (%p)\n", - hWindowStation); - - if (gpidLogon != PsGetCurrentProcessId()) - { - ERR("Unauthorized process attempted to lock the window station!\n"); - EngSetLastError(ERROR_ACCESS_DENIED); - return FALSE; - } - - Status = IntValidateWindowStationHandle( - hWindowStation, - UserMode, - 0, - &Object, - 0); - if (!NT_SUCCESS(Status)) - { - TRACE("Validation of window station handle (%p) failed\n", - hWindowStation); - SetLastNtError(Status); - return FALSE; - } - - Object->Flags |= WSS_LOCKED; - - ObDereferenceObject(Object); - return TRUE; + PWINSTATION_OBJECT Object; + NTSTATUS Status; + + TRACE("About to set process window station with handle (%p)\n", + hWindowStation); + + if (gpidLogon != PsGetCurrentProcessId()) + { + ERR("Unauthorized process attempted to lock the window station!\n"); + EngSetLastError(ERROR_ACCESS_DENIED); + return FALSE; + } + + Status = IntValidateWindowStationHandle(hWindowStation, + UserMode, + 0, + &Object, + 0); + if (!NT_SUCCESS(Status)) + { + TRACE("Validation of window station handle (%p) failed\n", + hWindowStation); + SetLastNtError(Status); + return FALSE; + } + + Object->Flags |= WSS_LOCKED; + + ObDereferenceObject(Object); + return TRUE; }
/* @@ -1047,315 +1038,319 @@ BOOL APIENTRY NtUserUnlockWindowStation(HWINSTA hWindowStation) { - PWINSTATION_OBJECT Object; - NTSTATUS Status; - BOOL Ret; - - TRACE("About to set process window station with handle (%p)\n", - hWindowStation); - - if (gpidLogon != PsGetCurrentProcessId()) - { - ERR("Unauthorized process attempted to unlock the window station!\n"); - EngSetLastError(ERROR_ACCESS_DENIED); - return FALSE; - } - - Status = IntValidateWindowStationHandle( - hWindowStation, - UserMode, - 0, - &Object, - 0); - if (!NT_SUCCESS(Status)) - { - TRACE("Validation of window station handle (%p) failed\n", - hWindowStation); - SetLastNtError(Status); - return FALSE; - } - - Ret = (Object->Flags & WSS_LOCKED) == WSS_LOCKED; - Object->Flags &= ~WSS_LOCKED; - - ObDereferenceObject(Object); - return Ret; + PWINSTATION_OBJECT Object; + NTSTATUS Status; + BOOL Ret; + + TRACE("About to set process window station with handle (%p)\n", + hWindowStation); + + if (gpidLogon != PsGetCurrentProcessId()) + { + ERR("Unauthorized process attempted to unlock the window station!\n"); + EngSetLastError(ERROR_ACCESS_DENIED); + return FALSE; + } + + Status = IntValidateWindowStationHandle(hWindowStation, + UserMode, + 0, + &Object, + 0); + if (!NT_SUCCESS(Status)) + { + TRACE("Validation of window station handle (%p) failed\n", + hWindowStation); + SetLastNtError(Status); + return FALSE; + } + + Ret = (Object->Flags & WSS_LOCKED) == WSS_LOCKED; + Object->Flags &= ~WSS_LOCKED; + + ObDereferenceObject(Object); + return Ret; }
static NTSTATUS FASTCALL BuildWindowStationNameList( - ULONG dwSize, - PVOID lpBuffer, - PULONG pRequiredSize) -{ - OBJECT_ATTRIBUTES ObjectAttributes; - NTSTATUS Status; - HANDLE DirectoryHandle; - char InitialBuffer[256], *Buffer; - ULONG Context, ReturnLength, BufferSize; - DWORD EntryCount; - POBJECT_DIRECTORY_INFORMATION DirEntry; - WCHAR NullWchar; - - /* - * Try to open the directory. - */ - InitializeObjectAttributes( - &ObjectAttributes, - &gustrWindowStationsDir, - OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, - NULL, - NULL); - - Status = ZwOpenDirectoryObject( - &DirectoryHandle, - DIRECTORY_QUERY, - &ObjectAttributes); - - if (!NT_SUCCESS(Status)) - { - return Status; - } - - /* First try to query the directory using a fixed-size buffer */ - Context = 0; - Buffer = NULL; - Status = ZwQueryDirectoryObject(DirectoryHandle, InitialBuffer, sizeof(InitialBuffer), - FALSE, TRUE, &Context, &ReturnLength); - if (NT_SUCCESS(Status)) - { - if (STATUS_NO_MORE_ENTRIES == ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, - FALSE, &Context, NULL)) - { - /* Our fixed-size buffer is large enough */ - Buffer = InitialBuffer; - } - } - - if (NULL == Buffer) - { - /* Need a larger buffer, check how large exactly */ - Status = ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, TRUE, &Context, - &ReturnLength); - if (!NT_SUCCESS(Status)) - { - ERR("ZwQueryDirectoryObject failed\n"); - ZwClose(DirectoryHandle); - return Status; - } - - BufferSize = ReturnLength; - Buffer = ExAllocatePoolWithTag(PagedPool, BufferSize, TAG_WINSTA); - if (NULL == Buffer) - { - ZwClose(DirectoryHandle); - return STATUS_NO_MEMORY; - } - - /* We should have a sufficiently large buffer now */ - Context = 0; - Status = ZwQueryDirectoryObject(DirectoryHandle, Buffer, BufferSize, - FALSE, TRUE, &Context, &ReturnLength); - if (! NT_SUCCESS(Status) || - STATUS_NO_MORE_ENTRIES != ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, - FALSE, &Context, NULL)) - { - /* Something went wrong, maybe someone added a directory entry? Just give up. */ - ExFreePoolWithTag(Buffer, TAG_WINSTA); - ZwClose(DirectoryHandle); - return NT_SUCCESS(Status) ? STATUS_INTERNAL_ERROR : Status; - } - } - - ZwClose(DirectoryHandle); - - /* - * Count the required size of buffer. - */ - ReturnLength = sizeof(DWORD); - EntryCount = 0; - for (DirEntry = (POBJECT_DIRECTORY_INFORMATION) Buffer; 0 != DirEntry->Name.Length; + ULONG dwSize, + PVOID lpBuffer, + PULONG pRequiredSize) +{ + OBJECT_ATTRIBUTES ObjectAttributes; + NTSTATUS Status; + HANDLE DirectoryHandle; + char InitialBuffer[256], *Buffer; + ULONG Context, ReturnLength, BufferSize; + DWORD EntryCount; + POBJECT_DIRECTORY_INFORMATION DirEntry; + WCHAR NullWchar; + + /* + * Try to open the directory. + */ + InitializeObjectAttributes(&ObjectAttributes, + &gustrWindowStationsDir, + OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, + NULL, + NULL); + + Status = ZwOpenDirectoryObject(&DirectoryHandle, + DIRECTORY_QUERY, + &ObjectAttributes); + + if (!NT_SUCCESS(Status)) + { + return Status; + } + + /* First try to query the directory using a fixed-size buffer */ + Context = 0; + Buffer = NULL; + Status = ZwQueryDirectoryObject(DirectoryHandle, + InitialBuffer, + sizeof(InitialBuffer), + FALSE, + TRUE, + &Context, + &ReturnLength); + if (NT_SUCCESS(Status)) + { + if (STATUS_NO_MORE_ENTRIES == ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, + FALSE, &Context, NULL)) + { + /* Our fixed-size buffer is large enough */ + Buffer = InitialBuffer; + } + } + + if (NULL == Buffer) + { + /* Need a larger buffer, check how large exactly */ + Status = ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, TRUE, &Context, + &ReturnLength); + if (!NT_SUCCESS(Status)) + { + ERR("ZwQueryDirectoryObject failed\n"); + ZwClose(DirectoryHandle); + return Status; + } + + BufferSize = ReturnLength; + Buffer = ExAllocatePoolWithTag(PagedPool, BufferSize, TAG_WINSTA); + if (NULL == Buffer) + { + ZwClose(DirectoryHandle); + return STATUS_NO_MEMORY; + } + + /* We should have a sufficiently large buffer now */ + Context = 0; + Status = ZwQueryDirectoryObject(DirectoryHandle, Buffer, BufferSize, + FALSE, TRUE, &Context, &ReturnLength); + if (! NT_SUCCESS(Status) || + STATUS_NO_MORE_ENTRIES != ZwQueryDirectoryObject(DirectoryHandle, NULL, 0, FALSE, + FALSE, &Context, NULL)) + { + /* Something went wrong, maybe someone added a directory entry? Just give up. */ + ExFreePoolWithTag(Buffer, TAG_WINSTA); + ZwClose(DirectoryHandle); + return NT_SUCCESS(Status) ? STATUS_INTERNAL_ERROR : Status; + } + } + + ZwClose(DirectoryHandle); + + /* + * Count the required size of buffer. + */ + ReturnLength = sizeof(DWORD); + EntryCount = 0; + for (DirEntry = (POBJECT_DIRECTORY_INFORMATION) Buffer; + 0 != DirEntry->Name.Length; DirEntry++) - { - ReturnLength += DirEntry->Name.Length + sizeof(WCHAR); - EntryCount++; - } - TRACE("Required size: %lu Entry count: %lu\n", ReturnLength, EntryCount); - if (NULL != pRequiredSize) - { - Status = MmCopyToCaller(pRequiredSize, &ReturnLength, sizeof(ULONG)); - if (! NT_SUCCESS(Status)) - { - if (Buffer != InitialBuffer) - { + { + ReturnLength += DirEntry->Name.Length + sizeof(WCHAR); + EntryCount++; + } + TRACE("Required size: %lu Entry count: %lu\n", ReturnLength, EntryCount); + if (NULL != pRequiredSize) + { + Status = MmCopyToCaller(pRequiredSize, &ReturnLength, sizeof(ULONG)); + if (! NT_SUCCESS(Status)) + { + if (Buffer != InitialBuffer) + { + ExFreePoolWithTag(Buffer, TAG_WINSTA); + } + return STATUS_BUFFER_TOO_SMALL; + } + } + + /* + * Check if the supplied buffer is large enough. + */ + if (dwSize < ReturnLength) + { + if (Buffer != InitialBuffer) + { ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - return STATUS_BUFFER_TOO_SMALL; - } - } - - /* - * Check if the supplied buffer is large enough. - */ - if (dwSize < ReturnLength) - { - if (Buffer != InitialBuffer) - { - ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - return STATUS_BUFFER_TOO_SMALL; - } - - /* - * Generate the resulting buffer contents. - */ - Status = MmCopyToCaller(lpBuffer, &EntryCount, sizeof(DWORD)); - if (! NT_SUCCESS(Status)) - { - if (Buffer != InitialBuffer) - { - ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - return Status; - } - lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(DWORD)); - - NullWchar = L'\0'; - for (DirEntry = (POBJECT_DIRECTORY_INFORMATION) Buffer; 0 != DirEntry->Name.Length; + } + return STATUS_BUFFER_TOO_SMALL; + } + + /* + * Generate the resulting buffer contents. + */ + Status = MmCopyToCaller(lpBuffer, &EntryCount, sizeof(DWORD)); + if (! NT_SUCCESS(Status)) + { + if (Buffer != InitialBuffer) + { + ExFreePoolWithTag(Buffer, TAG_WINSTA); + } + return Status; + } + lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(DWORD)); + + NullWchar = L'\0'; + for (DirEntry = (POBJECT_DIRECTORY_INFORMATION) Buffer; + 0 != DirEntry->Name.Length; DirEntry++) - { - Status = MmCopyToCaller(lpBuffer, DirEntry->Name.Buffer, DirEntry->Name.Length); - if (! NT_SUCCESS(Status)) - { - if (Buffer != InitialBuffer) - { - ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - return Status; - } - lpBuffer = (PVOID) ((PCHAR) lpBuffer + DirEntry->Name.Length); - Status = MmCopyToCaller(lpBuffer, &NullWchar, sizeof(WCHAR)); - if (! NT_SUCCESS(Status)) - { - if (Buffer != InitialBuffer) - { - ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - return Status; - } - lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(WCHAR)); - } - - /* - * Clean up - */ - if (Buffer != InitialBuffer) - { - ExFreePoolWithTag(Buffer, TAG_WINSTA); - } - - return STATUS_SUCCESS; + { + Status = MmCopyToCaller(lpBuffer, DirEntry->Name.Buffer, DirEntry->Name.Length); + if (! NT_SUCCESS(Status)) + { + if (Buffer != InitialBuffer) + { + ExFreePoolWithTag(Buffer, TAG_WINSTA); + } + return Status; + } + lpBuffer = (PVOID) ((PCHAR) lpBuffer + DirEntry->Name.Length); + Status = MmCopyToCaller(lpBuffer, &NullWchar, sizeof(WCHAR)); + if (! NT_SUCCESS(Status)) + { + if (Buffer != InitialBuffer) + { + ExFreePoolWithTag(Buffer, TAG_WINSTA); + } + return Status; + } + lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(WCHAR)); + } + + /* + * Clean up + */ + if (Buffer != InitialBuffer) + { + ExFreePoolWithTag(Buffer, TAG_WINSTA); + } + + return STATUS_SUCCESS; }
static NTSTATUS FASTCALL BuildDesktopNameList( - HWINSTA hWindowStation, - ULONG dwSize, - PVOID lpBuffer, - PULONG pRequiredSize) -{ - NTSTATUS Status; - PWINSTATION_OBJECT WindowStation; - PLIST_ENTRY DesktopEntry; - PDESKTOP DesktopObject; - DWORD EntryCount; - ULONG ReturnLength; - WCHAR NullWchar; - UNICODE_STRING DesktopName; - - Status = IntValidateWindowStationHandle(hWindowStation, - UserMode, - 0, - &WindowStation, - 0); - if (! NT_SUCCESS(Status)) - { - return Status; - } - - /* - * Count the required size of buffer. - */ - ReturnLength = sizeof(DWORD); - EntryCount = 0; - for (DesktopEntry = WindowStation->DesktopListHead.Flink; + HWINSTA hWindowStation, + ULONG dwSize, + PVOID lpBuffer, + PULONG pRequiredSize) +{ + NTSTATUS Status; + PWINSTATION_OBJECT WindowStation; + PLIST_ENTRY DesktopEntry; + PDESKTOP DesktopObject; + DWORD EntryCount; + ULONG ReturnLength; + WCHAR NullWchar; + UNICODE_STRING DesktopName; + + Status = IntValidateWindowStationHandle(hWindowStation, + UserMode, + 0, + &WindowStation, + 0); + if (! NT_SUCCESS(Status)) + { + return Status; + } + + /* + * Count the required size of buffer. + */ + ReturnLength = sizeof(DWORD); + EntryCount = 0; + for (DesktopEntry = WindowStation->DesktopListHead.Flink; DesktopEntry != &WindowStation->DesktopListHead; DesktopEntry = DesktopEntry->Flink) - { - DesktopObject = CONTAINING_RECORD(DesktopEntry, DESKTOP, ListEntry); - RtlInitUnicodeString(&DesktopName, DesktopObject->pDeskInfo->szDesktopName); - ReturnLength += DesktopName.Length + sizeof(WCHAR); - EntryCount++; - } - TRACE("Required size: %lu Entry count: %lu\n", ReturnLength, EntryCount); - if (NULL != pRequiredSize) - { - Status = MmCopyToCaller(pRequiredSize, &ReturnLength, sizeof(ULONG)); - if (! NT_SUCCESS(Status)) - { - ObDereferenceObject(WindowStation); - return STATUS_BUFFER_TOO_SMALL; - } - } - - /* - * Check if the supplied buffer is large enough. - */ - if (dwSize < ReturnLength) - { - ObDereferenceObject(WindowStation); - return STATUS_BUFFER_TOO_SMALL; - } - - /* - * Generate the resulting buffer contents. - */ - Status = MmCopyToCaller(lpBuffer, &EntryCount, sizeof(DWORD)); - if (! NT_SUCCESS(Status)) - { - ObDereferenceObject(WindowStation); - return Status; - } - lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(DWORD)); - - NullWchar = L'\0'; - for (DesktopEntry = WindowStation->DesktopListHead.Flink; + { + DesktopObject = CONTAINING_RECORD(DesktopEntry, DESKTOP, ListEntry); + RtlInitUnicodeString(&DesktopName, DesktopObject->pDeskInfo->szDesktopName); + ReturnLength += DesktopName.Length + sizeof(WCHAR); + EntryCount++; + } + TRACE("Required size: %lu Entry count: %lu\n", ReturnLength, EntryCount); + if (NULL != pRequiredSize) + { + Status = MmCopyToCaller(pRequiredSize, &ReturnLength, sizeof(ULONG)); + if (! NT_SUCCESS(Status)) + { + ObDereferenceObject(WindowStation); + return STATUS_BUFFER_TOO_SMALL; + } + } + + /* + * Check if the supplied buffer is large enough. + */ + if (dwSize < ReturnLength) + { + ObDereferenceObject(WindowStation); + return STATUS_BUFFER_TOO_SMALL; + } + + /* + * Generate the resulting buffer contents. + */ + Status = MmCopyToCaller(lpBuffer, &EntryCount, sizeof(DWORD)); + if (! NT_SUCCESS(Status)) + { + ObDereferenceObject(WindowStation); + return Status; + } + lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(DWORD)); + + NullWchar = L'\0'; + for (DesktopEntry = WindowStation->DesktopListHead.Flink; DesktopEntry != &WindowStation->DesktopListHead; DesktopEntry = DesktopEntry->Flink) - { - DesktopObject = CONTAINING_RECORD(DesktopEntry, DESKTOP, ListEntry); - RtlInitUnicodeString(&DesktopName, DesktopObject->pDeskInfo->szDesktopName); - Status = MmCopyToCaller(lpBuffer, DesktopName.Buffer, DesktopName.Length); - if (! NT_SUCCESS(Status)) - { - ObDereferenceObject(WindowStation); - return Status; - } - lpBuffer = (PVOID) ((PCHAR)lpBuffer + DesktopName.Length); - Status = MmCopyToCaller(lpBuffer, &NullWchar, sizeof(WCHAR)); - if (! NT_SUCCESS(Status)) - { - ObDereferenceObject(WindowStation); - return Status; - } - lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(WCHAR)); - } - - /* - * Clean up and return - */ - ObDereferenceObject(WindowStation); - return STATUS_SUCCESS; + { + DesktopObject = CONTAINING_RECORD(DesktopEntry, DESKTOP, ListEntry); + RtlInitUnicodeString(&DesktopName, DesktopObject->pDeskInfo->szDesktopName); + Status = MmCopyToCaller(lpBuffer, DesktopName.Buffer, DesktopName.Length); + if (! NT_SUCCESS(Status)) + { + ObDereferenceObject(WindowStation); + return Status; + } + lpBuffer = (PVOID) ((PCHAR)lpBuffer + DesktopName.Length); + Status = MmCopyToCaller(lpBuffer, &NullWchar, sizeof(WCHAR)); + if (! NT_SUCCESS(Status)) + { + ObDereferenceObject(WindowStation); + return Status; + } + lpBuffer = (PVOID) ((PCHAR) lpBuffer + sizeof(WCHAR)); + } + + /* + * Clean up and return + */ + ObDereferenceObject(WindowStation); + return STATUS_SUCCESS; }
/* @@ -1386,15 +1381,15 @@
NTSTATUS APIENTRY NtUserBuildNameList( - HWINSTA hWindowStation, - ULONG dwSize, - PVOID lpBuffer, - PULONG pRequiredSize) -{ - /* The WindowStation name list and desktop name list are build in completely - different ways. Call the appropriate function */ - return NULL == hWindowStation ? BuildWindowStationNameList(dwSize, lpBuffer, pRequiredSize) : - BuildDesktopNameList(hWindowStation, dwSize, lpBuffer, pRequiredSize); + HWINSTA hWindowStation, + ULONG dwSize, + PVOID lpBuffer, + PULONG pRequiredSize) +{ + /* The WindowStation name list and desktop name list are build in completely + different ways. Call the appropriate function */ + return NULL == hWindowStation ? BuildWindowStationNameList(dwSize, lpBuffer, pRequiredSize) : + BuildDesktopNameList(hWindowStation, dwSize, lpBuffer, pRequiredSize); }
/* @@ -1438,77 +1433,77 @@
UserLeave();
- return ret; + return ret; }
BOOL APIENTRY NtUserSetWindowStationUser( - HWINSTA hWindowStation, - PLUID pluid, - PSID psid, - DWORD size) -{ - NTSTATUS Status; - PWINSTATION_OBJECT WindowStation = NULL; - BOOL Ret = FALSE; - - UserEnterExclusive(); - - if (gpidLogon != PsGetCurrentProcessId()) - { - EngSetLastError(ERROR_ACCESS_DENIED); - goto Leave; - } - - Status = IntValidateWindowStationHandle(hWindowStation, - UserMode, - 0, - &WindowStation, - 0); - if (!NT_SUCCESS(Status)) - { - goto Leave; - } - - if (WindowStation->psidUser) - { - ExFreePoolWithTag(WindowStation->psidUser, USERTAG_SECURITY); - } - - WindowStation->psidUser = ExAllocatePoolWithTag(PagedPool, size, USERTAG_SECURITY); - if (WindowStation->psidUser == NULL) - { - EngSetLastError(ERROR_OUTOFMEMORY); - goto Leave; - } - - _SEH2_TRY - { - ProbeForRead( psid, size, 1); - ProbeForRead( pluid, sizeof(LUID), 1); - - RtlCopyMemory(WindowStation->psidUser, psid, size); - WindowStation->luidUser = *pluid; - } - _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) - { - Status = _SEH2_GetExceptionCode(); - } - _SEH2_END; - - if (!NT_SUCCESS(Status)) - { - ExFreePoolWithTag(WindowStation->psidUser, USERTAG_SECURITY); - WindowStation->psidUser = 0; - goto Leave; - } - - Ret = TRUE; + HWINSTA hWindowStation, + PLUID pluid, + PSID psid, + DWORD size) +{ + NTSTATUS Status; + PWINSTATION_OBJECT WindowStation = NULL; + BOOL Ret = FALSE; + + UserEnterExclusive(); + + if (gpidLogon != PsGetCurrentProcessId()) + { + EngSetLastError(ERROR_ACCESS_DENIED); + goto Leave; + } + + Status = IntValidateWindowStationHandle(hWindowStation, + UserMode, + 0, + &WindowStation, + 0); + if (!NT_SUCCESS(Status)) + { + goto Leave; + } + + if (WindowStation->psidUser) + { + ExFreePoolWithTag(WindowStation->psidUser, USERTAG_SECURITY); + } + + WindowStation->psidUser = ExAllocatePoolWithTag(PagedPool, size, USERTAG_SECURITY); + if (WindowStation->psidUser == NULL) + { + EngSetLastError(ERROR_OUTOFMEMORY); + goto Leave; + } + + _SEH2_TRY + { + ProbeForRead( psid, size, 1); + ProbeForRead( pluid, sizeof(LUID), 1); + + RtlCopyMemory(WindowStation->psidUser, psid, size); + WindowStation->luidUser = *pluid; + } + _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) + { + Status = _SEH2_GetExceptionCode(); + } + _SEH2_END; + + if (!NT_SUCCESS(Status)) + { + ExFreePoolWithTag(WindowStation->psidUser, USERTAG_SECURITY); + WindowStation->psidUser = 0; + goto Leave; + } + + Ret = TRUE;
Leave: - if (WindowStation) ObDereferenceObject(WindowStation); - UserLeave(); - return Ret; + if (WindowStation) ObDereferenceObject(WindowStation); + UserLeave(); + return Ret; }
-
tfaber＠svn.reactos.org