#include "../common/sys/kdef.h"
#include "../common/sys/type.h"
#include "mem.h"
#include "../common/lib/alloc/salloc/salloc.h"

#define KERNEL_HEAP_SIZE 4096

char kernel_heap[KERNEL_HEAP_SIZE];

void SAPI hal_write_pt_entry(void *const base, uint64_t const p_addr, uint64_t const attr)
{
    if (base == NULL)
        return;
    uint64_t entry = (p_addr & 0xFFFFFFFFFF000) | attr;
    ((uint8_t *) base)[0] = (uint8_t) (entry & 0xFF);
    ((uint8_t *) base)[1] = (uint8_t) ((entry >> 8) & 0xFF);
    ((uint8_t *) base)[2] = (uint8_t) ((entry >> 16) & 0xFF);
    ((uint8_t *) base)[3] = (uint8_t) ((entry >> 24) & 0xFF);
    ((uint8_t *) base)[4] = (uint8_t) ((entry >> 32) & 0xFF);
    ((uint8_t *) base)[5] = (uint8_t) ((entry >> 40) & 0xFF);
    ((uint8_t *) base)[6] = (uint8_t) ((entry >> 48) & 0xFF);
    ((uint8_t *) base)[7] = (uint8_t) ((entry >> 56) & 0xFF);
    return;
}

void SAPI hal_write_pd_entry(void *const base, uint64_t const pt_addr, uint64_t const attr)
{
    if (base == NULL)
        return;
    uint64_t entry = (pt_addr & 0xFFFFFFFFFF000) | attr;
    ((uint8_t *) base)[0] = (uint8_t) (entry & 0xFF);
    ((uint8_t *) base)[1] = (uint8_t) ((entry >> 8) & 0xFF);
    ((uint8_t *) base)[2] = (uint8_t) ((entry >> 16) & 0xFF);
    ((uint8_t *) base)[3] = (uint8_t) ((entry >> 24) & 0xFF);
    ((uint8_t *) base)[4] = (uint8_t) ((entry >> 32) & 0xFF);
    ((uint8_t *) base)[5] = (uint8_t) ((entry >> 40) & 0xFF);
    ((uint8_t *) base)[6] = (uint8_t) ((entry >> 48) & 0xFF);
    ((uint8_t *) base)[7] = (uint8_t) ((entry >> 56) & 0xFF);
    return;
}

void SAPI hal_write_pdpt_entry(void *const base, uint64_t const pd_addr, uint64_t const attr)
{
    if (base == NULL)
        return;
    uint64_t entry = (pd_addr & 0xFFFFFFFFFF000) | attr;
    ((uint8_t *) base)[0] = (uint8_t) (entry & 0xFF);
    ((uint8_t *) base)[1] = (uint8_t) ((entry >> 8) & 0xFF);
    ((uint8_t *) base)[2] = (uint8_t) ((entry >> 16) & 0xFF);
    ((uint8_t *) base)[3] = (uint8_t) ((entry >> 24) & 0xFF);
    ((uint8_t *) base)[4] = (uint8_t) ((entry >> 32) & 0xFF);
    ((uint8_t *) base)[5] = (uint8_t) ((entry >> 40) & 0xFF);
    ((uint8_t *) base)[6] = (uint8_t) ((entry >> 48) & 0xFF);
    ((uint8_t *) base)[7] = (uint8_t) ((entry >> 56) & 0xFF);
    return;
}

void SAPI hal_write_pml4_entry(void *const base, uint64_t const pdpt_addr, uint64_t const attr)
{
    if (base == NULL)
        return;
    uint64_t const entry = (pdpt_addr & 0xFFFFFFFFFF000) | attr;
    ((uint8_t *) base)[0] = (uint8_t) (entry & 0xFF);
    ((uint8_t *) base)[1] = (uint8_t) ((entry >> 8) & 0xFF);
    ((uint8_t *) base)[2] = (uint8_t) ((entry >> 16) & 0xFF);
    ((uint8_t *) base)[3] = (uint8_t) ((entry >> 24) & 0xFF);
    ((uint8_t *) base)[4] = (uint8_t) ((entry >> 32) & 0xFF);
    ((uint8_t *) base)[5] = (uint8_t) ((entry >> 40) & 0xFF);
    ((uint8_t *) base)[6] = (uint8_t) ((entry >> 48) & 0xFF);
    ((uint8_t *) base)[7] = (uint8_t) ((entry >> 56) & 0xFF);
    return;
}

void SAPI hal_write_segment_descriptor(void *const gdt, uint32_t const base, uint32_t const limit,
                                              uint64_t const attr)
{
    if (gdt == NULL)
        return;
    uint64_t const seg_desc = (((uint64_t) base & 0xFFFF) << 16) | ((((uint64_t) base >> 16) & 0xFF) << 32) |
                              ((((uint64_t) base >> 24) & 0xFF) << 56) | ((uint64_t) limit & 0xFFFF) |
                              ((((uint64_t) limit >> 16) & 0xF) << 48) | attr;
    ((uint8_t *) gdt)[0] = (uint8_t) (seg_desc & 0xFF);
    ((uint8_t *) gdt)[1] = (uint8_t) ((seg_desc >> 8) & 0xFF);
    ((uint8_t *) gdt)[2] = (uint8_t) ((seg_desc >> 16) & 0xFF);
    ((uint8_t *) gdt)[3] = (uint8_t) ((seg_desc >> 24) & 0xFF);
    ((uint8_t *) gdt)[4] = (uint8_t) ((seg_desc >> 32) & 0xFF);
    ((uint8_t *) gdt)[5] = (uint8_t) ((seg_desc >> 40) & 0xFF);
    ((uint8_t *) gdt)[6] = (uint8_t) ((seg_desc >> 48) & 0xFF);
    ((uint8_t *) gdt)[7] = (uint8_t) ((seg_desc >> 56) & 0xFF);
    return;
}

void* SAPI hal_alloc(uint32_t size)
{
    return salloc(kernel_heap,size);
}

void SAPI hal_free(void *ptr)
{
    return sfree(kernel_heap, ptr);
}

void SAPI hal_alloc_init()
{
    salloc_init(kernel_heap, KERNEL_HEAP_SIZE);
    return;
}