#ifndef _KERNEL_MEM_VMM_H
#define _KERNEL_MEM_VMM_H

#include <libkern/libkern.h>
#include <mem/bits/mmu.h>
#include <mem/bits/swap.h>
#include <mem/bits/vm.h>
#include <mem/boot.h>
#include <mem/pmm.h>
#include <mem/vm_address_space.h>
#include <platform/generic/vmm/consts.h>
#include <platform/generic/vmm/mmu.h>

#define IS_KERNEL_VADDR(vaddr) (vaddr >= KERNEL_BASE)
#define IS_USER_VADDR(vaddr) (vaddr <= USER_HIGH)

#define VMM_INVALID_PADDR ((uintptr_t)(-1))
#define VMM_INVALID_VADDR ((uintptr_t)(-1))

struct memzone;
struct vm_ops {
    int (*load_page_content)(struct memzone* zone, uintptr_t vaddr);
    int (*swap_page_mode)(struct memzone* zone, uintptr_t vaddr);
    int (*restore_swapped_page)(struct memzone* zone, uintptr_t vaddr);
};
typedef struct vm_ops vm_ops_t;

/**
 * PUBLIC FUNCTIONS
 */

struct dynamic_array;

int vmm_setup(boot_args_t* boot_args);
int vmm_setup_secondary_cpu();

int vmm_free_address_space(vm_address_space_t* vm_aspace);

int vmm_alloc_page(uintptr_t vaddr, mmu_flags_t mmu_flags);
int vmm_tune_page(uintptr_t vaddr, mmu_flags_t mmu_flags);
int vmm_tune_pages(uintptr_t vaddr, size_t length, mmu_flags_t mmu_flags);

int vmm_map_page(uintptr_t vaddr, uintptr_t paddr, mmu_flags_t mmu_flags);
int vmm_map_pages(uintptr_t vaddr, uintptr_t paddr, size_t n_pages, mmu_flags_t mmu_flags);
int vmm_unmap_page(uintptr_t vaddr);
int vmm_unmap_pages(uintptr_t vaddr, size_t n_pages);
int vmm_swap_page(ptable_entity_t* page_desc, struct memzone* zone, uintptr_t vaddr);

int vmm_map_page_locked(uintptr_t vaddr, uintptr_t paddr, mmu_flags_t mmu_flags);
int vmm_map_pages_locked(uintptr_t vaddr, uintptr_t paddr, size_t n_pages, mmu_flags_t mmu_flags);
int vmm_unmap_page_locked(uintptr_t vaddr);
int vmm_unmap_pages_locked(uintptr_t vaddr, size_t n_pages);

vm_address_space_t* vmm_new_address_space();
vm_address_space_t* vmm_new_forked_address_space();

bool vmm_is_copy_on_write(uintptr_t vaddr);

void vmm_ensure_writing_to_active_address_space(uintptr_t dest_vaddr, size_t length);
void vmm_ensure_reading_from_active_address_space(uintptr_t dest_vaddr, size_t length);
void vmm_copy_to_address_space(vm_address_space_t* vm_aspace, void* src, uintptr_t dest_vaddr, size_t length);

vm_address_space_t* vmm_get_active_address_space();
vm_address_space_t* vmm_get_kernel_address_space();

int vmm_switch_address_space_locked(vm_address_space_t* vm_aspace);
int vmm_switch_address_space(vm_address_space_t* vm_aspace);

int vmm_page_fault_handler(arch_pf_info_t info, uintptr_t vaddr);

uintptr_t vmm_convert_kernel_vaddr_to_paddr(uintptr_t vaddr);

#endif // _KERNEL_MEM_VMM_H