#include <u.h>
#include <a.out.h>
#include "fns.h"
#include "mem.h"

void
memset(void *p, int v, int n)
{
	uchar *d = p;
	while(n > 0){
		*d++ = v;
		n--;
	}
}

void
memmove(void *dst, void *src, int n)
{
	uchar *d = dst;
	uchar *s = src;

	if(d < s){
		while(n > 0){
			*d++ = *s++;
			n--;
		}
	} if(d > s){
		s += n;
		d += n;
		while(n > 0){
			*--d = *--s;
			n--;
		}
	}
}

int
memcmp(void *src, void *dst, int n)
{
	uchar *d = dst;
	uchar *s = src;
	int r = 0;
	while((n > 0) && (r = (*d++ - *s++)) == 0)
		n--;
	return r;
}

int
strlen(char *s)
{
	char *p = s;
	while(*p)
		p++;
	return p - s;
}

char*
strchr(char *s, int c)
{
	for(; *s; s++)
		if(*s == c)
			return s;
	return 0;
}

char*
strrchr(char *s, int c)
{
	char *r;
	r = 0;
	while(s = strchr(s, c))
		r = s++;
	return r;
}

void
print(char *s)
{
	while(*s)
		putc(*s++);
}

int
readn(void *f, void *data, int len)
{
	uchar *p, *e;

	putc(' ');
	p = data;
	e = p + len;
	while(p < e){
		if(((ulong)p & 0xF000) == 0){
			putc('\b');
			putc(hex[((ulong)p>>16)&0xF]);
		}
		if((len = read(f, p, e - p)) <= 0)
			break;
		p += len;
	}
	putc('\b');
	return p - (uchar*)data;
}

static int
readline(void *f, char buf[64])
{
	static char white[] = "\t ";
	char *p;

	p = buf;
	do{
		if(!f)
			putc('>');
		while(p < buf + 64-1){
			if(!f){
				putc(*p = getc());
				if(*p == '\r')
					putc('\n');
				else if(*p == '\b' && p > buf){
					p--;
					continue;
				}
			}else if(read(f, p, 1) <= 0)
				return 0;
			if(p == buf && strchr(white, *p))
				continue;
			if(strchr(crnl, *p))
				break;
			p++;
		}
		while(p > buf && strchr(white, p[-1]))
			p--;
	}while(p == buf);
	*p = 0;
	return p - buf;
}

static int
timeout(int ms)
{
	while(ms > 0){
		if(gotc())
			return 1;
		usleep(100000);
		ms -= 100;
	}
	return 0;
}

#define BOOTLINE	((char*)CONFADDR)
#define BOOTLINELEN	64
#define BOOTARGS	((char*)(CONFADDR+BOOTLINELEN))
#define	BOOTARGSLEN	(4096-0x200-BOOTLINELEN)

char *confend;

static void apmconf(int);
static void e820conf(void);

char*
configure(void *f, char *path)
{
	char line[64], *kern, *p;
	int inblock, n;
Clear:
	kern = 0;
	inblock = 0;

	memset(BOOTLINE, 0, BOOTLINELEN);

	confend = BOOTARGS;
	memset(confend, 0, BOOTARGSLEN);

	e820conf();
Loop:
	while((n = readline(f, line)) > 0){
		if(*line == 0 || strchr("#;=", *line))
			continue;
		if(*line == '['){
			inblock = memcmp("[common]", line, 8);
			continue;
		}
		if(!memcmp("clear", line, 6)){
			print("ok\r\n");
			goto Clear;
		}
		if(!memcmp("boot", line, 5))
			break;
		if(inblock || !strrchr(line, '='))
			continue;
		if(!memcmp("bootfile=", line, 9))
			memmove(kern = path, line+9, 1 + n-9);
		if(!memcmp("apm", line, 3) && line[4]=='='){
			apmconf('0' - line[3]);
			continue;
		}
		memmove(confend, line, n); confend += n;
		*confend++ = '\n';
		print(line); print(crnl);
	}
	*confend = 0;

	if(f){
		close(f);
		f = 0;

		if(kern && timeout(1000))
			goto Loop;
	}

	if(!kern){
		print("no bootfile\r\n");
		goto Loop;
	}
	if(p = strrchr(kern, '!'))
		kern = p+1;

	return kern;
}


static void
hexfmt(char *s, int i, ulong a)
{
	s += i;
	while(i > 0){
		*--s = hex[a&15];
		a >>= 4;
		i--;
	}
}

static void
addconfx(char *s, int w, ulong v)
{
	int n;

	n = strlen(s);
	memmove(confend, s, n);
	hexfmt(confend+n, w, v);
	confend += n+w;
	*confend = 0;
}

void apm(int id);

static void
apmconf(int id)
{
	uchar *a;
	char *s;

	a = (uchar*)CONFADDR;
	memset(a, 0, 20);

	apm(id);
	if(memcmp(a, "APM", 4))
		return;

	s = confend;

	addconfx("apm", 1, id);
	addconfx("=ax=", 4, *((ushort*)(a+4)));
	addconfx(" ebx=", 8, *((ulong*)(a+12)));
	addconfx(" cx=", 4, *((ushort*)(a+6)));
	addconfx(" dx=", 4, *((ushort*)(a+8)));
	addconfx(" di=", 4, *((ushort*)(a+10)));
	addconfx(" esi=", 8, *((ulong*)(a+16)));

	print(s); print(crnl);

	*confend++ = '\n';
}

ulong e820(ulong bx, void *p);

static void
e820conf(void)
{
	struct {
		uvlong	base;
		uvlong	len;
		ulong	typ;
		ulong	ext;
	} e;
	uvlong v;
	ulong bx;
	char *s;

	memset(&e, 0, sizeof(e));
	if((bx = e820(0, &e)) == 0)
		return;

	s = confend;
	memmove(confend, "e820=", 5);
	confend += 5;

	do{
		if(e.typ == 1 && (e.ext & 1) == 0 && e.len){
			v = e.base;
			addconfx("", 8, v>>32);
			addconfx("", 8, v&0xffffffff);
			v = e.base + e.len;
			addconfx(" ", 8, v>>32);
			addconfx("", 8, v&0xffffffff);
			*confend++ = ' ';
		}

		memset(&e, 0, sizeof(e));
	} while(bx = e820(bx, &e));

	*confend = 0;
	print(s); print(crnl);

	*confend++ = '\n';
}

static ushort
beswab(ushort s)
{
	uchar *p;

	p = (uchar*)&s;
	return (p[0]<<8) | p[1];
}

static ulong
beswal(ulong l)
{
	uchar *p;

	p = (uchar*)&l;
	return (p[0]<<24) | (p[1]<<16) | (p[2]<<8) | p[3];
}

void a20(void);

char*
bootkern(void *f)
{
	uchar *e, *d, *t;
	ulong n;
	Exec ex;

	a20();

	if(readn(f, &ex, sizeof(ex)) != sizeof(ex))
		return "bad header";
	if(beswal(ex.magic) != I_MAGIC)
		return "bad magic";

	e = (uchar*)(beswal(ex.entry) & ~0xF0000000UL);
	t = e;
	n = beswal(ex.text);

	if(readn(f, t, n) != n)
		goto Error;
	d = (uchar*)PGROUND((ulong)t + n);
	n = beswal(ex.data);

	if(readn(f, d, n) != n)
		goto Error;
	close(f);
	unload();

	print("boot");
	print(crnl);

	jump(e);

Error:		
	return "i/o error";
}