#include <klibc/string.h>
#include <lunaix/boot_generic.h>
#include <lunaix/mm/pmm.h>
#include <lunaix/mm/vmm.h>
#include <lunaix/spike.h>
#include <lunaix/kcmd.h>
#include <lunaix/sections.h>
#include <asm/mm_defs.h>

/**
 * @brief Reserve memory for kernel bootstrapping initialization
 *
 * @param bhctx
 */
void
boot_begin(struct boot_handoff* bhctx)
{
    bhctx->prepare(bhctx);

    boot_begin_arch_reserve(bhctx);
    
    // 将内核占据的页，包括前1MB，hhk_init 设为已占用
    size_t pg_count = leaf_count(to_kphysical(kernel_load_end));
    pmm_onhold_range(0, pg_count);

    size_t i;
    struct boot_mmapent* ent;
    for (i = 0; i < bhctx->mem.mmap_len; i++) {
        ent = &bhctx->mem.mmap[i];

        if (reserved_memregion(ent) || reclaimable_memregion(ent)) {
            unsigned int counts = leaf_count(ent->size);
            pmm_onhold_range(pfn(ent->start), counts);
        }
    }

    /* Reserve region for all loaded modules */
    for (i = 0; i < bhctx->mods.mods_num; i++) {
        struct boot_modent* mod = &bhctx->mods.entries[i];
        unsigned int counts = leaf_count(mod->end - mod->start);

        pmm_onhold_range(pfn(mod->start), counts);
    }
}

static void
__free_reclaimable()
{
    ptr_t start;
    pfn_t pgs;
    pte_t* ptep;

    start = reclaimable_start;
    pgs   = leaf_count(reclaimable_end - start);
    ptep  = mkptep_va(VMS_SELF, start);

    pmm_unhold_range(pfn(to_kphysical(start)), pgs);
    vmm_unset_ptes(ptep, pgs);
}

/**
 * @brief Release memory for kernel bootstrapping initialization
 *
 * @param bhctx
 */
void
boot_end(struct boot_handoff* bhctx)
{
    struct boot_mmapent* ent;
    for (size_t i = 0; i < bhctx->mem.mmap_len; i++) {
        ent = &bhctx->mem.mmap[i];

        if (reclaimable_memregion(ent)) {
            unsigned int counts = leaf_count(ent->size);
            pmm_unhold_range(pfn(ent->start), counts);
        }
    }

    bhctx->release(bhctx);

    boot_clean_arch_reserve(bhctx);

    __free_reclaimable();
}

void
boot_parse_cmdline(struct boot_handoff* bhctx) {
    if (!bhctx->kexec.len) {
        return;
    }

    kcmd_parse_cmdline(bhctx->kexec.cmdline);
}