1 files changed, 56 insertions, 8 deletions

diff --git a/src/ps2.rs b/src/ps2.rs
index 3a42c1a..4ea8b7a 100644
--- a/src/ps2.rs
+++ b/src/ps2.rs
@@ -158,9 +158,64 @@ static mut KBD: KbdState = KbdState {
     got_e0: false,
 };
 
-/// Initialise the PS/2 port for the keyboard — just drain stale bytes.
+/// Initialise the PS/2 port: drain stale bytes + enable keyboard interrupt.
 pub fn init() {
     drain_input();
+
+    // Enable the keyboard interrupt on the PS/2 controller.
+    // Read Configuration Byte (command 0x20), set bit 0 (IRQ1 enable),
+    // write back (command 0x60).
+    unsafe {
+        // Tell the keyboard controller we want to read the config byte.
+        // Need to wait for the controller to be ready first.
+        wait_write();
+        core::arch::asm!("out dx, al", in("dx") 0x64u16, in("al") 0x20u8,
+            options(nostack, preserves_flags));
+        wait_read();
+        let mut cfg: u8;
+        core::arch::asm!("in al, dx", out("al") cfg, in("dx") 0x60u16,
+            options(nostack, preserves_flags));
+
+        cfg |= 0x01; // enable keyboard interrupt
+
+        // Write config byte back
+        wait_write();
+        core::arch::asm!("out dx, al", in("dx") 0x64u16, in("al") 0x60u8,
+            options(nostack, preserves_flags));
+        wait_write();
+        core::arch::asm!("out dx, al", in("dx") 0x60u16, in("al") cfg,
+            options(nostack, preserves_flags));
+    }
+}
+
+/// Wait until the PS/2 controller is ready to receive a command.
+fn wait_write() {
+    unsafe {
+        loop {
+            let status: u8;
+            core::arch::asm!("in al, dx", out("al") status, in("dx") 0x64u16,
+                options(nostack, preserves_flags));
+            if status & 0x02 == 0 {
+                // Input buffer empty — controller ready to receive
+                break;
+            }
+        }
+    }
+}
+
+/// Wait until the PS/2 controller has data available to read.
+fn wait_read() {
+    unsafe {
+        loop {
+            let status: u8;
+            core::arch::asm!("in al, dx", out("al") status, in("dx") 0x64u16,
+                options(nostack, preserves_flags));
+            if status & 0x01 != 0 {
+                // Output buffer full — data ready to read
+                break;
+            }
+        }
+    }
 }
 
 fn drain_input() {
@@ -184,18 +239,11 @@ fn drain_input() {
 /// this function pops and decodes them.
 pub fn poll_key() -> KeyEvent {
     loop {
-        // Disable interrupts so the buffer-empty → hlt window is atomic.
-        unsafe { core::arch::asm!("cli", options(nostack, preserves_flags)); }
-
         if let Some(byte) = crate::interrupts::read_scancode() {
-            unsafe { core::arch::asm!("sti", options(nostack, preserves_flags)); }
             if let Some(ev) = process_byte(byte) {
                 return ev;
             }
         } else {
-            // Re-enable and halt.  If a keyboard IRQ arrived between the
-            // buffer check and `sti`, the CPU sees a pending interrupt
-            // and `hlt` returns immediately.
             unsafe {
                 core::arch::asm!("sti; hlt", options(nostack, preserves_flags));
             }