#include <types.h>
#include <config.h>

#include "multiboot.h"
#include "monitor.h"
#include "sys.h"
#include <task/idt.h>
#include <task/timer.h>
#include <task/task.h>
#include <mem/gdt.h>
#include <mem/paging.h>
#include <mem/mem.h>
#include <linker/elf.h>

#include <Object/Object.h>

/*	The kernel's main procedure. This function is called in loader_.asm.
	This function calls the initializer functions for all system parts.
	It then loads the modules the kernel was given by the bootloader. 
	This function never returns : once multitasking is started for good,
	the execution flow of this function is never returned to. */
void kmain(struct multiboot_info_t* mbd, int32_t magic) {
	ASSERT(magic == MULTIBOOT_BOOTLOADER_MAGIC);

	size_t totalRam = 0;
	uint32_t i;

	mem_placementAddr = ((size_t)&end & 0xFFFFF000) + 0x1000;
	mbd->cmdline += K_HIGHHALF_ADDR; mbd->mods_addr += K_HIGHHALF_ADDR;
	struct module_t *mods = (struct module_t*)mbd->mods_addr;
	for (i = 0; i < mbd->mods_count; i++) {
		mods[i].mod_start += K_HIGHHALF_ADDR;
		mods[i].mod_end += K_HIGHHALF_ADDR;
		mods[i].string += K_HIGHHALF_ADDR;
		if (mods[i].mod_end > mem_placementAddr)
			mem_placementAddr = (mods[i].mod_end & 0xFFFFF000) + 0x1000;
	}

	monitor_clear();

	monitor_write(" -> This is ");
	monitor_write(K_OS_NAME);
	monitor_write(" version ");
	monitor_write(K_OS_VER);
	monitor_write(" codename '");
	monitor_write(K_OS_CODENAME);
	monitor_write("', now starting up :\n");


	idt_init();

	totalRam = ((mbd->mem_upper + mbd->mem_lower) * 1024);
	paging_init(totalRam);
	_no_more_ksbrk = 1;
	gdt_init();
	paging_cleanup();

	timer_init(30);
	tasking_init();

	setup_object_system();
	
	for (i = 0; i < mbd->mods_count; i++) {
		monitor_write("\n * Load multiboot module '");
		monitor_write((char*)mods[i].string);
		if (elf_check((uint8_t*)mods[i].mod_start)) {
			monitor_write("' : Invalid ELF file");
		} else {
			struct process *pr = elf_exec((uint8_t*)mods[i].mod_start, PL_USER);
			if (pr == 0) {
				monitor_write("' : Error loading");
			} else {
				monitor_write("' : OK, pid="); monitor_writeDec(pr->pid);
			}
		}
	}

	monitor_write("\nUserland processes now RULE THE WORLD !\n\n");
	sti();
	schedule();
	PANIC("Should never happen. Something probably went wrong with multitasking.");
}