Enable caching while in SMM.

[netwatch.git] / aseg-paging / smi.c

diff --git a/aseg-paging/smi.c b/aseg-paging/smi.c
index 33536ebb1be5e48fb1d5668b2dc021fb3db8cff7..d519341a9370446fdbe8e1e65e56419271f4abb4 100644 (file)
--- a/aseg-paging/smi.c
+++ b/aseg-paging/smi.c
@@ -4,54 +4,41 @@
 #include <minilib.h>
 #include <smi.h>
 #include <pci-bother.h>
+#include <serial.h>
+#include <fb.h>
+#include <output.h>
+#include <msr.h>
 #include "../net/net.h"
 #include "vga-overlay.h"
 
-#include "pagetable.h"
-
 unsigned int counter = 0;
-unsigned int lastctr = 0;
 unsigned long pcisave = 0;
 unsigned char vgasave = 0;
 
-void set_cr0(unsigned int);
-
-#define get_cr0() \
-    ({ \
-        register unsigned int _temp__; \
-        asm volatile("mov %%cr0, %0" : "=r" (_temp__)); \
-        _temp__; \
-    })
-
-
-#define set_cr3(value) \
-    { \
-        register unsigned int _temp__ = (value); \
-        asm volatile("mov %0, %%cr3" : : "r" (_temp__)); \
-     }
-#define        CR0_PG  0x80000000
-
-
 void smi_entry(void)
 {
        char statstr[512];
-       outb(0x80, 0x0B);
+
+       /* Reenable caching on SMRAM. */
+       WRMSR(0x202, (RDMSR(0x202) & ~(0xFFULL)) | 0x06ULL);
+
        pcisave = inl(0xCF8);
        vgasave = inb(0x3D4);
-       outb(0x80, 0x1B);
-/*
        pci_unbother_all();
- */
+       
+       serial_init();
+       
+       if (fb)
+               fb->getvmode(fb->priv);
+
        counter++;
-       outb(0x80, 0x2B);
-       sprintf(statstr, "15-412! %08x %08x", smi_status(), counter);
-       outb(0x80, 0x3B);
-       strblit(statstr, 0, 0);
-       outb(0x80, 0x4B);
+       if (!fb || fb->curmode.text)
+       {
+               sprintf(statstr, "NetWatch! %08x %08x", smi_status(), counter);
+               strblit(statstr, 0, 0, 0);
+       }
        
-       /*
        eth_poll();
-       */
        
        if (inl(0x840) & 0x1000)
        {
@@ -62,35 +49,40 @@ void smi_entry(void)
                outl(0x840, 0x0100);
        }
 
-/*
+
        smi_poll();
        
        pci_bother_all();
- */
        outl(0xCF8, pcisave);
        outb(0x3D4, vgasave);
+       
+       /* Disable caching on SMRAM again, to prevent the user from whacking us. */
+       WRMSR(0x202, RDMSR(0x202) & ~(0xFFULL));
 }
 
-void timer_handler(smi_event_t ev)
-{
-        static unsigned int ticks = 0;
-
-        smi_disable_event(SMI_EVENT_FAST_TIMER);
-        smi_enable_event(SMI_EVENT_FAST_TIMER);
-
-        outb(0x80, (ticks++) & 0xFF);
+extern void timer_handler(smi_event_t ev);
+extern void kbc_handler(smi_event_t ev);
+extern void gbl_rls_handler(smi_event_t ev);
 
-        outlog();
-}
+void __firstrun_stub() {
 
+        /* Try really hard to shut up USB_LEGKEY. */
+        pci_write16(0, 31, 2, 0xC0, pci_read16(0, 31, 2, 0xC0));
+        pci_write16(0, 31, 2, 0xC0, 0);
+        pci_write16(0, 31, 4, 0xC0, pci_read16(0, 31, 4, 0xC0));
+        pci_write16(0, 31, 4, 0xC0, 0);
 
-void __firstrun_start() {
-       smi_disable();
-       outb(0x80, 0x41);
+        /* Turn on the SMIs we want */
+        smi_disable();
 
         smi_register_handler(SMI_EVENT_FAST_TIMER, timer_handler);
         smi_enable_event(SMI_EVENT_FAST_TIMER);
 
+        smi_register_handler(SMI_EVENT_DEVTRAP_KBC, kbc_handler);
+        smi_enable_event(SMI_EVENT_DEVTRAP_KBC);
+
+        smi_register_handler(SMI_EVENT_GBL_RLS, gbl_rls_handler);
+        smi_enable_event(SMI_EVENT_GBL_RLS);
+
         smi_enable();
 }
-