From ad68bb1dc2defbfab845e57e829b383e48410d18 Mon Sep 17 00:00:00 2001
From: Eric Kohl <eric.kohl@reactos.org>
Date: Thu, 21 Jul 2016 20:53:43 +0000
Subject: [PATCH] [USETUP] Keyboard Modifier LEDs not working in usetup. Patch
 by Carlo Bramini. CORE-3368 #resolve #comment Thanks a lot!

svn path=/trunk/; revision=71976
---
 .../base/setup/usetup/native/utils/console.c  |  2 +-
 .../base/setup/usetup/native/utils/keytrans.c | 77 ++++++++++++++-----
 .../base/setup/usetup/native/utils/keytrans.h |  2 +-
 3 files changed, 58 insertions(+), 23 deletions(-)

diff --git a/reactos/base/setup/usetup/native/utils/console.c b/reactos/base/setup/usetup/native/utils/console.c
index 5aec605f168..aa2a5661dad 100644
--- a/reactos/base/setup/usetup/native/utils/console.c
+++ b/reactos/base/setup/usetup/native/utils/console.c
@@ -218,7 +218,7 @@ ReadConsoleInput(
         return FALSE;
 
     lpBuffer->EventType = KEY_EVENT;
-    Status = IntTranslateKey(&InputData, &lpBuffer->Event.KeyEvent);
+    Status = IntTranslateKey(hConsoleInput, &InputData, &lpBuffer->Event.KeyEvent);
     if (!NT_SUCCESS(Status))
         return FALSE;
 
diff --git a/reactos/base/setup/usetup/native/utils/keytrans.c b/reactos/base/setup/usetup/native/utils/keytrans.c
index e047bbe180e..61279634b30 100644
--- a/reactos/base/setup/usetup/native/utils/keytrans.c
+++ b/reactos/base/setup/usetup/native/utils/keytrans.c
@@ -117,13 +117,6 @@ static WORD KeyTableEnhanced[] = {
 	0,		0,		0,		0
 };
 
-/*
- * Note how the keyboard lights are not handled, so while NUMLOCK_ON can
- * be on, the light will never be. If this starts to be a problem it can be
- * fixed, but it's too much work for too little gain to do now.
- * Look in win32k/ntuser/input.c for an example.
- */
-
 static WORD KeyTableNumlock[] = {
 /* 0x00 */
 	0,		0,		0,		0,
@@ -255,13 +248,16 @@ SCANTOASCII ScanToAscii[] = {
 
 
 static void
-IntUpdateControlKeyState(LPDWORD State, PKEYBOARD_INPUT_DATA InputData)
+IntUpdateControlKeyState(HANDLE hConsoleInput, LPDWORD State, PKEYBOARD_INPUT_DATA InputData)
 {
 	DWORD Value = 0;
+    DWORD oldState, newState;
 
 	if (InputData->Flags & KEY_E1) /* Only the pause key has E1 */
 		return;
 
+    oldState = newState = *State;
+
 	if (!(InputData->Flags & KEY_E0)) {
 		switch (InputData->MakeCode) {
 			case 0x2a:
@@ -278,22 +274,19 @@ IntUpdateControlKeyState(LPDWORD State, PKEYBOARD_INPUT_DATA InputData)
 				break;
 
 			case 0x3A:
-				Value = CAPSLOCK_ON;
 				if (!(InputData->Flags & KEY_BREAK))
-					*State ^= Value;
-				return;
+					newState ^= CAPSLOCK_ON;
+				break;
 
 			case 0x45:
-				Value = NUMLOCK_ON;
 				if (!(InputData->Flags & KEY_BREAK))
-					*State ^= Value;
-				return;
+					newState ^= NUMLOCK_ON;
+				break;
 
 			case 0x46:
-				Value = SCROLLLOCK_ON;
 				if (!(InputData->Flags & KEY_BREAK))
-					*State ^= Value;
-				return;
+					newState ^= SCROLLLOCK_ON;
+				break;
 
 			default:
 				return;
@@ -313,10 +306,52 @@ IntUpdateControlKeyState(LPDWORD State, PKEYBOARD_INPUT_DATA InputData)
 		}
 	}
 
+    /* Check if the state of the indicators has been changed */
+    if ((oldState ^ newState) & (NUMLOCK_ON | CAPSLOCK_ON | SCROLLLOCK_ON))
+    {
+        IO_STATUS_BLOCK               IoStatusBlock;
+        NTSTATUS                      Status;
+        KEYBOARD_INDICATOR_PARAMETERS kip;
+
+        kip.LedFlags = 0;
+        kip.UnitId   = 0;
+
+        if ((newState & NUMLOCK_ON))
+            kip.LedFlags |= KEYBOARD_NUM_LOCK_ON;
+
+        if ((newState & CAPSLOCK_ON))
+            kip.LedFlags |= KEYBOARD_CAPS_LOCK_ON;
+
+        if ((newState & SCROLLLOCK_ON))
+            kip.LedFlags |= KEYBOARD_SCROLL_LOCK_ON;
+
+        /* Update the state of the leds on primary keyboard */
+        DPRINT("NtDeviceIoControlFile dwLeds=%x\n", kip.LedFlags);
+
+        Status = NtDeviceIoControlFile(
+              hConsoleInput,
+              NULL,
+              NULL,
+              NULL,
+              &IoStatusBlock,
+              IOCTL_KEYBOARD_SET_INDICATORS,
+		      &kip,
+              sizeof(kip),
+		      NULL,
+              0);
+
+        if (!NT_SUCCESS(Status))
+        {
+            DPRINT1("NtDeviceIoControlFile(IOCTL_KEYBOARD_SET_INDICATORS) failed (Status %lx)\n", Status);
+        }
+    } else
+    /* Normal press/release state handling */
 	if (InputData->Flags & KEY_BREAK)
-		*State &= ~Value;
+		newState &= ~Value;
 	else
-		*State |= Value;
+		newState |= Value;
+
+    *State = newState;
 }
 
 static DWORD
@@ -379,7 +414,7 @@ IntAsciiFromInput(PKEYBOARD_INPUT_DATA InputData, DWORD KeyState)
  * in the app so I'll just fill the others with somewhat sane values
  */
 NTSTATUS
-IntTranslateKey(PKEYBOARD_INPUT_DATA InputData, KEY_EVENT_RECORD *Event)
+IntTranslateKey(HANDLE hConsoleInput, PKEYBOARD_INPUT_DATA InputData, KEY_EVENT_RECORD *Event)
 {
 	static DWORD dwControlKeyState;
 
@@ -395,7 +430,7 @@ IntTranslateKey(PKEYBOARD_INPUT_DATA InputData, KEY_EVENT_RECORD *Event)
 
 	DPRINT("Translating: %x\n", InputData->MakeCode);
 
-	IntUpdateControlKeyState(&dwControlKeyState, InputData);
+	IntUpdateControlKeyState(hConsoleInput, &dwControlKeyState, InputData);
 	Event->dwControlKeyState = dwControlKeyState;
 
 	if (InputData->Flags & KEY_E0)
diff --git a/reactos/base/setup/usetup/native/utils/keytrans.h b/reactos/base/setup/usetup/native/utils/keytrans.h
index 9198a5f4eda..893a6995e5c 100644
--- a/reactos/base/setup/usetup/native/utils/keytrans.h
+++ b/reactos/base/setup/usetup/native/utils/keytrans.h
@@ -28,6 +28,6 @@
 #include <ntddkbd.h>
 
 NTSTATUS
-IntTranslateKey(PKEYBOARD_INPUT_DATA InputData, KEY_EVENT_RECORD *Event);
+IntTranslateKey(HANDLE hConsoleInput, PKEYBOARD_INPUT_DATA InputData, KEY_EVENT_RECORD *Event);
 
 /* EOF */
-- 
2.17.1