/*
 * Copyright (c) 2020 AIIT XUOS Lab
 * XiUOS is licensed under Mulan PSL v2.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *        http://license.coscl.org.cn/MulanPSL2
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 */

/**
 * @file:    kalloc.c
 * @brief:   the general management of system memory
 * @version: 3.0
 * @author:  AIIT XUOS Lab
 * @date:    2023/8/28
 */
/*************************************************
File name: kalloc.c
Description: the general management of system memory
Others:
History:
1. Date: 2023-08-28
Author: AIIT XUOS Lab
Modification:
1. first version
*************************************************/
#include "kalloc.h"
#include "assert.h"

#include "actracer.h"
#include "buddy.h"

struct KBuddy kern_virtmem_buddy;
struct KBuddy user_phy_freemem_buddy;

extern uintptr_t kernel_data_end[];
bool module_phymem_init()
{
    uintptr_t kern_freemem_start = V2P(&kernel_data_end);
    uintptr_t kern_freemem_end = PHY_USER_FREEMEM_BASE;
    uintptr_t user_freemem_start = PHY_USER_FREEMEM_BASE;
    uintptr_t user_freemem_end = PHY_MEM_STOP;
    KBuddySysInit(&kern_virtmem_buddy, kern_freemem_start, kern_freemem_end);
    KBuddyInit(&user_phy_freemem_buddy, user_freemem_start, user_freemem_end);
    return true;
}

char* kalloc(size_t size)
{
    char* mem_alloc = KBuddyAlloc(&kern_virtmem_buddy, size);
    if (mem_alloc == NULL) {
        return NULL;
    }
    mem_alloc = P2V(mem_alloc);
    if ((uintptr_t)mem_alloc < KERN_MEM_BASE) {
        DEBUG("Error Alloc: %x by size: %d (Caused by double free)\n", mem_alloc, size);
    }
    memset(mem_alloc, 0, size);
    return mem_alloc;
}

bool kfree(char* vaddr)
{
    return KBuddyFree(&kern_virtmem_buddy, V2P_WO(vaddr));
}

char* raw_alloc(size_t size)
{
    char* mem_alloc = KBuddyAlloc(&user_phy_freemem_buddy, size);
    if (mem_alloc == NULL) {
        return NULL;
    }
    return mem_alloc;
}

bool raw_free(char* paddr)
{
    return KBuddyFree(&user_phy_freemem_buddy, paddr);
}

void show_phymem_info()
{
    KFreePagesInfo(&user_phy_freemem_buddy);
}