X-Git-Url: http://git.joshuawise.com/netwatch.git/blobdiff_plain/245f29105322dacb79709d220365045eca28b83e..cdde55f54a0880bde5694fcdea8e386ada0cd5ce:/aseg-paging/main.c?ds=inline

diff --git a/aseg-paging/main.c b/aseg-paging/main.c
index 1bc502e..56e856c 100644
--- a/aseg-paging/main.c
+++ b/aseg-paging/main.c
@@ -9,7 +9,7 @@
 #include "../net/net.h"
 #include "vga-overlay.h"
 #include "../aseg/packet.h"
-#include "../aseg/keyboard.h"
+#include "keyboard.h"
 
 unsigned int lastctr = 0;
 extern unsigned int counter;
@@ -17,6 +17,33 @@ extern unsigned int counter;
 static int _ibf_ready = 0;
 static int _waiting_for_data = 0;
 static int curdev = 0;
+static int adding_locks_from_time_to_time = 0;
+
+static int _inject_ready()
+{
+	return _ibf_ready && !_waiting_for_data && !adding_locks_from_time_to_time;
+}
+
+void _try_inject()
+{
+	if (kbd_has_injected_scancode() && _inject_ready())
+	{
+		smi_disable_event(SMI_EVENT_DEVTRAP_KBC);
+		int i = 1000;
+		while (inb(0x64) & 0x02)	/* Busy? */
+			;
+		outb(0x64, 0xD2);	/* "Inject, please!" */
+		while (inb(0x64) & 0x02)	/* Busy? */
+			;
+		outb(0x60, kbd_get_injected_scancode());	/* data */
+		while ((inb(0x64) & 0x02) && i--)	/* wait for completion */
+			;
+		outl(0x844, 0x1000);
+		adding_locks_from_time_to_time++;
+		smi_enable_event(SMI_EVENT_DEVTRAP_KBC);
+	} else if (kbd_has_injected_scancode())
+		outputf("Would like to inject, but %d %d", _ibf_ready, _waiting_for_data);
+}
 
 void pci_dump() {
 	unsigned long cts;
@@ -38,18 +65,28 @@ void pci_dump() {
 			b = inb(0x64);
 			
 			curdev = (b & 0x20 /* KBD_STAT_MOUSE_OBF */) ? 1 : 0;
+			
+			if ((curdev == 0) && kbd_has_injected_scancode())
+				b |= 0x1;
+
 			_ibf_ready = (b & 0x2 /* IBF */) ? 0 : 1;
 			
 			break;
 		case 0x60:
-			b = inb(0x60);
+			if ((curdev == 0) && kbd_has_injected_scancode() && !adding_locks_from_time_to_time)
+				b = kbd_get_injected_scancode();
+			else
+				b = inb(0x60);
+			if (adding_locks_from_time_to_time)
+				adding_locks_from_time_to_time--;
 			if ((curdev == 0) && (b == 0x01)) {	/* Escape */
 				outb(0xCF9, 0x4);	/* Reboot */
 				return;
 			}
 			break;
 		}
-
+		
+		dologf("READ : %08x (%02x)", cts, b);
 		*(unsigned char*)0xAFFD0 /* EAX */ = b;
 		break;
 	}
@@ -83,11 +120,6 @@ void pci_dump() {
 	outl(0x848, 0x1000);
 }
 
-static int _inject_ready()
-{
-	return _ibf_ready && !_waiting_for_data;
-}
-
 void timer_handler(smi_event_t ev)
 {
 	static unsigned int ticks = 0;
@@ -95,20 +127,7 @@ void timer_handler(smi_event_t ev)
 	smi_disable_event(SMI_EVENT_FAST_TIMER);
 	smi_enable_event(SMI_EVENT_FAST_TIMER);
 	
-	if (kbd_has_injected_scancode() && _inject_ready())
-	{
-		smi_disable_event(SMI_EVENT_DEVTRAP_KBC);
-		outputf("Injecting key");
-		/* Actually do the inject. */
-		outb(0x64, 0xD2);	/* "Inject, please!" */
-		while (inb(0x64) & 0x02)	/* Busy? */
-			;
-		outb(0x60, kbd_get_injected_scancode());	/* data */
-		while (inb(0x64) & 0x02)	/* wait for completion */
-			;
-		smi_enable_event(SMI_EVENT_DEVTRAP_KBC);
-	} else if (kbd_has_injected_scancode())
-		outputf("Would like to inject, but %d %d", _ibf_ready, _waiting_for_data);
+	_try_inject();
 	
 	outb(0x80, (ticks++) & 0xFF);
 	
@@ -140,7 +159,8 @@ void gbl_rls_handler(smi_event_t ev)
 		dump_log((char *)packet->data);
 		*(unsigned long*)0xAFFD4 = 42;
 	} else if (packet->type == 0xAA) {
-		kbd_inject_key('A');
+		kbd_inject_keysym('A', 1);
+		kbd_inject_keysym('A', 0);
 	} else {
 		*(unsigned long*)0xAFFD4 = 0x2BADD00D;
 	}