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, PS2Keyboard, KB_Install, NULL, NULL);
  28 tDevFS_Driver   gKB_DevInfo = {
  29         NULL, "PS2Keyboard",
  30         {
  31         .NumACLs = 0,
  32         .Size = 0,
  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                 if(ch != 0 && gKB_Callback)
 126                         gKB_Callback( ch & 0x80000000 );
 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)             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 0: break;
 161                 case '`':       ch = '~';       break;
 162                 case '1':       ch = '!';       break;
 163                 case '2':       ch = '@';       break;
 164                 case '3':       ch = '#';       break;
 165                 case '4':       ch = '$';       break;
 166                 case '5':       ch = '%';       break;
 167                 case '6':       ch = '^';       break;
 168                 case '7':       ch = '&';       break;
 169                 case '8':       ch = '*';       break;
 170                 case '9':       ch = '(';       break;
 171                 case '0':       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                 case '/':       ch = '?';       break;
 182                 default:
 183                         if('a' <= ch && ch <= 'z')
 184                                 ch -= 0x20;
 185                         break;
 186                 }
 187         }
 188
 189         if(gKB_Callback && ch != 0)     gKB_Callback(ch);
 190 }
 191
 192 /**
 193  * \fn void KB_UpdateLEDs()
 194  * \brief Updates the status of the keyboard LEDs
 195  */
 196 void KB_UpdateLEDs()
 197 {
 198         Uint8   leds;
 199
 200         leds = (gbKB_CapsState ? 4 : 0);
 201
 202         while( inb(0x64) & 2 ); // Wait for bit 2 to unset
 203         outb(0x60, 0xED);       // Send update command
 204
 205         while( inb(0x64) & 2 ); // Wait for bit 2 to unset
 206         outb(0x60, leds);
 207 }
 208
 209 /**
 210  * \fn int KB_IOCtl(tVFS_Node *Node, int Id, void *Data)
 211  * \brief Calls an IOCtl Command
 212  */
 213 int KB_IOCtl(tVFS_Node *Node, int Id, void *Data)
 214 {
 215         switch(Id)
 216         {
 217         case DRV_IOCTL_TYPE:    return DRV_TYPE_KEYBOARD;
 218         case DRV_IOCTL_IDENT:   memcpy(Data, "KB\0\0", 4);      return 1;
 219         case DRV_IOCTL_VERSION: return 0x100;
 220         case DRV_IOCTL_LOOKUP:  return 0;
 221
 222         // Sets the Keyboard Callback
 223         case KB_IOCTL_SETCALLBACK:
 224                 // Sanity Check
 225                 if((Uint)Data < KERNEL_BASE)    return 0;
 226                 // Can only be set once
 227                 if(gKB_Callback != NULL)        return 0;
 228                 // Set Callback
 229                 gKB_Callback = Data;
 230                 return 1;
 231
 232         default:
 233                 return 0;
 234         }
 235 }