#include "process.h"
#include "arch/x86_64/hal/obj.h"
#include "hal/obj.h"
#include "hal/process.h"
#include "hal/smp.h"
#include "lib/string.h"
#include "init.h"
#include "smp.h"
#include "paging.h"
#include "jove.h"

PAGEALIGN uint8_t s_init_iopb[0x2000];
PAGEALIGN char s_init_pcb_bytes[0x1000] = { 0 };
static pcb_t *s_init_pcb = (pcb_t*)&s_init_pcb_bytes;

jove_owned_obj_t*
process_current(void)
{
    return obj_take(((processor_t*)processor_current())->pcb_obj);
}

pcb_t*
pcb_from_obj(jove_owned_obj_t *obj)
{
    if(obj_gettype(obj) != JOVE_OBJ_PROCESS) {
        reterr(EBADOBJ);
    }
    return obj_deref(obj);
}

pid_t
pcb_pid(pcb_t *pcb)
{
    return pcb->pid;
}

jove_owned_obj_t*
pcb_rootdir(pcb_t *pcb)
{
    return pcb->rootdir;
}

void
processing_setup(void (*process_main))
{
    memset(s_init_iopb, 0xFF, 0x2000);
    uintptr_t ksp = ((uintptr_t)&s_init_pcb_bytes) + 0xFF8;

    s_init_pcb->pid = 1;
    s_init_pcb->ksp = (void*)ksp;
    s_init_pcb->pml4 = paging_get_pml4();
    s_init_pcb->rootdir = _rootdir_obj;

    jove_owned_obj_t *iopb_obj = objdir_getlocal(_rootdir_obj, ROOTDIR_IOPB_I);
    obj_settype(iopb_obj, JOVE_OBJ_IOPB);
    obj_setvalue(iopb_obj, (uintptr_t)s_init_iopb);
    s_init_pcb->iopb = iopb_obj;

    processor_t *proc = processor_current();
    proc->tss.rsp[0] = (uintptr_t)s_init_pcb->ksp;
    memcpy(proc->iopb, s_init_iopb, 0x2000);

    {
        jove_owned_obj_t *process_obj = objdir_getlocal(_rootdir_obj, ROOTDIR_PCB_I);
        obj_settype(process_obj, JOVE_OBJ_PROCESS);
        obj_setvalue(process_obj, (uintptr_t)s_init_pcb);

        proc->pcb_obj = obj_release(process_obj);
        proc->pcb = s_init_pcb;
    }

    __asm__ volatile("\
            movq %0, %%rsp; \
            movq $0, %%rbp; \
            jmpq %1"::
            "r"(s_init_pcb->ksp), "r"(process_main));
}