Enable caching while in SMM.

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

diff --git a/aseg-paging/smi.c b/aseg-paging/smi.c
index bd29f64f81713da06febd76e5b8cbc028ae368f1..d519341a9370446fdbe8e1e65e56419271f4abb4 100644 (file)
--- a/aseg-paging/smi.c
+++ b/aseg-paging/smi.c
@@ -5,6 +5,9 @@
 #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"
 
@@ -16,15 +19,25 @@ void smi_entry(void)
 {
        char statstr[512];
 
+       /* Reenable caching on SMRAM. */
+       WRMSR(0x202, (RDMSR(0x202) & ~(0xFFULL)) | 0x06ULL);
+
        pcisave = inl(0xCF8);
        vgasave = inb(0x3D4);
        pci_unbother_all();
+       
+       serial_init();
+       
+       if (fb)
+               fb->getvmode(fb->priv);
 
        counter++;
-       sprintf(statstr, "NetWatch! %08x %08x", smi_status(), counter);
-       strblit(statstr, 0, 0);
+       if (!fb || fb->curmode.text)
+       {
+               sprintf(statstr, "NetWatch! %08x %08x", smi_status(), counter);
+               strblit(statstr, 0, 0, 0);
+       }
        
-       serial_init();
        eth_poll();
        
        if (inl(0x840) & 0x1000)
@@ -42,6 +55,9 @@ void smi_entry(void)
        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));
 }
 
 extern void timer_handler(smi_event_t ev);