[netwatch.git] / netwatch / smi.c

/* smi.c
 * First-run SMI C entry point
 * NetWatch system management mode administration console
 *
 * Copyright (c) 2008 Jacob Potter and Joshua Wise.  All rights reserved.
 * This program is free software; you can redistribute and/or modify it under
 * the terms found in the file LICENSE in the root of this source tree.
 *
 */

#include <io.h>
#include <smram.h>
#include <video_defines.h>
#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"

unsigned int counter = 0;
unsigned long pcisave = 0;
unsigned char vgasave = 0;

unsigned long lastentry = 0;
unsigned long lastlength = 0;

unsigned long rdtsc()
{
        unsigned long tsc;
        asm volatile ("mov %%cr4, %%eax;"
             "and $~2, %%eax;"
             "mov %%eax, %%cr4;"
             "rdtsc" : "=a"(tsc) : : "edx");
        return tsc;
}

void smi_entry(void)
{
        char statstr[512];
        unsigned long entrytime;

        /* Reenable caching on SMRAM. */
        WRMSR(0x202, (RDMSR(0x202) & ~(0xFFULL)) | 0x06ULL);

        entrytime = rdtsc();

        pcisave = inl(0xCF8);
        vgasave = inb(0x3D4);
        pci_unbother_all();
        
        serial_init();

        if (fb)
                fb->getvmode(fb->priv);

        counter++;
        if (!fb || fb->curmode.text)
        {
                int totcyc, pct;
                
                if (entrytime < lastentry)
                        totcyc = entrytime + (0xFFFFFFFFUL - lastentry) + 1;
                else
                        totcyc = entrytime - lastentry;
                if (totcyc == 0)
                        totcyc = 1;     /* argh */
                totcyc /= 1000;
                if (totcyc == 0)
                        totcyc = 1;
                
                pct = lastlength / totcyc;
                
                sprintf(statstr, "NetWatch! %08x %08x, %2d.%d%%", smi_status(), counter, pct/10, pct%10);
                strblit(statstr, 0, 0, 0);
        }
        
        eth_poll();
        
        smi_poll();
        
        pci_bother_all();
        outl(0xCF8, pcisave);
        outb(0x3D4, vgasave);
        
        lastentry = entrytime;
        entrytime = rdtsc();
        if (entrytime < lastentry)
                lastlength = entrytime + (0xFFFFFFFFUL - lastentry) + 1;
        else
                lastlength = entrytime - lastentry;
        
        /* 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);
extern void kbc_handler(smi_event_t ev);
extern void gbl_rls_handler(smi_event_t ev);

void __firstrun_stub()
{
        /* 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();
}