From 4460003077460e40e8d04ff3f6f27e252581f463 Mon Sep 17 00:00:00 2001
From: James Tabor <james.tabor@reactos.org>
Date: Fri, 11 Jul 2008 01:24:47 +0000
Subject: [PATCH] Improve NtUserWaitForInputIdle, still a work in progress.

svn path=/trunk/; revision=34419
---
 .../subsystems/win32/win32k/ntuser/message.c  | 27 ++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/reactos/subsystems/win32/win32k/ntuser/message.c b/reactos/subsystems/win32/win32k/ntuser/message.c
index d32339f9ec9..65b595fd273 100644
--- a/reactos/subsystems/win32/win32k/ntuser/message.c
+++ b/reactos/subsystems/win32/win32k/ntuser/message.c
@@ -1858,11 +1858,35 @@ NtUserWaitForInputIdle(
   DWORD ret;
   LARGE_INTEGER Timeout; 
   HANDLE handles[2];
+  PEPROCESS Process;
+  PW32PROCESS W32Process;
+  NTSTATUS Status;
 
   UserEnterExclusive();
 
+  Status = ObReferenceObjectByHandle(hProcess,
+                                     PROCESS_QUERY_INFORMATION,
+                                     PsProcessType,
+                                     UserMode,
+                                     (PVOID*)&Process,
+                                     NULL);
+
+  if (!NT_SUCCESS(Status))
+  {
+     SetLastNtError(Status);
+     return (DWORD)-1;
+  }
+
+  W32Process = (PW32PROCESS)Process->Win32Process;
+  if (!W32Process)
+  {
+      ObDereferenceObject(Process);
+      SetLastWin32Error(ERROR_INVALID_PARAMETER);
+      return (DWORD)-1;
+  }
+
   handles[0] = hProcess;
-  handles[1] = &PsGetCurrentProcessWin32Process()->InputIdleEvent; // Fixme!
+  handles[1] = &W32Process->InputIdleEvent; // Fixme!
 
   if (!handles[1]) return 0;  /* no event to wait on */
 
@@ -1920,6 +1944,7 @@ NtUserWaitForInputIdle(
   }
   while (1);
 WaitExit:
+  ObDereferenceObject(Process);
   UserLeave();
   return ret;
 }