[PATCH 3/7] win32u: Use the thread input shared memory in GetKeyboardState.

[Rémi Bernon]

From: Rémi Bernon

---
 dlls/win32u/input.c | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/dlls/win32u/input.c b/dlls/win32u/input.c
index ba54cd3d7bb..36d23eb1d1b 100644
--- a/dlls/win32u/input.c
+++ b/dlls/win32u/input.c
@@ -972,21 +972,19 @@ SHORT WINAPI NtUserGetKeyState( INT vkey )
  */
 BOOL WINAPI NtUserGetKeyboardState( BYTE *state )
 {
-    BOOL ret;
+    struct object_lock lock = OBJECT_LOCK_INIT;
+    const input_shm_t *input_shm;
+    NTSTATUS status;
     UINT i;

TRACE("(%p)\n", state);

-    memset( state, 0, 256 );
-    SERVER_START_REQ( get_key_state )
-    {
-        req->key = -1;
-        wine_server_set_reply( req, state, 256 );
-        ret = !wine_server_call_err( req );
-        for (i = 0; i < 256; i++) state[i] &= 0x81;
-    }
-    SERVER_END_REQ;
-    return ret;
+    while ((status = get_shared_input( GetCurrentThreadId(), &lock, &input_shm )) == STATUS_PENDING)
+        memcpy( state, (const void *)input_shm->keystate, 256 );
+    if (status) memset( state, 0, 256 );
+
+    for (i = 0; i < 256; i++) state[i] &= 0x81;
+    return !status;
 }

/***********************************************************************