Hit paging to clean it up a bit. Add a reboot button.

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

#include <io.h>
#include <smram.h>
#include <video_defines.h>
#include <minilib.h>
#include <smi.h>
#include <pci-bother.h>
#include <serial.h>
#include <output.h>
#include "traps.h"
#include "../net/net.h"
#include "vga-overlay.h"

extern void smi_init();
#include "vm_flags.h"

extern void smi_entry();
void set_cr0(unsigned int);
void ps_switch_stack (void (*call)(), int stack);

#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

#define MAP_FLAGS	(PTE_PRESENT | PTE_READ_WRITE)

static int initialized = 0;
static int paging_enb = 0;
static unsigned long *pd;

extern int _bss, _bssend, _end;

unsigned long v2p(void *virt)
{
	unsigned long _virt = (unsigned long)virt;
	
	if (!paging_enb)
		return _virt;
	
	if (_virt >= 0xA0000 && _virt < 0xC0000)
		return _virt;
	if (_virt >= 0x200000 && _virt < 0x300000)
		return _virt - 0x200000 + /* XXX */ 0x1FF82000;
	outputf("WARNING: v2p(%08x)", _virt);
	return 0xFFFFFFFF;
}

void *p2v(unsigned long phys)
{
	if (!paging_enb)
		return (void*)phys;
	
	if (phys >= 0xA0000 && phys < 0xC0000)
		return (void*)phys;
	if (phys >= 0x1FF80000 && phys < 0x20000000)
		return (void*)(phys - 0x1FF82000 + 0x200000);
	outputf("WARNING: p2v(%08x)", phys);
	return (void *)0xFFFFFFFF;
}


inline int pt_addmap(unsigned long *pd, unsigned long vaddr, unsigned long paddr)
{
	unsigned long pde = ((unsigned long *)p2v((unsigned long)pd))[PDE_FOR(vaddr)];
	unsigned long *pt;
	
	if (!(pde & PTE_PRESENT))
		return -1;
	
	pt = (unsigned long *)p2v(ADDR_12_MASK(pde));
	pt[PTE_FOR(vaddr)] = paddr | PTE_PRESENT | PTE_READ_WRITE;
	
	return 0;
}

static void * pt_setup(int tseg_start, int tseg_size) {
	int i;

	/* The page directory and page table live at TSEG and TSEG + 0x1000,
	 * respectively. */
	unsigned long *pagedirectory = (unsigned long *) tseg_start;
	unsigned long *pagetable = (unsigned long *) (tseg_start + 0x1000);

	/* Clear out the page directory except for one entry pointing to the
	 * page table, and clear the page table entirely. */
	pagedirectory[0] = (tseg_start + 0x1000) | PTE_PRESENT | PTE_READ_WRITE;
	for (i = 1; i < 1024; i++)
		pagedirectory[i] = 0;

	for (i = 0; i < 1024; i++)
		pagetable[i] = 0;

	/* Map 0x0A0000:0x0BFFFF to itself. */
	for (i = 0; i < 32; i++)
		pt_addmap(pagedirectory, 0xA0000 + i * 0x1000, 0xA0000 + i * 0x1000);

	/* Map 0x200000:0x300000 to TSEG */
	for (i = 0; i < 256; i++)
		pt_addmap(pagedirectory, 0x200000 + i * 0x1000, tseg_start + 0x2000 + i * 0x1000);

	/* Map 0x300000:0x400000 to 0x200000, so we can copy our code out of
	 * RAM the first time around */
	for (i = 0; i < 256; i++)
		pt_addmap(pagedirectory, 0x300000 + i * 0x1000, 0x200000 + i * 0x1000);

	return pagedirectory;
}

void c_entry(void)
{
	paging_enb = 0;

	outb(0x80, 0x01);	
	if (!initialized)
		pd = pt_setup(0x1FF80000, 0x80000);
	outb(0x80, 0x02);
		
	/* Enable paging. */
	set_cr3((unsigned long)pd);
	set_cr0(get_cr0() | CR0_PG);
	outb(0x80, 0x03);
	paging_enb = 1;

	/* If this is the first goround, copy in data. */
	if (!initialized)
	{
		unsigned char *p;
		
		outb(0x80, 0x04);
		for (p = (void *)0x200000; (void *)p < (void *)&_bss; p++)
			*p = *(p + 0x100000);
		for (p = (void *)&_bss; (void *)p < (void *)&_bssend; p++)
			*p = 0;
		outb(0x80, 0x05);
		
		/* Only now is it safe to call other functions. */
		serial_init();
		dolog("Evacuation to TSEG complete.");
	}
	
	outb(0x80, 0x06);

	traps_install();
	
	outb(0x80, 0x07);

	if (!initialized)
	{
		smi_init();	/* Run the firstrun. */
		outb(0x80, 0x08);
		
		initialized = 1;
	}
	
	outb(0x80, 0x09);
	ps_switch_stack(smi_entry, 0x270000);
	outb(0x80, 0xFA);
}