X-Git-Url: http://git.joshuawise.com/netwatch.git/blobdiff_plain/5ae81164883c737f0204dc26beb7bb7488693a1d..037ce545946a8bc2d482d24e3bbdb6174a93ca46:/aseg/counter.c?ds=inline

diff --git a/aseg/counter.c b/aseg/counter.c
index d523168..ab200c1 100644
--- 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);
-	
 }