#include "smp.h"
#include "proc/init.h"
#include "config.h"
#include "jove.h"
#include "paging.h"
#include "msr.h"
#include "hal/obj.h"
#include "proc/process.h"
#include "lib/printf.h"
#include "lib/string.h"
#include "sys/objdir.h"

processor_t s_processors[SMP_PROCESSORS_MAX];

PAGEALIGN jove_objdir_t s_processor_objdir;

extern void
gdt_setup(processor_t *processor);

extern void
idt_init(void);

extern void
idt_setup(processor_t *processor);

static void
init_processor(processor_t *processor, jove_owned_obj_t *obj)
{
    gdt_setup(processor);
    idt_setup(processor);

    msr_enable_sce();
    msr_set_gs_base((uintptr_t)processor);

    obj_settype(obj, JOVE_OBJ_PROCESSOR);
    obj_setvalue(obj, (uintptr_t)processor);
}

void
smp_init_bsp(void)
{
    memset(s_processors, 0, sizeof(s_processors));
    memset(&s_processor_objdir, 0, 0x1000);

    processor_t *bsp = &s_processors[0];
    {
        jove_owned_obj_t *processor_objdir_obj = objdir_getlocal(_rootdir_obj, INIT_ROOTDIR_PROCESSORS_I);
        obj_settype(processor_objdir_obj, JOVE_OBJ_OBJDIR);
        obj_setvalue(processor_objdir_obj, (uintptr_t)&s_processor_objdir);

        jove_owned_obj_t *bsp_obj = objdir_getlocal(processor_objdir_obj, 0);
        init_processor(bsp, bsp_obj);

        obj_release(bsp_obj);
        obj_release(processor_objdir_obj);
    }

    idt_init();
    paging_pml4_init();
}

inline processor_t *processor_current(void)
{
    return (processor_t*)msr_get_gs_base();
}

inline jove_owned_obj_t *processor_getpcb(processor_t *processor)
{
    return processor->pcb_obj;
}

void 
processor_setpcb(processor_t *processor, jove_owned_obj_t *pcb_obj)
{
    pcb_t *pcb = obj_deref(pcb_obj);
    if(processor->pcb) {
        obj_release(processor_getpcb(processor));
    }
    processor->pcb_obj = pcb_obj;
    processor->pcb = pcb;
}