Kernel/drv/kb.c

   1 /*
   2  * Acess2
   3  * PS2 Keyboard Driver
   4  */
   5 #include <common.h>
   6 #include <modules.h>
   7 #include <fs_devfs.h>
   8 #include <tpl_drv_common.h>
   9 #include <tpl_drv_keyboard.h>
  10 #include "kb_kbdus.h"
  11
  12 // === CONSTANTS ===
  13 #define KB_BUFFER_SIZE  1024
  14
  15 // === IMPORTS ===
  16 void    Threads_Dump();
  17
  18 // === PROTOTYPES ===
  19  int    KB_Install(char **Arguments);
  20 void    KB_IRQHandler();
  21 void    KB_AddBuffer(char ch);
  22 Uint64  KB_Read(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Dest);
  23 void    KB_UpdateLEDs();
  24  int    KB_IOCtl(tVFS_Node *Node, int Id, void *Data);
  25
  26 // === GLOBALS ===
  27 MODULE_DEFINE(0, 0x0100, PS2Keybard, KB_Install, NULL, NULL);
  28 tDevFS_Driver   gKB_DevInfo = {
  29         NULL, "PS2Keyboard",
  30         {
  31         .NumACLs = 0,
  32         .Size = -1,
  33         .Read = KB_Read,
  34         .IOCtl = KB_IOCtl
  35         }
  36 };
  37 tKeybardCallback        gKB_Callback = NULL;
  38 Uint8   **gpKB_Map = gpKBDUS;
  39 Uint8   gbaKB_States[256];
  40  int    gbKB_ShiftState = 0;
  41  int    gbKB_CapsState = 0;
  42  int    gbKB_KeyUp = 0;
  43  int    giKB_KeyLayer = 0;
  44 //Uint64        giKB_ReadBase = 0;
  45 Uint8   gaKB_Buffer[KB_BUFFER_SIZE];    //!< Keyboard Ring Buffer
  46 volatile int    giKB_InsertPoint = 0;   //!< Writing location marker
  47 volatile int    giKB_ReadPoint = 0;     //!< Reading location marker
  48 volatile int    giKB_InUse = 0;         //!< Lock marker
  49
  50 // === CODE ===
  51 /**
  52  * \fn int KB_Install(char **Arguments)
  53  */
  54 int KB_Install(char **Arguments)
  55 {
  56         IRQ_AddHandler(1, KB_IRQHandler);
  57         DevFS_AddDevice( &gKB_DevInfo );
  58         return 1;
  59 }
  60
  61 /**
  62  * \fn void KB_IRQHandler()
  63  * \brief Called on a keyboard IRQ
  64  */
  65 void KB_IRQHandler()
  66 {
  67         Uint8   scancode;
  68         Uint32  ch;
  69         // int  keyNum;
  70
  71         //if( inportb(0x64) & 0x20 )    return;
  72
  73         scancode = inb(0x60); // Read from the keyboard's data buffer
  74
  75         // Ignore ACKs
  76         if(scancode == 0xFA) {
  77                 //kb_lastChar = KB_ACK;
  78                 return;
  79         }
  80
  81         // Layer +1
  82         if(scancode == 0xE0) {
  83                 giKB_KeyLayer = 1;
  84                 return;
  85         }
  86         // Layer +2
  87         if(scancode == 0xE1) {
  88                 giKB_KeyLayer = 2;
  89                 return;
  90         }
  91
  92         #if KB_ALT_SCANCODES
  93         if(scancode == 0xF0)
  94         {
  95                 gbKB_KeyUp = 1;
  96                 return;
  97         }
  98         #else
  99         if(scancode & 0x80)
 100         {
 101                 scancode &= 0x7F;
 102                 gbKB_KeyUp = 1;
 103         }
 104         #endif
 105
 106         // Translate
 107         ch = gpKB_Map[giKB_KeyLayer][scancode];
 108         //keyNum = giKB_KeyLayer * 256 + scancode;
 109         // Check for unknown key
 110         if(!ch && !gbKB_KeyUp)
 111                 Warning("UNK %i %x", giKB_KeyLayer, scancode);
 112
 113         // Reset Layer
 114         giKB_KeyLayer = 0;
 115
 116         // Key Up?
 117         if (gbKB_KeyUp)
 118         {
 119                 gbKB_KeyUp = 0;
 120                 gbaKB_States[ ch ] = 0; // Unset key state flag
 121
 122                 if( !gbaKB_States[KEY_LSHIFT] && !gbaKB_States[KEY_RSHIFT] )
 123                         gbKB_ShiftState = 0;
 124
 125                 KB_AddBuffer(KEY_KEYUP);
 126                 KB_AddBuffer(ch);
 127
 128                 return;
 129         }
 130
 131         // Set the bit relating to the key
 132         gbaKB_States[ch] = 1;
 133         if(ch == KEY_LSHIFT || ch == KEY_RSHIFT)
 134                 gbKB_ShiftState = 1;
 135
 136         // Check for Caps Lock
 137         if(ch == KEY_CAPSLOCK) {
 138                 gbKB_CapsState = !gbKB_CapsState;
 139                 KB_UpdateLEDs();
 140         }
 141
 142         // Ignore Non-Printable Characters
 143         if(ch == 0 || ch & 0x80)                return;
 144
 145         // --- Check for Kernel Magic Combos
 146         if(gbaKB_States[KEY_LCTRL] && gbaKB_States[KEY_LALT])
 147         {
 148                 switch(ch)
 149                 {
 150                 case 'd':       __asm__ __volatile__ ("xchg %bx, %bx"); break;
 151                 case 'p':       Threads_Dump(); break;
 152                 }
 153         }
 154
 155         // Is shift pressed
 156         if(gbKB_ShiftState ^ gbKB_CapsState)
 157         {
 158                 switch(ch)
 159                 {
 160                 case '`':       ch = '~';       break;
 161                 case '1':       ch = '!';       break;
 162                 case '2':       ch = '@';       break;
 163                 case '3':       ch = '#';       break;
 164                 case '4':       ch = '$';       break;
 165                 case '5':       ch = '%';       break;
 166                 case '6':       ch = '^';       break;
 167                 case '7':       ch = '&';       break;
 168                 case '8':       ch = '*';       break;
 169                 case '9':       ch = '(';       break;
 170                 case '0':       ch = ')';       break;
 171                 case '-':       ch = '_';       break;
 172                 case '=':       ch = '+';       break;
 173                 case '[':       ch = '{';       break;
 174                 case ']':       ch = '}';       break;
 175                 case '\\':      ch = '|';       break;
 176                 case ';':       ch = ':';       break;
 177                 case '\'':      ch = '"';       break;
 178                 case ',':       ch = '<';       break;
 179                 case '.':       ch = '>';       break;
 180                 case '/':       ch = '?';       break;
 181                 default:
 182                         if('a' <= ch && ch <= 'z')
 183                                 ch -= 0x20;
 184                         break;
 185                 }
 186         }
 187
 188         if(gKB_Callback)        gKB_Callback(ch);
 189 }
 190
 191 /**
 192  * \fn void KB_AddBuffer(char ch)
 193  * \brief Add to the keyboard ring buffer
 194  */
 195 void KB_AddBuffer(char ch)
 196 {
 197         // Add to buffer
 198         gaKB_Buffer[ giKB_InsertPoint++ ] = ch;
 199         // - Wrap
 200         if( giKB_InsertPoint == KB_BUFFER_SIZE )        giKB_InsertPoint = 0;
 201         // - Force increment read pointer
 202         if( giKB_InsertPoint == giKB_ReadPoint )        giKB_ReadPoint ++;
 203 }
 204
 205 /**
 206  * \fn Uint64 KB_Read(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Dest)
 207  * \brief Read from the ring buffer
 208  * \param Node  Unused
 209  * \param Offset        Unused (Character Device)
 210  * \param Length        Number of bytes to read
 211  * \param Dest  Destination
 212  */
 213 Uint64 KB_Read(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Dest)
 214 {
 215          int    pos = 0;
 216         char    *dstbuf = Dest;
 217
 218         if(giKB_InUse)  return -1;
 219         giKB_InUse = 1;
 220
 221         while(giKB_ReadPoint != giKB_InsertPoint && pos < Length)
 222         {
 223                 dstbuf[pos++] = gaKB_Buffer[ giKB_ReadPoint++ ];
 224                 if( giKB_ReadPoint == KB_BUFFER_SIZE )  giKB_InsertPoint = 0;
 225         }
 226
 227         giKB_InUse = 0;
 228
 229         return Length;
 230 }
 231
 232 /**
 233  * \fn void KB_UpdateLEDs()
 234  * \brief Updates the status of the keyboard LEDs
 235  */
 236 void KB_UpdateLEDs()
 237 {
 238         Uint8   leds;
 239
 240         leds = (gbKB_CapsState ? 4 : 0);
 241
 242         while( inb(0x64) & 2 ); // Wait for bit 2 to unset
 243         outb(0x60, 0xED);       // Send update command
 244
 245         while( inb(0x64) & 2 ); // Wait for bit 2 to unset
 246         outb(0x60, leds);
 247 }
 248
 249 /**
 250  * \fn int KB_IOCtl(tVFS_Node *Node, int Id, void *Data)
 251  * \brief Calls an IOCtl Command
 252  */
 253 int KB_IOCtl(tVFS_Node *Node, int Id, void *Data)
 254 {
 255         switch(Id)
 256         {
 257         case DRV_IOCTL_TYPE:    return DRV_TYPE_KEYBOARD;
 258         case DRV_IOCTL_IDENT:   memcpy(Data, "KB\0\0", 4);      return 1;
 259         case DRV_IOCTL_VERSION: return 0x100;
 260         case DRV_IOCTL_LOOKUP:  return 0;
 261
 262         // Sets the Keyboard Callback
 263         case KB_IOCTL_SETCALLBACK:
 264                 // Sanity Check
 265                 if((Uint)Data < KERNEL_BASE)    return 0;
 266                 // Can only be set once
 267                 if(gKB_Callback != NULL)        return 0;
 268                 // Set Callback
 269                 gKB_Callback = Data;
 270                 return 1;
 271
 272         default:
 273                 return 0;
 274         }
 275 }