Merge nyus.joshuawise.com:/storage/git/netwatch

[netwatch.git] / aseg-paging / main.c

diff --git a/aseg-paging/main.c b/aseg-paging/main.c
index 555a6a19f694b60618b18f389a9a11bf2514d4b0..41465e2629059f41417a08c35977a23f7ae8cae6 100644 (file)
--- 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,31 @@ 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;
-               }
 
+               default:
+                       b = inb(cts & 0xFFFF);
+               }
+               
+               dologf("READ : %08x (%02x)", cts, b);
                *(unsigned char*)0xAFFD0 /* EAX */ = b;
                break;
        }
@@ -83,11 +123,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,19 +130,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);
-       }
+       _try_inject();
        
        outb(0x80, (ticks++) & 0xFF);
        
@@ -139,7 +162,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;
        }