From ef2d8ed00cdf152aa36dc5b51f1f413128ef344a Mon Sep 17 00:00:00 2001
From: Matheus <matheusuky@gmail.com>
Date: Thu, 4 May 2023 19:58:24 -0300
Subject: [PATCH] kernel: ps2kbd

---
 kernel-src/arch/x86-64/main.c           |   3 +-
 kernel-src/arch/x86-64/ps2kbd.c         | 163 ++++++++++++++++++++++++
 kernel-src/include/x86-64/arch/ps2kbd.h |   6 +
 3 files changed, 171 insertions(+), 1 deletion(-)
 create mode 100644 kernel-src/arch/x86-64/ps2kbd.c
 create mode 100644 kernel-src/include/x86-64/arch/ps2kbd.h

diff --git a/kernel-src/arch/x86-64/main.c b/kernel-src/arch/x86-64/main.c
index 1fed4075..5ddd9876 100644
--- a/kernel-src/arch/x86-64/main.c
+++ b/kernel-src/arch/x86-64/main.c
@@ -20,6 +20,7 @@
 #include <kernel/devfs.h>
 #include <kernel/pseudodevices.h>
 #include <arch/ps2.h>
+#include <kernel/keyboard.h>
 
 static cpu_t bsp_cpu;
 
@@ -58,7 +59,7 @@ void kernel_entry() {
 	__assert(vfs_mount(NULL, vfsroot, "/dev", "devfs", NULL) == 0);
 
 	pseudodevices_init();
-
+	keyboard_init();
 	arch_ps2_init();
 
 	__assert(!"kernel entry end");
diff --git a/kernel-src/arch/x86-64/ps2kbd.c b/kernel-src/arch/x86-64/ps2kbd.c
new file mode 100644
index 00000000..916bfdb0
--- /dev/null
+++ b/kernel-src/arch/x86-64/ps2kbd.c
@@ -0,0 +1,163 @@
+#include <arch/ps2kbd.h>
+#include <stdbool.h>
+#include <arch/io.h>
+#include <arch/cpu.h>
+#include <kernel/interrupt.h>
+#include <arch/apic.h>
+#include <kernel/keyboard.h>
+#include <arch/ps2.h>
+#include <logging.h>
+
+static char codes[128] = {
+	KEYCODE_RESERVED,
+	KEYCODE_ESCAPE,
+	KEYCODE_1,
+	KEYCODE_2,
+	KEYCODE_3,
+	KEYCODE_4,
+	KEYCODE_5,
+	KEYCODE_6,
+	KEYCODE_7,
+	KEYCODE_8,
+	KEYCODE_9,
+	KEYCODE_0,
+	KEYCODE_MINUS,
+	KEYCODE_EQUAL,
+	KEYCODE_BACKSPACE,
+	KEYCODE_TAB,
+	KEYCODE_Q,
+	KEYCODE_W,
+	KEYCODE_E,
+	KEYCODE_R,
+	KEYCODE_T,
+	KEYCODE_Y,
+	KEYCODE_U,
+	KEYCODE_I,
+	KEYCODE_O,
+	KEYCODE_P,
+	KEYCODE_LEFTBRACE,
+	KEYCODE_RIGHTBRACE,
+	KEYCODE_ENTER,
+	KEYCODE_LEFTCTRL,
+	KEYCODE_A,
+	KEYCODE_S,
+	KEYCODE_D,
+	KEYCODE_F,
+	KEYCODE_G,
+	KEYCODE_H,
+	KEYCODE_J,
+	KEYCODE_K,
+	KEYCODE_L,
+	KEYCODE_SEMICOLON,
+	KEYCODE_APOSTROPHE,
+	KEYCODE_GRAVE,
+	KEYCODE_LEFTSHIFT,
+	KEYCODE_BACKSLASH,
+	KEYCODE_Z,
+	KEYCODE_X,
+	KEYCODE_C,
+	KEYCODE_V,
+	KEYCODE_B,
+	KEYCODE_N,
+	KEYCODE_M,
+	KEYCODE_COMMA,
+	KEYCODE_DOT,
+	KEYCODE_SLASH,
+	KEYCODE_RIGHTSHIFT,
+	KEYCODE_KEYPADASTERISK,
+	KEYCODE_LEFTALT,
+	KEYCODE_SPACE,
+	KEYCODE_CAPSLOCK,
+	KEYCODE_F1,
+	KEYCODE_F2,
+	KEYCODE_F3,
+	KEYCODE_F4,
+	KEYCODE_F5,
+	KEYCODE_F6,
+	KEYCODE_F7,
+	KEYCODE_F8,
+	KEYCODE_F9,
+	KEYCODE_F10,
+	KEYCODE_NUMLOCK,
+	KEYCODE_SCROLLLOCK,
+	KEYCODE_KEYPAD7,
+	KEYCODE_KEYPAD8,
+	KEYCODE_KEYPAD9,
+	KEYCODE_KEYPADMINUS,
+	KEYCODE_KEYPAD4,
+	KEYCODE_KEYPAD5,
+	KEYCODE_KEYPAD6,
+	KEYCODE_KEYPADPLUS,
+	KEYCODE_KEYPAD1,
+	KEYCODE_KEYPAD2,
+	KEYCODE_KEYPAD3,
+	KEYCODE_KEYPAD0,
+	KEYCODE_KEYPADDOT,
+	0, 0, 0,
+	KEYCODE_F11,
+	KEYCODE_F12
+};
+
+static char extendedcodes[128] = {
+	[0x1C] = KEYCODE_KEYPADENTER,
+	[0x1D] = KEYCODE_RIGHTCTRL, 
+	[0x35] = KEYCODE_KEYPADSLASH, // k/
+	[0x38] = KEYCODE_RIGHTALT, // altgr
+	[0x47] = KEYCODE_HOME, // home
+	[0x48] = KEYCODE_UP, // up
+	[0x49] = KEYCODE_PAGEUP, // page up
+	[0x4B] = KEYCODE_LEFT, // left
+	[0x4D] = KEYCODE_RIGHT, // right
+	[0x4F] = KEYCODE_END, // end
+	[0x50] = KEYCODE_DOWN, // down
+	[0x51] = KEYCODE_PAGEDOWN, // page down
+	[0x52] = KEYCODE_INSERT, // insert
+	[0x53] = KEYCODE_DELETE // delete
+};
+
+#define KEYBOARDIRQ 1
+
+keyboard_t *kb;
+static bool extended = false;
+
+static void kbdisr(isr_t *isr, context_t *ctx) {
+	uint8_t scancode = inb(0x60);
+	if (scancode == 0xE0) {
+		extended = true;
+		return;
+	}
+
+	kbpacket_t packet;
+
+	packet.flags = 0;
+
+	if (scancode & 0x80) {
+		packet.flags = KBPACKET_FLAGS_RELEASED;
+		scancode &= 0x7F;
+	}
+
+	char *tab = codes;
+
+	if (extended) {
+		tab = extendedcodes;
+		extended = false;
+	}
+
+	packet.keycode = tab[scancode];
+
+	if (!packet.keycode)
+		return;
+
+	keyboard_sendpacket(kb, &packet);
+}
+
+void ps2kbd_init(){
+	// clear PS/2 buffer to make sure its empty
+	inb(PS2_PORT_DATA);
+	isr_t *isr = interrupt_allocate(kbdisr, arch_apic_eoi, IPL_KEYBOARD);
+	__assert(isr);
+	arch_ioapic_setirq(KEYBOARDIRQ, isr->id & 0xff, _cpu()->id, false);
+	kb = keyboard_new();
+	__assert(kb);
+	printf("ps2kbd: irq enabled with vector %u\n", isr->id & 0xff);
+}
diff --git a/kernel-src/include/x86-64/arch/ps2kbd.h b/kernel-src/include/x86-64/arch/ps2kbd.h
new file mode 100644
index 00000000..dda8a515
--- /dev/null
+++ b/kernel-src/include/x86-64/arch/ps2kbd.h
@@ -0,0 +1,6 @@
+#ifndef _PS2KBD_H
+#define _PS2KBD_H
+
+void ps2kbd_init();
+
+#endif