Don't return characters is key is up
[reactos.git] / reactos / subsys / win32k / ntuser / keyboard.c
index 8631420..835a5b8 100644 (file)
@@ -16,8 +16,7 @@
  *  along with this program; if not, write to the Free Software
  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */
-/* $Id: keyboard.c,v 1.9 2003/08/19 11:48:49 weiden Exp $
- *
+/*
  * COPYRIGHT:        See COPYING in the top level directory
  * PROJECT:          ReactOS kernel
  * PURPOSE:          Messages
 
 /* INCLUDES ******************************************************************/
 
-#include <ddk/ntddk.h>
-#include <win32k/win32k.h>
-#include <internal/safe.h>
-#include <internal/kbd.h>
-#include <include/guicheck.h>
-#include <include/msgqueue.h>
-#include <include/window.h>
-#include <include/class.h>
-#include <include/error.h>
-#include <include/object.h>
-#include <include/winsta.h>
+#include <w32k.h>
 
 #define NDEBUG
 #include <debug.h>
 
-DWORD ModBits = 0;
-BYTE QueueKeyStateTable[256];
-static PVOID pkKeyboardLayout = 0;
+/* Directory to load key layouts from */
+#define SYSTEMROOT_DIR L"\\SystemRoot\\System32\\"
+/* Lock modifiers */
+#define CAPITAL_BIT   0x80000000
+#define NUMLOCK_BIT   0x40000000
+#define MOD_BITS_MASK 0x3fffffff
+#define MOD_KCTRL     0x02
+/* Key States */
+#define KS_DOWN_MASK     0xc0
+#define KS_DOWN_BIT      0x80
+#define KS_LOCK_BIT    0x01
+/* lParam bits */
+#define LP_EXT_BIT       (1<<24)
+/* From kbdxx.c -- Key changes with numlock */
+#define KNUMP         0x400 
 
-/* arty -- These should be phased out for the general kbdxx.dll tables */
+/* Lock the keyboard state to prevent unusual concurrent access */ 
+FAST_MUTEX QueueStateLock;
 
-struct accent_char
-{
-    BYTE ac_accent;
-    BYTE ac_char;
-    BYTE ac_result;
-};
+BYTE QueueKeyStateTable[256];
 
-static const struct accent_char accent_chars[] =
-{
-/* A good idea should be to read /usr/X11/lib/X11/locale/iso8859-x/Compose */
-    {'`', 'A', '\300'},  {'`', 'a', '\340'},
-    {'\'', 'A', '\301'}, {'\'', 'a', '\341'},
-    {'^', 'A', '\302'},  {'^', 'a', '\342'},
-    {'~', 'A', '\303'},  {'~', 'a', '\343'},
-    {'"', 'A', '\304'},  {'"', 'a', '\344'},
-    {'O', 'A', '\305'},  {'o', 'a', '\345'},
-    {'0', 'A', '\305'},  {'0', 'a', '\345'},
-    {'A', 'A', '\305'},  {'a', 'a', '\345'},
-    {'A', 'E', '\306'},  {'a', 'e', '\346'},
-    {',', 'C', '\307'},  {',', 'c', '\347'},
-    {'`', 'E', '\310'},  {'`', 'e', '\350'},
-    {'\'', 'E', '\311'}, {'\'', 'e', '\351'},
-    {'^', 'E', '\312'},  {'^', 'e', '\352'},
-    {'"', 'E', '\313'},  {'"', 'e', '\353'},
-    {'`', 'I', '\314'},  {'`', 'i', '\354'},
-    {'\'', 'I', '\315'}, {'\'', 'i', '\355'},
-    {'^', 'I', '\316'},  {'^', 'i', '\356'},
-    {'"', 'I', '\317'},  {'"', 'i', '\357'},
-    {'-', 'D', '\320'},  {'-', 'd', '\360'},
-    {'~', 'N', '\321'},  {'~', 'n', '\361'},
-    {'`', 'O', '\322'},  {'`', 'o', '\362'},
-    {'\'', 'O', '\323'}, {'\'', 'o', '\363'},
-    {'^', 'O', '\324'},  {'^', 'o', '\364'},
-    {'~', 'O', '\325'},  {'~', 'o', '\365'},
-    {'"', 'O', '\326'},  {'"', 'o', '\366'},
-    {'/', 'O', '\330'},  {'/', 'o', '\370'},
-    {'`', 'U', '\331'},  {'`', 'u', '\371'},
-    {'\'', 'U', '\332'}, {'\'', 'u', '\372'},
-    {'^', 'U', '\333'},  {'^', 'u', '\373'},
-    {'"', 'U', '\334'},  {'"', 'u', '\374'},
-    {'\'', 'Y', '\335'}, {'\'', 'y', '\375'},
-    {'T', 'H', '\336'},  {'t', 'h', '\376'},
-    {'s', 's', '\337'},  {'"', 'y', '\377'},
-    {'s', 'z', '\337'},  {'i', 'j', '\377'},
-       /* iso-8859-2 uses this */
-    {'<', 'L', '\245'},  {'<', 'l', '\265'},   /* caron */
-    {'<', 'S', '\251'},  {'<', 's', '\271'},
-    {'<', 'T', '\253'},  {'<', 't', '\273'},
-    {'<', 'Z', '\256'},  {'<', 'z', '\276'},
-    {'<', 'C', '\310'},  {'<', 'c', '\350'},
-    {'<', 'E', '\314'},  {'<', 'e', '\354'},
-    {'<', 'D', '\317'},  {'<', 'd', '\357'},
-    {'<', 'N', '\322'},  {'<', 'n', '\362'},
-    {'<', 'R', '\330'},  {'<', 'r', '\370'},
-    {';', 'A', '\241'},  {';', 'a', '\261'},   /* ogonek */
-    {';', 'E', '\312'},  {';', 'e', '\332'},
-    {'\'', 'Z', '\254'}, {'\'', 'z', '\274'},  /* acute */
-    {'\'', 'R', '\300'}, {'\'', 'r', '\340'},
-    {'\'', 'L', '\305'}, {'\'', 'l', '\345'},
-    {'\'', 'C', '\306'}, {'\'', 'c', '\346'},
-    {'\'', 'N', '\321'}, {'\'', 'n', '\361'},
-/*  collision whith S, from iso-8859-9 !!! */
-    {',', 'S', '\252'},  {',', 's', '\272'},   /* cedilla */
-    {',', 'T', '\336'},  {',', 't', '\376'},
-    {'.', 'Z', '\257'},  {'.', 'z', '\277'},   /* dot above */
-    {'/', 'L', '\243'},  {'/', 'l', '\263'},   /* slash */
-    {'/', 'D', '\320'},  {'/', 'd', '\360'},
-    {'(', 'A', '\303'},  {'(', 'a', '\343'},   /* breve */
-    {'\275', 'O', '\325'}, {'\275', 'o', '\365'},      /* double acute */
-    {'\275', 'U', '\334'}, {'\275', 'u', '\374'},
-    {'0', 'U', '\332'},  {'0', 'u', '\372'},   /* ring above */
-       /* iso-8859-3 uses this */
-    {'/', 'H', '\241'},  {'/', 'h', '\261'},   /* slash */
-    {'>', 'H', '\246'},  {'>', 'h', '\266'},   /* circumflex */
-    {'>', 'J', '\254'},  {'>', 'j', '\274'},
-    {'>', 'C', '\306'},  {'>', 'c', '\346'},
-    {'>', 'G', '\330'},  {'>', 'g', '\370'},
-    {'>', 'S', '\336'},  {'>', 's', '\376'},
-/*  collision whith G( from iso-8859-9 !!!   */
-    {'(', 'G', '\253'},  {'(', 'g', '\273'},   /* breve */
-    {'(', 'U', '\335'},  {'(', 'u', '\375'},
-/*  collision whith I. from iso-8859-3 !!!   */
-    {'.', 'I', '\251'},  {'.', 'i', '\271'},   /* dot above */
-    {'.', 'C', '\305'},  {'.', 'c', '\345'},
-    {'.', 'G', '\325'},  {'.', 'g', '\365'},
-       /* iso-8859-4 uses this */
-    {',', 'R', '\243'},  {',', 'r', '\263'},   /* cedilla */
-    {',', 'L', '\246'},  {',', 'l', '\266'},
-    {',', 'G', '\253'},  {',', 'g', '\273'},
-    {',', 'N', '\321'},  {',', 'n', '\361'},
-    {',', 'K', '\323'},  {',', 'k', '\363'},
-    {'~', 'I', '\245'},  {'~', 'i', '\265'},   /* tilde */
-    {'-', 'E', '\252'},  {'-', 'e', '\272'},   /* macron */
-    {'-', 'A', '\300'},  {'-', 'a', '\340'},
-    {'-', 'I', '\317'},  {'-', 'i', '\357'},
-    {'-', 'O', '\322'},  {'-', 'o', '\362'},
-    {'-', 'U', '\336'},  {'-', 'u', '\376'},
-    {'/', 'T', '\254'},  {'/', 't', '\274'},   /* slash */
-    {'.', 'E', '\314'},  {'.', 'e', '\344'},   /* dot above */
-    {';', 'I', '\307'},  {';', 'i', '\347'},   /* ogonek */
-    {';', 'U', '\331'},  {';', 'u', '\371'},
-       /* iso-8859-9 uses this */
-       /* iso-8859-9 has really bad choosen G( S, and I. as they collide
-        * whith the same letters on other iso-8859-x (that is they are on
-        * different places :-( ), if you use turkish uncomment these and
-        * comment out the lines in iso-8859-2 and iso-8859-3 sections
-        * FIXME: should be dynamic according to chosen language
-        *        if/when Wine has turkish support.  
-        */ 
-/*  collision whith G( from iso-8859-3 !!!   */
-/*  {'(', 'G', '\320'},  {'(', 'g', '\360'}, */        /* breve */
-/*  collision whith S, from iso-8859-2 !!! */
-/*  {',', 'S', '\336'},  {',', 's', '\376'}, */        /* cedilla */
-/*  collision whith I. from iso-8859-3 !!!   */
-/*  {'.', 'I', '\335'},  {'.', 'i', '\375'}, */        /* dot above */
-};
+#define IntLockQueueState \
+  ExAcquireFastMutex(&QueueStateLock)
+
+#define IntUnLockQueueState \
+  ExReleaseFastMutex(&QueueStateLock)
 
 /* FUNCTIONS *****************************************************************/
 
+/* Initialization -- Right now, just zero the key state and init the lock */
+NTSTATUS FASTCALL InitKeyboardImpl(VOID) {
+  ExInitializeFastMutex(&QueueStateLock);
+  RtlZeroMemory(&QueueKeyStateTable,0x100);
+  return STATUS_SUCCESS;
+}
+
 /*** Statics used by TranslateMessage ***/
 
-static VOID STDCALL SetKeyState(DWORD key, BOOL down) {
-  if( key >= 'a' && key <= 'z' ) key += 'A' - 'a';
-  QueueKeyStateTable[key] = down;
+/*** Shift state code was out of hand, sorry. --- arty */
+
+static UINT DontDistinguishShifts( UINT ret ) {
+    if( ret == VK_LSHIFT || ret == VK_RSHIFT ) ret = VK_LSHIFT;
+    if( ret == VK_LCONTROL || ret == VK_RCONTROL ) ret = VK_LCONTROL;
+    if( ret == VK_LMENU || ret == VK_RMENU ) ret = VK_LMENU;
+    return ret;
+}
+
+static VOID STDCALL SetKeyState(DWORD key, DWORD vk, DWORD ext, BOOL down) {
+  ASSERT(vk <= 0xff);
+
+  /* Special handling for toggles like numpad and caps lock */
+  if (vk == VK_CAPITAL || vk == VK_NUMLOCK) {
+    if (down) QueueKeyStateTable[vk] ^= KS_LOCK_BIT;
+  } 
+
+  if (ext && vk == VK_LSHIFT)
+    vk = VK_RSHIFT;
+  if (ext && vk == VK_LCONTROL)
+    vk = VK_RCONTROL;
+  if (ext && vk == VK_LMENU)
+    vk = VK_RMENU;
+
+  if (down)
+    QueueKeyStateTable[vk] |= KS_DOWN_BIT;
+  else
+    QueueKeyStateTable[vk] &= ~KS_DOWN_MASK;
+
+  if (vk == VK_LSHIFT || vk == VK_RSHIFT) {
+    if ((QueueKeyStateTable[VK_LSHIFT] & KS_DOWN_BIT) ||
+        (QueueKeyStateTable[VK_RSHIFT] & KS_DOWN_BIT)) {
+      QueueKeyStateTable[VK_SHIFT] |= KS_DOWN_BIT;
+    } else {
+      QueueKeyStateTable[VK_SHIFT] &= ~KS_DOWN_MASK;
+    }
+  }
+
+  if (vk == VK_LCONTROL || vk == VK_RCONTROL) {
+    if ((QueueKeyStateTable[VK_LCONTROL] & KS_DOWN_BIT) ||
+        (QueueKeyStateTable[VK_RCONTROL] & KS_DOWN_BIT)) {
+      QueueKeyStateTable[VK_CONTROL] |= KS_DOWN_BIT;
+    } else {
+      QueueKeyStateTable[VK_CONTROL] &= ~KS_DOWN_MASK;
+    }
+  }
+
+  if (vk == VK_LMENU || vk == VK_RMENU) {
+    if ((QueueKeyStateTable[VK_LMENU] & KS_DOWN_BIT) ||
+        (QueueKeyStateTable[VK_RMENU] & KS_DOWN_BIT)) {
+      QueueKeyStateTable[VK_MENU] |= KS_DOWN_BIT;
+    } else {
+      QueueKeyStateTable[VK_MENU] &= ~KS_DOWN_MASK;
+    }
+  }
+}
+
+VOID DumpKeyState( PBYTE KeyState ) {
+  int i;
+
+  DbgPrint( "KeyState { " );
+  for( i = 0; i < 0x100; i++ ) {
+    if( KeyState[i] ) DbgPrint( "%02x(%02x) ", i, KeyState[i] );
+  }
+  DbgPrint( "};\n" );
+}
+
+static BYTE KeysSet( PKBDTABLES pkKT, PBYTE KeyState, 
+                    int FakeModLeft, int FakeModRight ) {
+  if( !KeyState || !pkKT ) return 0;
+
+  /* Search special codes first */
+  if( FakeModLeft && KeyState[FakeModLeft] )
+      return KeyState[FakeModLeft];
+  else if( FakeModRight && KeyState[FakeModRight] )
+      return KeyState[FakeModRight];
+
+  return 0;
 }
 
-static BOOL SetModKey( PKBDTABLES pkKT, WORD wVK, BOOL down ) {
+/* Search the keyboard layout modifiers table for the shift bit.  I don't
+ * want to count on the shift bit not moving, because it can be specified
+ * in the layout */
+
+static DWORD FASTCALL GetShiftBit( PKBDTABLES pkKT, DWORD Vk ) {
   int i;
+
+  for( i = 0; pkKT->pCharModifiers->pVkToBit[i].Vk; i++ ) 
+    if( pkKT->pCharModifiers->pVkToBit[i].Vk == Vk ) 
+      return pkKT->pCharModifiers->pVkToBit[i].ModBits;
+
+  return 0;
+}
+
+static DWORD ModBits( PKBDTABLES pkKT, PBYTE KeyState ) {
+  DWORD ModBits = 0;
+
+  if( !KeyState ) return 0;
+
+  /* DumpKeyState( KeyState ); */
+
+  if (KeysSet( pkKT, KeyState, VK_LSHIFT, VK_RSHIFT ) & 
+      KS_DOWN_BIT)
+      ModBits |= GetShiftBit( pkKT, VK_SHIFT );
   
-  for( i = 0; pkKT->pCharModifiers->pVkToBit[i].Vk; i++ ) {
-    DbgPrint( "vk[%d] = { %04x, %x }\n", i, 
-       pkKT->pCharModifiers->pVkToBit[i].Vk,
-       pkKT->pCharModifiers->pVkToBit[i].ModBits );
-    if( pkKT->pCharModifiers->pVkToBit[i].Vk == wVK ) {
-      if( down ) ModBits |= pkKT->pCharModifiers->pVkToBit[i].ModBits;
-      else ModBits &= ~pkKT->pCharModifiers->pVkToBit[i].ModBits;
-      DbgPrint( "ModBits: %x\n", ModBits );
-      return TRUE;
+  if (KeysSet( pkKT, KeyState, VK_LCONTROL, VK_RCONTROL ) & 
+      KS_DOWN_BIT )
+      ModBits |= GetShiftBit( pkKT, VK_CONTROL );
+
+  if (KeysSet( pkKT, KeyState, VK_LMENU, VK_RMENU ) &
+      KS_DOWN_BIT )
+      ModBits |= GetShiftBit( pkKT, VK_MENU );
+
+  /* Handle Alt+Gr */
+  if (KeysSet( pkKT, KeyState, VK_RMENU, 0 ) &
+      KS_DOWN_BIT )
+      ModBits |= GetShiftBit( pkKT, VK_CONTROL );
+      
+      /* Deal with VK_CAPITAL */
+  if (KeysSet( pkKT, KeyState, VK_CAPITAL, 0 ) & KS_LOCK_BIT) 
+    {
+      ModBits |= CAPITAL_BIT;
     }
-  }
 
-  return FALSE;
+  /* Deal with VK_NUMLOCK */
+  if (KeysSet( pkKT, KeyState, VK_NUMLOCK, 0 ) & KS_LOCK_BIT) 
+    {
+      ModBits |= NUMLOCK_BIT;
+    }
+
+  DPRINT( "Current Mod Bits: %x\n", ModBits );
+
+  return ModBits;
 }
 
-static BOOL TryToTranslateChar( WORD wVirtKey,
-                               PVK_TO_WCHAR_TABLE vtwTbl, 
-                               DWORD ModBits,
-                               PBOOL pbDead,
-                               PBOOL pbLigature,
-                               PWCHAR pwcTranslatedChar ) {
-  int i,j;
-  size_t size_this_entry = vtwTbl->cbSize;
-  int nStates = vtwTbl->nModifications;
+static BOOL TryToTranslateChar(WORD wVirtKey,
+                              DWORD ModBits,
+                              PBOOL pbDead,
+                              PBOOL pbLigature,
+                              PWCHAR pwcTranslatedChar,
+                              PKBDTABLES keyLayout ) 
+{
+  PVK_TO_WCHAR_TABLE vtwTbl;
   PVK_TO_WCHARS10 vkPtr;
+  size_t size_this_entry;
+  int nMod, shift;
+  DWORD CapsMod = 0, CapsState = 0;
+
+  CapsState = ModBits & ~MOD_BITS_MASK;
+  ModBits = ModBits & MOD_BITS_MASK;
+
+  DPRINT ( "TryToTranslate: %04x %x\n", wVirtKey, ModBits ); 
+
+  if (ModBits > keyLayout->pCharModifiers->wMaxModBits)
+    {
+      return FALSE;
+    }
+  shift = keyLayout->pCharModifiers->ModNumber[ModBits];
+  
+  for (nMod = 0; keyLayout->pVkToWcharTable[nMod].nModifications; nMod++)
+    {
+      vtwTbl = &keyLayout->pVkToWcharTable[nMod];
+      size_this_entry = vtwTbl->cbSize;
+      vkPtr = (PVK_TO_WCHARS10)((BYTE *)vtwTbl->pVkToWchars);
+      while(vkPtr->VirtualKey)
+        {
+          if( wVirtKey == (vkPtr->VirtualKey & 0xff) )
+           {
+             CapsMod = 
+               shift | ((CapsState & CAPITAL_BIT) ? vkPtr->Attributes : 0);
+
+             if( CapsMod > keyLayout->pVkToWcharTable[nMod].nModifications ) {
+                 DWORD MaxBit = 1;
+                 while( MaxBit < 
+                        keyLayout->pVkToWcharTable[nMod].nModifications )
+                     MaxBit <<= 1;
 
-  for( i = 0;; i++ ) {
-    vkPtr = (PVK_TO_WCHARS10)
-      (((BYTE *)vtwTbl->pVkToWchars) + i * size_this_entry);
-
-    if( !vkPtr->VirtualKey ) return FALSE;
-    if( wVirtKey == vkPtr->VirtualKey ) {
-      for( j = 0; j < nStates; j++ ) {
-       if( j == (int) ModBits ) { /* OK, we found a wchar with the correct
-                               shift state and vk */
-         *pbDead = vkPtr->wch[j] == WCH_DEAD;
-         *pbLigature = vkPtr->wch[j] == WCH_LGTR;
-         *pwcTranslatedChar = vkPtr->wch[j];
-         if( *pbDead ) {
-           i++;
-           vkPtr = (PVK_TO_WCHARS10)
-             (((BYTE *)vtwTbl->pVkToWchars) + i * size_this_entry);
-           if( vkPtr->VirtualKey != 0xff ) {
-             DPRINT( "Found dead key with no trailer in the table.\n" );
-             DPRINT( "VK: %04x, ADDR: %08x\n", wVirtKey, (int)vkPtr );
-             return FALSE;
+                 CapsMod &= MaxBit - 1; /* Guarantee that CapsMod lies
+                                           in bounds. */
+             }
+             
+             *pbDead = vkPtr->wch[CapsMod] == WCH_DEAD;
+             *pbLigature = vkPtr->wch[CapsMod] == WCH_LGTR;
+             *pwcTranslatedChar = vkPtr->wch[CapsMod];
+             
+             DPRINT("%d %04x: CapsMod %08x CapsState %08x shift %08x Char %04x\n",
+                      nMod, wVirtKey,
+                      CapsMod, CapsState, shift, *pwcTranslatedChar);
+
+             if( *pbDead ) 
+               {
+                  vkPtr = (PVK_TO_WCHARS10)(((BYTE *)vkPtr) + size_this_entry);
+                 if( vkPtr->VirtualKey != 0xff ) 
+                   {
+                     DPRINT( "Found dead key with no trailer in the table.\n" );
+                     DPRINT( "VK: %04x, ADDR: %08x\n", wVirtKey, (int)vkPtr );
+                     return FALSE;
+                   }
+                 *pwcTranslatedChar = vkPtr->wch[shift];
+               }
+               return TRUE;
            }
-           *pwcTranslatedChar = vkPtr->wch[j];
-         }
-         return TRUE;
+          vkPtr = (PVK_TO_WCHARS10)(((BYTE *)vkPtr) + size_this_entry);
        }
-      }
     }
-  }
+  return FALSE;
 }
 
 static
@@ -243,38 +287,31 @@ ToUnicodeInner(UINT wVirtKey,
               LPWSTR pwszBuff,
               int cchBuff,
               UINT wFlags,
-              DWORD ModBits,
               PKBDTABLES pkKT)
 {
-  int i;
+  WCHAR wcTranslatedChar;
+  BOOL bDead;
+  BOOL bLigature;
 
-  DbgPrint("wVirtKey=%08x, wScanCode=%08x, lpKeyState=[], "
-          "pwszBuff=%S, cchBuff=%d, wFlags=%x\n",
-          wVirtKey, wScanCode, /* lpKeyState, */ pwszBuff,
-          cchBuff, wFlags );
-
-  for( i = 0; pkKT->pVkToWcharTable[i].nModifications; i++ ) {
-    WCHAR wcTranslatedChar;
-    BOOL bDead;
-    BOOL bLigature;
-
-    if( TryToTranslateChar( wVirtKey,
-                           &pkKT->pVkToWcharTable[i], 
-                           ModBits,
-                           &bDead,
-                           &bLigature,
-                           &wcTranslatedChar ) ) {
-      if( bLigature ) {
-       DPRINT("Not handling ligature (yet)\n" );
-       return 0;
-      }
+  if( !pkKT ) return 0;
+
+  if( TryToTranslateChar( wVirtKey,
+                         ModBits( pkKT, lpKeyState ),
+                         &bDead,
+                         &bLigature,
+                         &wcTranslatedChar, 
+                         pkKT ) ) 
+    {
+      if( bLigature ) 
+        {
+         DPRINT("Not handling ligature (yet)\n" );
+         return 0;
+        }
 
       if( cchBuff > 0 ) pwszBuff[0] = wcTranslatedChar;
 
-      if( bDead ) return -1;
-      else return 1;
+      return bDead ? -1 : 1;
     }
-  }
 
   return 0;
 }
@@ -284,13 +321,45 @@ STDCALL
 NtUserGetKeyState(
   DWORD key)
 {
-  DWORD ret;
+  DWORD ret = 0;
 
-    if (key >= 'a' && key <= 'z') key += 'A' - 'a';
-    ret = ((DWORD)(QueueKeyStateTable[key] & 0x80) << 8 ) |
-              (QueueKeyStateTable[key] & 0x80) |
-              (QueueKeyStateTable[key] & 0x01);
-    return ret;
+  IntLockQueueState;
+  if( key < 0x100 ) {
+    ret = ((DWORD)(QueueKeyStateTable[key] & KS_DOWN_BIT) << 8 ) |
+      (QueueKeyStateTable[key] & KS_LOCK_BIT);
+  }
+  IntUnLockQueueState;
+  return ret;
+}
+
+int STDCALL ToUnicodeEx( UINT wVirtKey,
+                        UINT wScanCode,
+                        PBYTE lpKeyState,
+                        LPWSTR pwszBuff,
+                        int cchBuff,
+                        UINT wFlags,
+                        HKL dwhkl ) {
+  int ToUnicodeResult = 0;
+
+  if (0 == (lpKeyState[wVirtKey] & KS_DOWN_BIT))
+    {
+      ToUnicodeResult = 0;
+    }
+  else
+    {
+      IntLockQueueState;
+      ToUnicodeResult = ToUnicodeInner( wVirtKey,
+                                       wScanCode,
+                                       lpKeyState,
+                                       pwszBuff,
+                                       cchBuff,
+                                       wFlags,
+                                       PsGetWin32Thread() ? 
+                                       PsGetWin32Thread()->KeyboardLayout : 0 );
+      IntUnLockQueueState;
+    }
+
+  return ToUnicodeResult;
 }
 
 int STDCALL ToUnicode( UINT wVirtKey,
@@ -299,14 +368,122 @@ int STDCALL ToUnicode( UINT wVirtKey,
                       LPWSTR pwszBuff,
                       int cchBuff,
                       UINT wFlags ) {
-  return ToUnicodeInner( wVirtKey,
-                        wScanCode,
-                        QueueKeyStateTable,
-                        pwszBuff,
-                        cchBuff,
-                        wFlags,
-                        ModBits,
-                        pkKeyboardLayout );
+  return ToUnicodeEx( wVirtKey,
+                     wScanCode,
+                     QueueKeyStateTable,
+                     pwszBuff,
+                     cchBuff,
+                     wFlags,
+                     0 );
+}
+
+/* 
+ * Utility to copy and append two unicode strings.
+ *
+ * IN OUT PUNICODE_STRING ResultFirst -> First string and result
+ * IN     PUNICODE_STRING Second      -> Second string to append
+ * IN     BOOL            Deallocate  -> TRUE: Deallocate First string before
+ *                                       overwriting.
+ *
+ * Returns NTSTATUS.
+ */
+
+NTSTATUS NTAPI AppendUnicodeString(PUNICODE_STRING ResultFirst,
+                                  PUNICODE_STRING Second,
+                                  BOOL Deallocate) {
+    NTSTATUS Status;
+    PWSTR new_string = 
+       ExAllocatePoolWithTag(PagedPool,
+                             (ResultFirst->Length + Second->Length + sizeof(WCHAR)),
+                             TAG_STRING);
+    if( !new_string ) {
+       return STATUS_NO_MEMORY;
+    }
+    memcpy( new_string, ResultFirst->Buffer, 
+           ResultFirst->Length );
+    memcpy( new_string + ResultFirst->Length / sizeof(WCHAR),
+           Second->Buffer,
+           Second->Length );
+    if( Deallocate ) RtlFreeUnicodeString(ResultFirst);
+    ResultFirst->Length += Second->Length;
+    ResultFirst->MaximumLength = ResultFirst->Length;
+    new_string[ResultFirst->Length / sizeof(WCHAR)] = 0;
+    Status = RtlCreateUnicodeString(ResultFirst,new_string) ? 
+       STATUS_SUCCESS : STATUS_NO_MEMORY;
+    ExFreePool(new_string);
+    return Status;
+}
+
+/*
+ * Utility function to read a value from the registry more easily.
+ *
+ * IN  PUNICODE_STRING KeyName       -> Name of key to open
+ * IN  PUNICODE_STRING ValueName     -> Name of value to open
+ * OUT PUNICODE_STRING ReturnedValue -> String contained in registry
+ *
+ * Returns NTSTATUS
+ */
+
+static NTSTATUS NTAPI ReadRegistryValue( PUNICODE_STRING KeyName,
+                                        PUNICODE_STRING ValueName,
+                                        PUNICODE_STRING ReturnedValue ) {
+    NTSTATUS Status;
+    HANDLE KeyHandle;
+    OBJECT_ATTRIBUTES KeyAttributes;
+    PKEY_VALUE_PARTIAL_INFORMATION KeyValuePartialInfo;
+    ULONG Length = 0;
+    ULONG ResLength = 0;
+    UNICODE_STRING Temp;
+    
+    InitializeObjectAttributes(&KeyAttributes, KeyName, OBJ_CASE_INSENSITIVE,
+                              NULL, NULL);
+    Status = ZwOpenKey(&KeyHandle, KEY_ALL_ACCESS, &KeyAttributes);
+    if( !NT_SUCCESS(Status) ) {
+       return Status;
+    }
+    
+    Status = ZwQueryValueKey(KeyHandle, ValueName, KeyValuePartialInformation,
+                            0,
+                            0,
+                            &ResLength);
+    
+    if( Status != STATUS_BUFFER_TOO_SMALL ) {
+       NtClose(KeyHandle);
+       return Status;
+    }
+    
+    ResLength += sizeof( *KeyValuePartialInfo );
+    KeyValuePartialInfo = 
+       ExAllocatePoolWithTag(PagedPool, ResLength, TAG_STRING);
+    Length = ResLength;
+    
+    if( !KeyValuePartialInfo ) {
+       NtClose(KeyHandle);
+       return STATUS_NO_MEMORY;
+    }
+    
+    Status = ZwQueryValueKey(KeyHandle, ValueName, KeyValuePartialInformation,
+                            (PVOID)KeyValuePartialInfo,
+                            Length,
+                            &ResLength);
+    
+    if( !NT_SUCCESS(Status) ) {
+       NtClose(KeyHandle);
+       ExFreePool(KeyValuePartialInfo);
+       return Status;
+    }
+    
+    Temp.Length = Temp.MaximumLength = KeyValuePartialInfo->DataLength;
+    Temp.Buffer = (PWCHAR)KeyValuePartialInfo->Data;
+    
+    /* At this point, KeyValuePartialInfo->Data contains the key data */
+    RtlInitUnicodeString(ReturnedValue,L"");
+    AppendUnicodeString(ReturnedValue,&Temp,FALSE);
+    
+    ExFreePool(KeyValuePartialInfo);
+    NtClose(KeyHandle);
+    
+    return Status;
 }
 
 typedef PVOID (*KbdLayerDescriptor)(VOID);
@@ -315,110 +492,189 @@ NTSTATUS STDCALL LdrGetProcedureAddress(PVOID module,
                                        DWORD flags,
                                        PVOID *func_addr);
 
-void InitKbdLayout( PVOID *pkKeyboardLayout ) {
+void InitKbdLayout( PVOID *pkKeyboardLayout )
+{
+  WCHAR LocaleBuffer[16];
+  UNICODE_STRING LayoutKeyName;
+  UNICODE_STRING LayoutValueName;
+  UNICODE_STRING DefaultLocale;
+  UNICODE_STRING LayoutFile;
+  UNICODE_STRING FullLayoutPath;
+  LCID LocaleId;
+  PWCHAR KeyboardLayoutWSTR;
   HMODULE kbModule = 0;
+  NTSTATUS Status;
   ANSI_STRING kbdProcedureName;
-  //NTSTATUS Status;
-
   KbdLayerDescriptor layerDescGetFn;
 
-  kbModule = EngLoadImage(L"\\SystemRoot\\system32\\kbdus.dll");
+  #define XX_STATUS(x) if (!NT_SUCCESS(Status = (x))) continue;
 
-  if( !kbModule ) {
-    DbgPrint( "Foo: No kbdus.dll\n" );
-    return;
-  }
+  do {
+    Status = ZwQueryDefaultLocale(FALSE, &LocaleId);
+    if (!NT_SUCCESS(Status))
+    {
+      DPRINT1("Could not get default locale (%08lx).\n", Status);
+    }
+    else
+    {
+      DPRINT("DefaultLocale = %lx\n", LocaleId);
+      swprintf(LocaleBuffer, L"%08lx", LocaleId);
+      DPRINT("DefaultLocale = %S\n", LocaleBuffer);
+      RtlInitUnicodeString(&DefaultLocale, LocaleBuffer);
 
-  RtlInitAnsiString( &kbdProcedureName, "KbdLayerDescriptor" );
-  LdrGetProcedureAddress((PVOID)kbModule,
-                        &kbdProcedureName,
-                        0,
-                        (PVOID*)&layerDescGetFn);
-  if( layerDescGetFn ) {
-    *pkKeyboardLayout = layerDescGetFn();
+      RtlInitUnicodeString(&LayoutKeyName,
+                          L"\\REGISTRY\\Machine\\SYSTEM\\CurrentControlSet"
+                          L"\\Control\\KeyboardLayouts\\");
+
+      AppendUnicodeString(&LayoutKeyName,&DefaultLocale,FALSE);
+
+      RtlInitUnicodeString(&LayoutValueName,L"Layout File");
+
+      Status = ReadRegistryValue(&LayoutKeyName,&LayoutValueName,&LayoutFile);
+      RtlInitUnicodeString(&FullLayoutPath,SYSTEMROOT_DIR);
+
+      if( !NT_SUCCESS(Status) ) {
+       DPRINT1("Got default locale but not layout file. (%08x)\n",
+                Status);
+       RtlFreeUnicodeString(&LayoutFile);
+      } else {
+       DPRINT("Read registry and got %wZ\n", &LayoutFile);
+    
+       RtlFreeUnicodeString(&LayoutKeyName);
+
+       AppendUnicodeString(&FullLayoutPath,&LayoutFile,FALSE);
+
+       DPRINT("Loading Keyboard DLL %wZ\n", &FullLayoutPath);
+
+       RtlFreeUnicodeString(&LayoutFile);
+
+       KeyboardLayoutWSTR = 
+         ExAllocatePoolWithTag(PagedPool,
+                               FullLayoutPath.Length + sizeof(WCHAR), 
+                               TAG_STRING);
+
+       if( !KeyboardLayoutWSTR ) {
+         DPRINT1("Couldn't allocate a string for the keyboard layout name.\n");
+         RtlFreeUnicodeString(&FullLayoutPath);
+         return;
+       }
+       memcpy(KeyboardLayoutWSTR,FullLayoutPath.Buffer,
+              FullLayoutPath.Length + sizeof(WCHAR));
+       KeyboardLayoutWSTR[FullLayoutPath.Length / sizeof(WCHAR)] = 0;
+
+       kbModule = EngLoadImage(KeyboardLayoutWSTR);
+       DPRINT( "Load Keyboard Layout: %S\n", KeyboardLayoutWSTR );
+
+        if( !kbModule )
+         DPRINT1( "Load Keyboard Layout: No %wZ\n", &FullLayoutPath );
+      }
+
+      RtlFreeUnicodeString(&FullLayoutPath);
+    }
+
+    if( !kbModule )
+    {
+      DPRINT1("Trying to load US Keyboard Layout\n");
+      kbModule = EngLoadImage(L"\\SystemRoot\\system32\\kbdus.dll");
+      
+      if (!kbModule)
+      {
+        DPRINT1("Failed to load any Keyboard Layout\n");
+        return;
+           }
+    }
+
+    RtlInitAnsiString( &kbdProcedureName, "KbdLayerDescriptor" );
+
+    LdrGetProcedureAddress((PVOID)kbModule,
+                          &kbdProcedureName,
+                          0,
+                          (PVOID*)&layerDescGetFn);
+    
+    if( layerDescGetFn ) {
+      *pkKeyboardLayout = layerDescGetFn();
+    }
+  } while (FALSE);
+
+  if( !*pkKeyboardLayout ) {
+    DPRINT1("Failed to load the keyboard layout.\n");
   }
+
+#undef XX_STATUS
+}
+
+PKBDTABLES W32kGetDefaultKeyLayout() {
+  PKBDTABLES pkKeyboardLayout = 0;
+  InitKbdLayout( (PVOID) &pkKeyboardLayout );
+  return pkKeyboardLayout;
 }
 
-BOOL STDCALL
-NtUserTranslateMessage(LPMSG lpMsg,
-                      DWORD Unknown1) /* Used to pass the kbd layout */
+BOOL FASTCALL
+IntTranslateKbdMessage(LPMSG lpMsg,
+                       HKL dwhkl)
 {
   static INT dead_char = 0;
   LONG UState = 0;
   WCHAR wp[2] = { 0 };
   MSG NewMsg = { 0 };
-  PUSER_MESSAGE UMsg;
-
-  /* FIXME: Should pass current keyboard layout for this thread. */
-  /* At the moment, the keyboard layout is global. */
-  /* Also, we're fixed at kbdus.dll ... */
-  if( !pkKeyboardLayout ) InitKbdLayout( &pkKeyboardLayout );
-  if( !pkKeyboardLayout ) {
-    DbgPrint( "Not Translating due to empty layout.\n" );
+  PKBDTABLES keyLayout;
+  BOOL Result = FALSE;
+  DWORD ScanCode = 0;
+
+
+  keyLayout = PsGetWin32Thread()->KeyboardLayout;
+  if( !keyLayout )
     return FALSE;
-  }
 
   if (lpMsg->message != WM_KEYDOWN && lpMsg->message != WM_SYSKEYDOWN)
-    {
-      if (lpMsg->message == WM_KEYUP) {
-       DbgPrint( "About to SetKeyState( %04x, FALSE );\n", lpMsg->wParam );
-       SetKeyState( lpMsg->wParam, FALSE ); /* Release key */
-        DbgPrint( "About to SetModKey();\n" );
-       SetModKey( pkKeyboardLayout, lpMsg->wParam, FALSE );
-       /* Release Mod if any */
-       DbgPrint( "Done with keys.\n" );
-      }
-      return(FALSE);
-    }
+    return FALSE;
 
-  DbgPrint( "About to SetKeyState( %04x, TRUE );\n", lpMsg->wParam );
-  SetKeyState( lpMsg->wParam, TRUE ); /* Strike key */
+  ScanCode = (lpMsg->lParam >> 16) & 0xff;
 
-  /* Pass 1: Search for modifiers */
-  DbgPrint( "About to SetModKey();\n" );
-  if( SetModKey( pkKeyboardLayout, lpMsg->wParam, TRUE ) ) return TRUE;
-  DbgPrint( "Done with keys.\n" );
+  IntLockQueueState;
 
-  /* Pass 2: Get Unicode Character */
-  DbgPrint( "Calling ToUnicodeString()\n" );
-  UState = ToUnicodeInner(lpMsg->wParam, HIWORD(lpMsg->lParam),
-                         QueueKeyStateTable, wp, 2, 0, ModBits,
-                         pkKeyboardLayout);
+  UState = ToUnicodeInner(lpMsg->wParam, HIWORD(lpMsg->lParam) & 0xff,
+                         QueueKeyStateTable, wp, 2, 0, 
+                         keyLayout );
 
-  DbgPrint( "UState is %d after key %04x\n", UState, wp[0] );
   if (UState == 1)
     {
       NewMsg.message = (lpMsg->message == WM_KEYDOWN) ? WM_CHAR : WM_SYSCHAR;
       if (dead_char)
         {
          ULONG i;
-         
-         if (wp[0] == ' ') wp[0] =  dead_char;
-         if (dead_char == 0xa2) dead_char = '(';
-         else if (dead_char == 0xa8) dead_char = '"';
-         else if (dead_char == 0xb2) dead_char = ';';
-         else if (dead_char == 0xb4) dead_char = '\'';
-         else if (dead_char == 0xb7) dead_char = '<';
-         else if (dead_char == 0xb8) dead_char = ',';
-         else if (dead_char == 0xff) dead_char = '.';
-         for (i = 0; i < sizeof(accent_chars)/sizeof(accent_chars[0]); i++)
+         WCHAR first, second;
+         DPRINT("PREVIOUS DEAD CHAR: %c\n", dead_char);
+
+         for( i = 0; keyLayout->pDeadKey[i].dwBoth; i++ )
            {
-             if ((accent_chars[i].ac_accent == dead_char) &&
-                 (accent_chars[i].ac_char == wp[0]))
-                {
-                 wp[0] = accent_chars[i].ac_result;
-                 break;
-                }
-             dead_char = 0;
+             first = keyLayout->pDeadKey[i].dwBoth >> 16;
+             second = keyLayout->pDeadKey[i].dwBoth;
+             if (first == dead_char && second == wp[0]) 
+               {
+                 wp[0] = keyLayout->pDeadKey[i].wchComposed;
+                 dead_char = 0;
+                 break;
+               }
            }
-        }
+
+         DPRINT("FINAL CHAR: %c\n", wp[0]);
+       }
+      if (dead_char) 
+       {
+         NewMsg.hwnd = lpMsg->hwnd;
+         NewMsg.wParam = dead_char;
+         NewMsg.lParam = lpMsg->lParam;
+         dead_char = 0;
+         MsqPostMessage(PsGetWin32Thread()->MessageQueue, &NewMsg, FALSE, QS_KEY);
+       }
+      
       NewMsg.hwnd = lpMsg->hwnd;
       NewMsg.wParam = wp[0];
       NewMsg.lParam = lpMsg->lParam;
-      UMsg = MsqCreateMessage(&NewMsg);
-      DbgPrint( "CHAR='%c' %04x %08x\n", wp[0], wp[0], lpMsg->lParam );
-      MsqPostMessage(PsGetWin32Thread()->MessageQueue, UMsg);
-      return(TRUE);
+      DPRINT( "CHAR='%c' %04x %08x\n", wp[0], wp[0], lpMsg->lParam );
+      MsqPostMessage(PsGetWin32Thread()->MessageQueue, &NewMsg, FALSE, QS_KEY);
+      Result = TRUE;
     }
   else if (UState == -1)
     {
@@ -428,17 +684,12 @@ NtUserTranslateMessage(LPMSG lpMsg,
       NewMsg.wParam = wp[0];
       NewMsg.lParam = lpMsg->lParam;
       dead_char = wp[0];
-      UMsg = MsqCreateMessage(&NewMsg);
-      MsqPostMessage(PsGetWin32Thread()->MessageQueue, UMsg);
-      return(TRUE);
+      MsqPostMessage(PsGetWin32Thread()->MessageQueue, &NewMsg, FALSE, QS_KEY);
+      Result = TRUE;
     }
-  return(FALSE);
-}
 
-HWND STDCALL
-NtUserSetFocus(HWND hWnd)
-{
-  return IntSetFocusWindow(hWnd);
+  IntUnLockQueueState;
+  return Result;
 }
 
 DWORD
@@ -446,11 +697,15 @@ STDCALL
 NtUserGetKeyboardState(
   LPBYTE lpKeyState)
 {
+  BOOL Result = TRUE;
+
+  IntLockQueueState;
   if (lpKeyState) {
        if(!NT_SUCCESS(MmCopyToCaller(lpKeyState, QueueKeyStateTable, 256)))
-               return FALSE;
+         Result = FALSE;
   }
-  return TRUE;
+  IntUnLockQueueState;
+  return Result;
 }
 
 DWORD
@@ -458,12 +713,320 @@ STDCALL
 NtUserSetKeyboardState(
   LPBYTE lpKeyState)
 {
-       if (lpKeyState) {
-               if(! NT_SUCCESS(MmCopyFromCaller(QueueKeyStateTable, lpKeyState, 256)))
-                       return FALSE;
-       }
-    return TRUE;
+  BOOL Result = TRUE;
+
+ IntLockQueueState;
+  if (lpKeyState) {
+    if(! NT_SUCCESS(MmCopyFromCaller(QueueKeyStateTable, lpKeyState, 256)))
+      Result = FALSE;
+  }
+  IntUnLockQueueState;
+  
+  return Result;
 }
 
+static UINT VkToScan( UINT Code, BOOL ExtCode, PKBDTABLES pkKT ) {
+  int i;
+
+  for( i = 0; i < pkKT->bMaxVSCtoVK; i++ ) {
+    if( pkKT->pusVSCtoVK[i] == Code ) { return i; }
+  }
+
+  return 0;
+}
+
+UINT ScanToVk( UINT Code, BOOL ExtKey, PKBDTABLES pkKT ) {
+  if( !pkKT ) {
+    DPRINT("ScanToVk: No layout\n");
+    return 0;
+  }
+
+  if( ExtKey ) {
+    int i;
+
+    for( i = 0; pkKT->pVSCtoVK_E0[i].Vsc; i++ ) {
+      if( pkKT->pVSCtoVK_E0[i].Vsc == Code ) 
+       return pkKT->pVSCtoVK_E0[i].Vk & 0xff;
+    }
+    for( i = 0; pkKT->pVSCtoVK_E1[i].Vsc; i++ ) {
+      if( pkKT->pVSCtoVK_E1[i].Vsc == Code ) 
+       return pkKT->pVSCtoVK_E1[i].Vk & 0xff;
+    }
+
+    return 0;
+  } else {
+    if( Code >= pkKT->bMaxVSCtoVK ) { return 0; }
+    return pkKT->pusVSCtoVK[Code] & 0xff;
+  }
+}
+
+/*
+ * Map a virtual key code, or virtual scan code, to a scan code, key code,
+ * or unshifted unicode character.
+ *
+ * Code: See Below
+ * Type:
+ * 0 -- Code is a virtual key code that is converted into a virtual scan code
+ *      that does not distinguish between left and right shift keys.
+ * 1 -- Code is a virtual scan code that is converted into a virtual key code
+ *      that does not distinguish between left and right shift keys.
+ * 2 -- Code is a virtual key code that is converted into an unshifted unicode
+ *      character.
+ * 3 -- Code is a virtual scan code that is converted into a virtual key code
+ *      that distinguishes left and right shift keys.
+ * KeyLayout: Keyboard layout handle (currently, unused)
+ *
+ * @implemented
+ */
+
+static UINT IntMapVirtualKeyEx( UINT Code, UINT Type, PKBDTABLES keyLayout ) {
+  UINT ret = 0;
+
+  switch( Type ) {
+  case 0:
+    if( Code == VK_RSHIFT ) Code = VK_LSHIFT;
+    if( Code == VK_RMENU ) Code = VK_LMENU;
+    if( Code == VK_RCONTROL ) Code = VK_LCONTROL;
+    ret = VkToScan( Code, FALSE, keyLayout );
+    break;
+
+  case 1:
+    ret = 
+      DontDistinguishShifts
+      (IntMapVirtualKeyEx( Code, 3, keyLayout ) );
+    break;
+
+  case 2: {
+    WCHAR wp[2];
+
+    ret = VkToScan( Code, FALSE, keyLayout );
+    ToUnicodeInner( Code, ret, 0, wp, 2, 0, keyLayout );
+    ret = wp[0];
+  } break;
+
+  case 3:
+      
+    ret = ScanToVk( Code, FALSE, keyLayout );
+    break;
+  }
+
+  return ret;
+}
+
+UINT
+STDCALL
+NtUserMapVirtualKeyEx( UINT Code, UINT Type, DWORD keyboardId, HKL dwhkl ) {
+  PKBDTABLES keyLayout = PsGetWin32Thread() ? 
+    PsGetWin32Thread()->KeyboardLayout : 0;
 
+  if( !keyLayout ) return 0;
+
+  return IntMapVirtualKeyEx( Code, Type, keyLayout );
+}
+
+
+int
+STDCALL
+NtUserToUnicodeEx(
+                 UINT wVirtKey,
+                 UINT wScanCode,
+                 PBYTE lpKeyState,
+                 LPWSTR pwszBuff,
+                 int cchBuff,
+                 UINT wFlags,
+                 HKL dwhkl ) {
+  BYTE KeyStateBuf[0x100];
+  PWCHAR OutPwszBuff = 0;
+  int ret = 0;
+  
+
+  if( !NT_SUCCESS(MmCopyFromCaller(KeyStateBuf,
+                                  lpKeyState,
+                                  sizeof(KeyStateBuf))) ) {
+    DPRINT1( "Couldn't copy key state from caller.\n" );
+    return 0;
+  }
+  OutPwszBuff = ExAllocatePoolWithTag(NonPagedPool,sizeof(WCHAR) * cchBuff, TAG_STRING);
+  if( !OutPwszBuff ) {
+    DPRINT1( "ExAllocatePool(%d) failed\n", sizeof(WCHAR) * cchBuff);
+    return 0;
+  }
+  RtlZeroMemory( OutPwszBuff, sizeof( WCHAR ) * cchBuff );
+
+  ret = ToUnicodeEx( wVirtKey,
+                    wScanCode,
+                    KeyStateBuf,
+                    OutPwszBuff,
+                    cchBuff,
+                    wFlags,
+                    dwhkl );  
+
+  MmCopyToCaller(pwszBuff,OutPwszBuff,sizeof(WCHAR)*cchBuff);
+  ExFreePool(OutPwszBuff);
+
+  return ret;
+}
+
+static int W32kSimpleToupper( int ch ) {
+  if( ch >= 'a' && ch <= 'z' ) ch = ch - 'a' + 'A';
+  return ch;
+}
+
+DWORD
+STDCALL
+NtUserGetKeyNameText( LONG lParam, LPWSTR lpString, int nSize ) {
+  int i;
+  DWORD ret = 0;
+  UINT CareVk = 0;
+  UINT VkCode = 0;
+  UINT ScanCode = (lParam >> 16) & 0xff;
+  BOOL ExtKey = lParam & (1<<24) ? TRUE : FALSE;
+  PKBDTABLES keyLayout = 
+    PsGetWin32Thread() ? 
+    PsGetWin32Thread()->KeyboardLayout : 0;
+
+  if( !keyLayout || nSize < 1 ) return 0;
+
+  if( lParam & (1<<25) ) {
+    CareVk = VkCode = ScanToVk( ScanCode, ExtKey, keyLayout );
+    if( VkCode == VK_LSHIFT || VkCode == VK_RSHIFT )
+      VkCode = VK_LSHIFT;
+    if( VkCode == VK_LCONTROL || VkCode == VK_RCONTROL )
+      VkCode = VK_LCONTROL;
+    if( VkCode == VK_LMENU || VkCode == VK_RMENU )
+      VkCode = VK_LMENU;
+  } else {
+    VkCode = ScanToVk( ScanCode, ExtKey, keyLayout );
+  }
+
+  VSC_LPWSTR *KeyNames = 0;
+
+  if( CareVk != VkCode ) 
+    ScanCode = VkToScan( VkCode, ExtKey, keyLayout );
+  
+  if( ExtKey ) 
+    KeyNames = keyLayout->pKeyNamesExt;
+  else 
+    KeyNames = keyLayout->pKeyNames;
+
+  for( i = 0; KeyNames[i].pwsz; i++ ) {
+    if( KeyNames[i].vsc == ScanCode ) {
+      UINT StrLen = wcslen(KeyNames[i].pwsz);
+      UINT StrMax = StrLen > (nSize - 1) ? (nSize - 1) : StrLen;
+      WCHAR null_wc = 0;
+      if( NT_SUCCESS( MmCopyToCaller( lpString, 
+                                     KeyNames[i].pwsz, 
+                                     StrMax * sizeof(WCHAR) ) ) &&
+         NT_SUCCESS( MmCopyToCaller( lpString + StrMax,
+                                     &null_wc,
+                                     sizeof( WCHAR ) ) ) ) {
+       ret = StrMax;
+       break;
+      }
+    }
+  }
+
+  if( ret == 0 ) {
+    WCHAR UCName[2];
+
+    UCName[0] = W32kSimpleToupper(IntMapVirtualKeyEx( VkCode, 2, keyLayout ));
+    UCName[1] = 0;
+    ret = 1;
+
+    if( !NT_SUCCESS(MmCopyToCaller( lpString, UCName, 2 * sizeof(WCHAR) )) )
+      return 0;
+  }
+
+  return ret;
+}
+
+/*
+ * Filter this message according to the current key layout, setting wParam
+ * appropriately.
+ */
+
+VOID FASTCALL W32kKeyProcessMessage(LPMSG Msg, PKBDTABLES KeyboardLayout) {
+  DWORD ScanCode = 0, ModifierBits = 0;
+  DWORD i = 0;
+  DWORD BaseMapping = 0;
+  DWORD RawVk = 0;
+  static WORD NumpadConversion[][2] = 
+    { { VK_DELETE, VK_DECIMAL },
+      { VK_INSERT, VK_NUMPAD0 },
+      { VK_END,    VK_NUMPAD1 },
+      { VK_DOWN,   VK_NUMPAD2 },
+      { VK_NEXT,   VK_NUMPAD3 },
+      { VK_LEFT,   VK_NUMPAD4 },
+      { VK_CLEAR,  VK_NUMPAD5 },
+      { VK_RIGHT,  VK_NUMPAD6 },
+      { VK_HOME,   VK_NUMPAD7 },
+      { VK_UP,     VK_NUMPAD8 },
+      { VK_PRIOR,  VK_NUMPAD9 },
+      { 0,0 } };
+
+  if( !KeyboardLayout || !Msg || 
+      (Msg->message != WM_KEYDOWN && Msg->message != WM_SYSKEYDOWN &&
+       Msg->message != WM_KEYUP   && Msg->message != WM_SYSKEYUP) ) 
+    {
+      return;
+    }
+
+  IntLockQueueState;
+
+  /* arty -- handle numpad -- On real windows, the actual key produced 
+   * by the messaging layer is different based on the state of numlock. */
+  ModifierBits = ModBits(KeyboardLayout,QueueKeyStateTable);
+
+  /* Get the raw scan code, so we can look up whether the key is a numpad
+   * key
+   *
+   * Shift and the LP_EXT_BIT cancel. */
+  ScanCode = (Msg->lParam >> 16) & 0xff;
+  BaseMapping = Msg->wParam = 
+    IntMapVirtualKeyEx( ScanCode, 1, KeyboardLayout );
+  if( ScanCode >= KeyboardLayout->bMaxVSCtoVK )
+    RawVk = 0;
+  else
+    RawVk = KeyboardLayout->pusVSCtoVK[ScanCode];
+
+  if ((ModifierBits & NUMLOCK_BIT) && 
+      !(ModifierBits & GetShiftBit(KeyboardLayout, VK_SHIFT)) && 
+      (RawVk & KNUMP) &&
+      !(Msg->lParam & LP_EXT_BIT))
+    {
+      /* The key in question is a numpad key.  Search for a translation. */
+      for (i = 0; NumpadConversion[i][0]; i++) 
+       {
+           if ((BaseMapping & 0xff) == NumpadConversion[i][0]) /* RawVk? */
+           {
+             Msg->wParam = NumpadConversion[i][1];
+             break;
+           }
+       }
+    }
+
+  DPRINT("Key: [%04x -> %04x]\n", BaseMapping, Msg->wParam);
+
+  /* Now that we have the VK, we can set the keymap appropriately
+   * This is a better place for this code, as it's guaranteed to be
+   * run, unlike translate message. */
+  if (Msg->message == WM_KEYDOWN || Msg->message == WM_SYSKEYDOWN)
+    {
+      SetKeyState( ScanCode, Msg->wParam, Msg->lParam & LP_EXT_BIT,
+                  TRUE ); /* Strike key */
+    }
+  else if (Msg->message == WM_KEYUP || Msg->message == WM_SYSKEYUP)
+    {
+      SetKeyState( ScanCode, Msg->wParam, Msg->lParam & LP_EXT_BIT,
+                  FALSE ); /* Release key */
+    }
+
+  /* We need to unset SYSKEYDOWN if the ALT key is an ALT+Gr */
+  if( QueueKeyStateTable[VK_RMENU] & KS_DOWN_BIT ) {
+      if( Msg->message == WM_SYSKEYDOWN ) Msg->message = WM_KEYDOWN;
+      else Msg->message = WM_KEYUP;
+  }
+
+  IntUnLockQueueState;
+}
 /* EOF */