more keys

[netwatch.git] / aseg / counter.c

diff --git a/aseg/counter.c b/aseg/counter.c
index f5a2602da9344fe7d1a4d7dbd9338034934e0997..0ab5d4c6355efa6d9e4dadc51ccddd5231efa0b8 100644 (file)
--- a/aseg/counter.c
+++ b/aseg/counter.c
@@ -7,6 +7,7 @@
 #include "../net/net.h"
 #include "vga-overlay.h"
 #include "packet.h"
+#include "keyboard.h"
 
 unsigned int counter = 0;
 unsigned long pcisave;
@@ -15,6 +16,7 @@ unsigned char vgasave;
 void pci_dump() {
        unsigned long cts;
        static int curdev = 0;  /* 0 if kbd, 1 if mouse */
+       static int takeover = 0;
                
        cts = inl(0x84C);
        
@@ -25,12 +27,37 @@ void pci_dump() {
        case 0x20000:
        {
                unsigned char b;
-               b = inb(cts & 0xFFFF);
-               //dologf("READ: %08x (%02x)", cts, b);
-               if ((cts & 0xFFFF) == 0x64)
-                       curdev = (b & 0x20) ? 1 : 0;
-               if ((curdev == 0) && ((cts & 0xFFFF) == 0x60) && (b == 0x01))
-                       outb(0xCF9, 0x4);
+               
+               switch (cts & 0xFFFF)
+               {
+               case 0x64:
+                       /* Read the real hardware and mask in our OBF if need be. */
+                       b = inb(0x64);
+                       if (kbd_has_injected_scancode())
+                       {
+                               takeover = 1;
+                               b |= 0x01;
+                               b &= ~0x20;     /* no mouse for you! */
+                               curdev = 0;
+                       } else 
+                               curdev = (b & 0x20) ? 1 : 0;
+                       *(unsigned char*)0xAFFD0 /* EAX */ = b;
+                       break;
+               case 0x60:
+                       if (takeover)
+                       {
+                               b = kbd_get_injected_scancode();
+                               takeover = 0;
+                       } else
+                               b = inb(0x60);
+                       if ((curdev == 0) && (b == 0x01)) {     /* Escape */
+                               outb(0xCF9, 0x4);       /* Reboot */
+                               return;
+                       }
+                       *(unsigned char*)0xAFFD0 /* EAX */ = b;
+                       break;
+               }
+
                *(unsigned char*)0xAFFD0 /* EAX */ = b;
                break;
        }
@@ -86,6 +113,8 @@ void gbl_rls_handler(smi_event_t ev)
        if (packet->type == 42) {
                dump_log((char *)packet->data);
                *(unsigned long*)0xAFFD4 = 42;
+       } else if (packet->type == 0xAA) {
+               kbd_inject_key('A');
        } else {
                *(unsigned long*)0xAFFD4 = 0x2BADD00D;
        }