X-Git-Url: http://git.joshuawise.com/netwatch.git/blobdiff_plain/36ce375d87e8dd0fdee14de4fbc35c1aef56d7e9..03bcc4db7995b38ecda927e65d7bf2e657477664:/aseg/counter.c diff --git a/aseg/counter.c b/aseg/counter.c index ccfff72..ab200c1 100644 --- a/aseg/counter.c +++ b/aseg/counter.c @@ -2,102 +2,81 @@ #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; -char thestr[512]; -char logents[4][41] = {{0}}; +static int curdev = 0; /* 0 if kbd, 1 if mouse */ -#define VRAM_BASE 0xA0000UL -#define TEXT_CONSOLE_OFFSET 0x18000UL - -#define TEXT_CONSOLE_BASE (VRAM_BASE + TEXT_CONSOLE_OFFSET) - -#define COLOR 0x1F - -unsigned char vga_read(unsigned char idx) -{ - outb(CRTC_IDX_REG, idx); - return inb(CRTC_DATA_REG); -} - -char * vga_base() -{ - return (char *) ( - TEXT_CONSOLE_BASE - | (((unsigned int) vga_read(CRTC_START_ADDR_LSB_IDX)) << 9) - | (((unsigned int) vga_read(CRTC_START_ADDR_MSB_IDX)) << 1) - ); -} - -void strblit(char *src, int row, int col) +static void cause_kbd_irq() { - char *destp = vga_base() + row * 80 * 2 + col * 2; - smram_state_t old_state = smram_save_state(); - - smram_aseg_set_state(SMRAM_ASEG_SMMCODE); - - while (*src) - { - *(destp++) = *(src++); - *(destp++) = COLOR; - } - - smram_restore_state(old_state); -} - -void outlog() -{ - int y, x; - char *basep = vga_base(); - - smram_state_t old_state = smram_save_state(); - - smram_aseg_set_state(SMRAM_ASEG_SMMCODE); - - 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; - } - - smram_restore_state(old_state); - - 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() { - 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; } @@ -105,78 +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) { - 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; + } + + 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; } - if (inl(0x834) & 0x4000) - dolog("Long periodic timer"); +} + +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)) - { - 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); - }