Joshua Wise's Git repositories - netwatch.git/blame_incremental

... / ...

Commit	Line	Data
	1	#include "console.h"
	2	#include "loader.h"
	3
	4	#include <minilib.h>
	5	#include <io.h>
	6	#include <smram.h>
	7	#include <multiboot.h>
	8	#include <smi.h>
	9	#include <pci.h>
	10
	11	#define INFO_SIGNATURE 0x5754454E
	12
	13	extern char _binary_realmode_bin_start[];
	14	extern int _binary_realmode_bin_size;
	15
	16
	17
	18	struct info_section
	19	{
	20	unsigned int signature;
	21	void (*firstrun)();
	22	};
	23
	24	void panic(const char *msg)
	25	{
	26	puts("PANIC: ");
	27	puts(msg);
	28	puts("\nSystem halted\n");
	29	while(1) { __asm__("hlt"); }
	30	}
	31
	32	void c_start(unsigned int magic, struct mb_info *mbinfo)
	33	{
	34	struct mod_info *mods = mbinfo->mods;
	35	smram_state_t old_smramc;
	36	struct info_section * info;
	37	int i;
	38
	39	void (realmode)() = (void ()()) 0x4000;
	40
	41	show_cursor();
	42	puts("NetWatch loader\n");
	43
	44	if (magic != MULTIBOOT_LOADER_MAGIC)
	45	panic("Bootloader was not multiboot compliant; cannot continue.");
	46
	47	for (i = 0; i < mbinfo->mod_cnt; i++)
	48	{
	49	puts("Module found:\n");
	50	puts(" Start: "); puthex((unsigned long) mods[i].mod_start); puts("\n");
	51	puts(" Size: "); puthex((unsigned long)mods[i].mod_end - (unsigned long)mods[i].mod_start); puts("\n");
	52	puts(" Name: "); puts(mods[i].mod_string); puts("\n");
	53	}
	54
	55	if (mbinfo->mod_cnt != 1)
	56	panic("Expected exactly one module; cannot continue.");
	57
	58	puts("Current USB state is: "); puthex(pci_read16(0, 31, 2, 0xC0)); puts(" "); puthex(pci_read16(0, 31, 4, 0xC0)); puts("\n");
	59	puts("Current SMI state is: "); puthex(inl(0x830)); puts("\n");
	60	puts("Current SMRAMC state is: "); puthex(pci_read8(0, 0, 0, 0x70)); puts("\n");
	61
	62	smi_disable();
	63
	64	/* Try really hard to shut up USB_LEGKEY. */
	65	pci_write16(0, 31, 2, 0xC0, pci_read16(0, 31, 2, 0xC0));
	66	pci_write16(0, 31, 2, 0xC0, 0);
	67	pci_write16(0, 31, 4, 0xC0, pci_read16(0, 31, 4, 0xC0));
	68	pci_write16(0, 31, 4, 0xC0, 0);
	69
	70	/* Open the SMRAM aperture and load our ELF. */
	71	old_smramc = smram_save_state();
	72
	73	if (smram_aseg_set_state(SMRAM_ASEG_OPEN) != 0)
	74	panic("Opening SMRAM failed; cannot load ELF.");
	75
	76	load_elf(mods[0].mod_start, (unsigned long)mods[0].mod_end - (unsigned long)mods[0].mod_start);
	77
	78	info = (struct info_section *)0x10000;
	79	if (info->signature != INFO_SIGNATURE)
	80	{
	81	smram_restore_state(old_smramc); /* Restore so that video ram is touchable again. */
	82	panic("Info section signature mismatch.");
	83	}
	84
	85	info->firstrun();
	86	smram_restore_state(old_smramc);
	87
	88	puts("Waiting for a bit before returning to real mode...");
	89	for (i=0; i<0x500000; i++)
	90	{
	91	if ((i % 0x100000) == 0)
	92	puts(".");
	93	inb(0x80);
	94	}
	95	puts("\n");
	96
	97	puts("Now returning to real mode.\n");
	98	memcpy((void *)0x4000, _binary_realmode_bin_start, (int)&_binary_realmode_bin_size);
	99	realmode(); // goodbye!
	100	}