#include "sys.h"
#include <ui/vt.h>
#include <dev/vgatxt.h>

#include "mem/mem.h"

/*	These four functions are wrappers around ASM operations.
	These functions are used when comunicating with the system hardware. */

void outb(uint16_t port, uint8_t value) {
	asm volatile("outb %1, %0" : : "dN"(port), "a"(value));
}

void outw(uint16_t port, uint16_t value) {
	asm volatile("outw %1, %0" : : "dN"(port), "a"(value));
}

uint8_t inb(uint16_t port) {
	uint8_t ret;
	asm volatile("inb %1, %0" : "=a"(ret) : "dN"(port));
	return ret;
}

uint16_t inw(uint16_t port) {
	uint16_t ret;
	asm volatile("inw %1, %0" : "=a"(ret) : "dN"(port));
	return ret;
}

//////
void stack_trace(size_t bp) {
	uint32_t *stack = (uint32_t*)bp, i;
	for (i = 0; i < 5 && (size_t)stack > K_HIGHHALF_ADDR && (size_t)stack < (bp + 0x8000); i++) {
		ke_vt->writeStr("| "); ke_vt->writeHex((size_t)stack);
		ke_vt->writeStr("\tnext:"); ke_vt->writeHex(stack[0]); ke_vt->writeStr("\t\tret:"); ke_vt->writeHex(stack[1]);
		ke_vt->writeStr("\n");
		stack = (uint32_t*)stack[0];
	}
}

/*	For internal use only. Used by panic and panic_assert. */
static void panic_do(char* file, int line) {
	asm volatile("cli;");
	ke_vt->writeStr("\n File:\t\t"); ke_vt->writeStr(file); ke_vt->writeStr(":"); ke_vt->writeDec(line);
	ke_vt->writeStr("\nTrace:\n");
	size_t bp; asm volatile("mov %%ebp,%0" : "=r"(bp)); stack_trace(bp);
	ke_vt->writeStr("\n\t\tSystem halted -_-'");
	asm volatile("hlt");
}

/*	These functions stop the system, reporting an error message, because something bad happenned. */
void panic(char* message, char* file, int line) {
	ke_vt->fgcolor = TC_WHITE;
	ke_vt->bgcolor = TC_BLUE;
	ke_vt->outputTo(text_display);
	ke_vt->writeStr("\nPANIC:\t\t"); ke_vt->writeStr(message);
	panic_do(file, line);
}

void panic_assert(char* assertion, char* file, int line) {
	ke_vt->fgcolor = TC_WHITE;
	ke_vt->bgcolor = TC_RED;
	ke_vt->outputTo(text_display);
	ke_vt->writeStr("\nASSERT FAILED:\t"); ke_vt->writeStr(assertion);
	panic_do(file, line);
}

/* Global system mutex. See comments in sys.h. */

static uint32_t if_locks = 1;

void cli() {
	asm volatile("cli");
	if_locks++;
}

void sti() {
	if (if_locks > 0) if_locks--;
	if (if_locks == 0) asm volatile("sti");
}