X-Git-Url: http://git.joshuawise.com/netwatch.git/blobdiff_plain/434f1ddc94acd188ebbea1deae6c3cb67caf2275..6b91f09cf683c29989ab259f4a433fb7d1427b36:/aseg/counter.c diff --git a/aseg/counter.c b/aseg/counter.c index e4b5eae..ab200c1 100644 --- a/aseg/counter.c +++ b/aseg/counter.c @@ -1,82 +1,82 @@ #include +#include +#include +#include +#include +#include +#include "../net/net.h" +#include "vga-overlay.h" +#include "packet.h" +#include "keyboard.h" unsigned int counter = 0; +unsigned int lastctr = 0; unsigned long pcisave; unsigned char vgasave; -unsigned char thestr[512]; -unsigned char logents[4][41] = {0}; +static int curdev = 0; /* 0 if kbd, 1 if mouse */ -unsigned char vgaread(unsigned char idx) +static void cause_kbd_irq() { - outb(0x3D4, idx); - inb(0x3D5); -} - -void strblit(char *src, int r, int c) -{ - char *destp = (char*)(0xB8000UL | (((unsigned int)vgaread(0xC)) << 9) | (((unsigned int)vgaread(0xD)) << 1)) + r*80*2 + c*2; - unsigned char smramc; - - smramc = pci_read8(0, 0, 0, 0x70); - pci_write8(0, 0, 0, 0x70, (smramc & 0xF3) | 0x08); - while (*src) - { - *(destp++) = *(src++); - *(destp++) = 0x1F; - } - pci_write8(0, 0, 0, 0x70, smramc); -} - -void outlog() -{ - int y, x; - unsigned char smramc; - unsigned char *basep = (char*)(0xB8000UL | (((unsigned int)vgaread(0xC)) << 9) | (((unsigned int)vgaread(0xD)) << 1)); - - smramc = pci_read8(0, 0, 0, 0x70); - pci_write8(0, 0, 0, 0x70, (smramc & 0xF3) | 0x08); - for (y = 0; y < 4; y++) - for (x = 40; x < 80; x++) - { - basep[y*80*2+x*2] = ' '; - basep[y*80*2+x*2+1] = 0x1F; - } - pci_write8(0, 0, 0, 0x70, smramc); - - for (y = 0; y < 4; y++) - strblit(logents[y], y, 40); - -} - -void dolog(char *s) -{ - memmove(logents[0], logents[1], sizeof(logents[0])*3); - strcpy(logents[3], s); + outl(0x844, 0x0); + outl(0x848, 0x0); + while (inb(0x64) & 0x1) + inb(0x60); + outb(0x60, 0xee); /* Cause an IRQ. */ + while (inb(0x60) != 0xEE) + ; } void pci_dump() { - unsigned char s[40]; unsigned long cts; - static int curdev = 0; /* 0 if kbd, 1 if mouse */ cts = inl(0x84C); + outl(0x840, 0x0); outl(0x848, 0x0); - switch(cts&0xF0000) { case 0x20000: { unsigned char b; - strcpy(s, "READxxxxxxxxxxxxxxxx"); - tohex(s+4, cts); - b = inb(cts & 0xFFFF); - tohex(s+12, b); - if ((cts & 0xFFFF) == 0x64) - curdev = (b & 0x20) ? 1 : 0; - if ((curdev == 0) && ((cts & 0xFFFF) == 0x60) && (b == 0x01)) - outb(0xCF9, 0x4); - dolog(s); + + 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()) + { + dologf("OS wants to know; we have data"); + lastctr = counter; + 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 (kbd_has_injected_scancode()) + { + b = kbd_get_injected_scancode(); + lastctr = counter; + while (inb(0x64) & 0x1) + inb(0x60); + } else + b = inb(0x60); + if ((curdev == 0) && (b == 0x01)) { /* Escape */ + outb(0xCF9, 0x4); /* Reboot */ + return; + } + + /* If there is more nus to come, generate another IRQ. */ + if (kbd_has_injected_scancode()) + cause_kbd_irq(); + + *(unsigned char*)0xAFFD0 /* EAX */ = b; + break; + } + *(unsigned char*)0xAFFD0 /* EAX */ = b; break; } @@ -84,79 +84,95 @@ void pci_dump() { { unsigned char b; - strcpy(s, "WRITxxxxxxxxxxxxxxxx"); b = *(unsigned char*)0xAFFD0 /* EAX */; - tohex(s+4, cts); - tohex(s+12, b); - dolog(s); + dologf("WRITE: %08x (%02x)", cts, b); outb(cts & 0xFFFF, b); break; } default: dolog("Unhandled PCI cycle"); } + + outl(0x840, 0x0); + outl(0x844, 0x1000); + outl(0x848, 0x1000); } -void __start (void) +void timer_handler(smi_event_t ev) { - unsigned char smramc; - static int first = 1; + static unsigned int ticks = 0; - pcisave = inl(0xCF8); - vgasave = inb(0x3D4); + smi_disable_event(SMI_EVENT_FAST_TIMER); + smi_enable_event(SMI_EVENT_FAST_TIMER); - if (first) + if (kbd_has_injected_scancode() && (counter > (lastctr + 2))) { - first = 0; - dolog("NetWatch running..."); + smi_disable_event(SMI_EVENT_DEVTRAP_KBC); + dolog("Kicking timer"); + cause_kbd_irq(); + smi_enable_event(SMI_EVENT_DEVTRAP_KBC); } - - counter++; - outb(0x80, (counter & 0xFF)); - strcpy(thestr, "15-412! xxxxxxxx xxxxxxxx"); - tohex(thestr + 8, inl(0x0834)); - tohex(thestr + 17, counter); - strblit(thestr, 0, 0); + outb(0x80, (ticks++) & 0xFF); + + outlog(); +} + +void kbc_handler(smi_event_t ev) +{ + pci_dump(); +} + +void gbl_rls_handler(smi_event_t ev) +{ + unsigned long ecx; - if (inl(0x834) & 0x20) - dolog("Warning: unhandled APM access"); - if (inl(0x834) & 0x1000) + ecx = *(unsigned long*)0xAFFD4; + + packet_t * packet = check_packet(ecx); + if (!packet) { - if (inl(0x844) & 0x1000) /* devact_sts */ - { - pci_dump(); - outl(0x848, 0x1000); - outl(0x844, 0x1000); - } + dologf("WARN: bad packet at %08x", ecx); + return; } - if (inl(0x834) & 0x4000) - dolog("Long periodic timer"); + + dologf("Got packet: type %08x", packet->type); + + 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; + } +} + +void smi_entry(void) +{ + char statstr[512]; + + pcisave = inl(0xCF8); + vgasave = inb(0x3D4); + pci_unbother_all(); + + counter++; + sprintf(statstr, "15-412! %08x %08x", smi_status(), counter); + strblit(statstr, 0, 0); + + eth_poll(); + if (inl(0x840) & 0x1000) { - dolog("Caught device monitor trap"); pci_dump(); outl(0x840, 0x1100); outl(0x840, 0x0100); } - if (inl(0x834) & ~(0x4160)) - { - unsigned char s[40]; - strcpy(s, "Unknown: xxxxxxxx"); - tohex(s + 9, inl(0x834) & ~(0x140)); - dolog(s); - } - - outlog(); + smi_poll(); + pci_bother_all(); outl(0xCF8, pcisave); outb(0x3D4, vgasave); - - outl(0x848, 0x1000); - outl(0x834, /*0x40*/0xFFFF); // ack the periodic IRQ - outb(0x830, (inb(0x830) | 0x2) & ~0x40); - outb(0x830, inb(0x830) | 0x40); - }