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

[netwatch.git] / aseg / counter.c

diff --git a/aseg/counter.c b/aseg/counter.c
index d52316843f1c2f41b4459ff3b33548c251a91d8f..ab200c107c8c73e441c7ba9211a72688a80e70b4 100644 (file)
--- a/aseg/counter.c
+++ b/aseg/counter.c
@@ -2,38 +2,81 @@
 #include <smram.h>
 #include <video_defines.h>
 #include <minilib.h>
-
-char thestr[512];
-
+#include <smi.h>
+#include <pci-bother.h>
+#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;
+static int curdev = 0; /* 0 if kbd, 1 if mouse */
+
+static void cause_kbd_irq()
+{
+       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(0x848, 0x0);
        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;
        }
@@ -41,11 +84,8 @@ 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;
        }
@@ -53,67 +93,86 @@ void pci_dump() {
                dolog("Unhandled PCI cycle");
        }
        
+       outl(0x840, 0x0);
+       outl(0x844, 0x1000);
        outl(0x848, 0x1000);
-       outl(0x840, 0x0100);
 }
 
-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(0x844, 0x1000);    /* ack it */
-               }
+               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)
        {
                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);
-       
 }