1 files changed, 64 insertions, 0 deletions
diff --git a/src/kernel/mem/gdt.cpp b/src/kernel/mem/gdt.cpp
new file mode 100644
index 0000000..49fbd82
--- /dev/null
+++ b/src/kernel/mem/gdt.cpp
@@ -0,0 +1,64 @@
+#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] ");
+}