X-Git-Url: http://git.joshuawise.com/netwatch.git/blobdiff_plain/91cc13402921ddb5b3b4e695b036a43c3722e615..66cd7e8220e0f7dbb0e43f44f6aab60194548ad0:/aseg-paging/main.c?ds=sidebyside diff --git a/aseg-paging/main.c b/aseg-paging/main.c index d8fecdd..56e856c 100644 --- a/aseg-paging/main.c +++ b/aseg-paging/main.c @@ -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,19 +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); - } + _try_inject(); outb(0x80, (ticks++) & 0xFF);