Espresso 0.0.0f

drivers/ps2_keyboard.c

@@ -1,5 +1,6 @@
 #include <stdio.h>
 #include <port_io.h>
+#include <tty.h>
 
 #include <drivers/ps2_keyboard.h>
 
@@ -16,7 +17,74 @@
   0x64 	 Write 	     Command Register
 */
 
+/*
+  Note:
+   The interrupt number for keyboard input in 1.
+*/
+
+#define PS2_DATA_PORT   0x60
+#define PS2_STATUS_PORT 0x64
+
+#define KEYBOARD_IRQ    1
+
+
+/* State for shift key */
+static bool shift_pressed = false;
+
+static const char scancode_map[128] = {
+  0,  27, '1','2','3','4','5','6','7','8','9','0','-','=','\b', /* Backspace */
+  '\t', // Tab
+  'q','w','e','r','t','y','u','i','o','p','[',']','\n', /* Enter */
+  0, /* Control */
+  'a','s','d','f','g','h','j','k','l',';','\'','`',
+  0, /* Left shift */
+  '\\','z','x','c','v','b','n','m',',','.','/',
+  0, /* Right shift */
+  '*',
+  0, /* Alt */
+  ' ', /* Spacebar */
+  0, /* Caps lock */
+  /* The rest are function and control keys */
+};
+
+
 void keyboard_init(void)
 {
-  
+  outb(0x64, 0xAE); /* Enable keyboard interface (often optional, but here for safety) */
+}
+
+void keyboard_handler(void) {
+  uint8_t scancode = inb(PS2_DATA_PORT);
+
+  /* Handle shift press/release */
+  if (scancode == 0x2A || scancode == 0x36)
+  {
+    shift_pressed = true;
+    return;
+  }
+  else if (scancode == 0xAA || scancode == 0xB6)
+  {
+    shift_pressed = false;
+    return;
+  }
+
+  if (scancode & 0x80)
+  {
+    /* Key release event, ignore for now */
+  } else
+  {
+    char c = scancode_map[scancode];
+    if (shift_pressed && c >= 'a' && c <= 'z') {
+      c -= 32; /* Convert to uppercase */
+    }
+    else if (shift_pressed)
+    {
+      c = (char) terminal_get_shifted((unsigned char) c);
+    }
+
+    if (c)
+    {
+      printf("%c", c);
+    }
+  }
 }