#include <comus/memory.h>
#include <comus/asm.h>
#include <comus/mboot.h>
#include <comus/efi.h>
#include <lib.h>

#include "memory.h"
#include "paging.h"
#include "virtalloc.h"
#include "physalloc.h"

mem_ctx_t kernel_mem_ctx;
struct mem_ctx_s _kernel_mem_ctx;
extern volatile char kernel_pml4[];
extern struct virt_ctx kernel_virt_ctx;

void *kmapaddr(void *phys, void *virt, size_t len, unsigned int flags)
{
	return mem_mapaddr(kernel_mem_ctx, phys, virt, len, flags);
}

void kunmapaddr(void *virt)
{
	mem_unmapaddr(kernel_mem_ctx, virt);
}

void *kalloc_page(void)
{
	return mem_alloc_page(kernel_mem_ctx);
}

void *kalloc_pages(size_t count)
{
	return mem_alloc_pages(kernel_mem_ctx, count);
}

void kfree_pages(void *ptr)
{
	mem_free_pages(kernel_mem_ctx, ptr);
}

int kload_page(void *virt)
{
	return mem_load_page(kernel_mem_ctx, virt);
}

mem_ctx_t mem_ctx_alloc(void)
{
	panic("not yet implemented");
}

mem_ctx_t mem_ctx_clone(mem_ctx_t ctx, bool cow)
{
	(void)ctx;
	(void)cow;

	panic("not yet implemented");
}

void mem_ctx_free(mem_ctx_t ctx)
{
	(void)ctx;

	panic("not yet implemented");
}

void memory_init(void)
{
	struct memory_map mmap;
	if (mboot_get_mmap(&mmap))
		if (efi_get_mmap(&mmap))
			panic("failed to load memory map");

	kernel_mem_ctx = &_kernel_mem_ctx;
	kernel_mem_ctx->pml4 = kernel_pml4;
	kernel_mem_ctx->virtctx = &kernel_virt_ctx;

	cli();
	paging_init();
	virtaddr_init(kernel_mem_ctx->virtctx);
	physalloc_init(&mmap);
	sti();

	// identiy map EFI functions
	for (size_t i = 0; i < mmap.entry_count; i++) {
		struct memory_segment *seg = &mmap.entries[i];
		if (seg->type != SEG_TYPE_EFI)
			continue;
		kmapaddr((void *)seg->addr, (void *)seg->addr, seg->len, F_WRITEABLE);
	}
}