#include "gdt.h"
#include <stdlib_common.h>
#include <core/monitor.h>

extern "C" void gdt_flush(uint32_t);	//ASM (imported from idt_.asm)
extern "C" void tss_flush();

#define GDT_ENTRIES 6	// The contents of each entry is defined in gdt_init.

static tss_entry tss_entry;
static gdt_entry gdt_entries[GDT_ENTRIES];
static gdt_ptr gdt_ptr;

/*	This function is called by the task_switch function on each context switch.
	It updates the TSS so that an interrupt will be handled on the correct kernel stack
	(one kernel stack is allocated to each thread). */
void gdt_setKernelStack(uint32_t esp0) {
	tss_entry.esp0 = esp0;
}

/*	For internal use only. Writes one entry of the GDT with given parameters. */
static void gdt_setGate(int num, uint32_t base, uint32_t limit, uint8_t access, uint8_t gran) {
	gdt_entries[num].base_low = (base & 0xFFFF);
	gdt_entries[num].base_middle = (base >> 16) & 0xFF;
	gdt_entries[num].base_high = (base >> 24) & 0xFF;

	gdt_entries[num].limit_low = (limit & 0xFFFF);
	gdt_entries[num].granularity = (limit >> 16) & 0x0F;
	gdt_entries[num].granularity |= gran & 0xF0;
	gdt_entries[num].access = access;
}

/*	For internal use only. Writes one entry of the GDT, that entry being a pointer to the TSS. */
static void gdt_writeTss(int num, uint32_t ss0, uint32_t esp0) {
	uint32_t base = (uint32_t)&tss_entry;
	uint32_t limit = base + sizeof(tss_entry);

	gdt_setGate(num, base, limit, 0xE9, 0);

	memset(&tss_entry, 0, sizeof(tss_entry));

	tss_entry.ss0 = ss0; tss_entry.esp0 = esp0;

	tss_entry.cs = 0x0b;
	tss_entry.ss = tss_entry.ds = tss_entry.es = tss_entry.fs = tss_entry.gs = 0x13;
}

/*	Write data to the GDT and enable it. */
void gdt_init() {
	gdt_ptr.limit = (sizeof(gdt_entry) * GDT_ENTRIES) - 1;
	gdt_ptr.base = (uint32_t)&gdt_entries;

	gdt_setGate(0, 0, 0, 0, 0);					//Null segment
	gdt_setGate(1, 0, 0xFFFFFFFF, 0x9A, 0xCF);	//Kernel code segment	0x08
	gdt_setGate(2, 0, 0xFFFFFFFF, 0x92, 0xCF);	//Kernel data segment	0x10
	gdt_setGate(3, 0, 0xFFFFFFFF, 0xFA, 0xCF);	//User code segment		0x18
	gdt_setGate(4, 0, 0xFFFFFFFF, 0xF2, 0xCF);	//User data segment		0x20
	gdt_writeTss(5, 0x10, 0);

	gdt_flush((uint32_t)&gdt_ptr);
	tss_flush();

	monitor_write("[GDT] ");
}