forked from PoliEcho-holding-org/kern
73 lines
1.7 KiB
C
73 lines
1.7 KiB
C
#include <bitmap.h>
|
|
#include <kernel.h>
|
|
#include <paging.h>
|
|
#include <pmm.h>
|
|
|
|
#include <vmm.h>
|
|
|
|
bitmap_t vmm_bitmap;
|
|
|
|
int vmm_init(void) {
|
|
vmm_bitmap.ready = 0;
|
|
vmm_bitmap.membase = KERNEL_HEAP_BASE;
|
|
vmm_bitmap.blocks = dceil(KERNEL_HEAP_SIZE, BLOCK_SIZE);
|
|
vmm_bitmap.bytes = dceil(vmm_bitmap.blocks, 8);
|
|
uint32_t bitmap_loc = 0xC6000000;
|
|
|
|
size_t page_count = dceil(vmm_bitmap.bytes, PAGE_SIZE);
|
|
for (size_t i = 0; i < page_count; i++) {
|
|
void* fr = pmm_alloc();
|
|
if (fr == NULL) return 1;
|
|
pag_map(bitmap_loc + i * PAGE_SIZE, (uint32_t)fr, 0);
|
|
}
|
|
|
|
// Set all pages as
|
|
vmm_bitmap.map = (void*)bitmap_loc;
|
|
bmap_set_all(&vmm_bitmap, 0);
|
|
|
|
bmap_set(&vmm_bitmap, 0, 1);
|
|
bmap_set(&vmm_bitmap, 8, 1);
|
|
|
|
debugf(
|
|
"[vmm] Virtual memory manager initialized.\n"
|
|
"\tBitmap manages %d blocks\n"
|
|
"\tBitmap is at 0x%08x in virtual memory\n"
|
|
"\tBitmap is %d bytes over %d pages\n",
|
|
vmm_bitmap.blocks, vmm_bitmap.map,
|
|
vmm_bitmap.bytes, page_count
|
|
);
|
|
|
|
vmm_bitmap.ready = 1;
|
|
return 0;
|
|
}
|
|
|
|
void* vmm_alloc(size_t pages) {
|
|
size_t idx = bmap_get_area(&vmm_bitmap, pages, 0);
|
|
if (idx == BMAP_NOT_FOUND) return NULL;
|
|
void* ptr = bmap_ptr(&vmm_bitmap, idx);
|
|
|
|
bmap_set_area(&vmm_bitmap, idx * BLOCK_SIZE, pages * PAGE_SIZE, 1);
|
|
|
|
for (size_t i = 0; i < pages; i++) {
|
|
void* fr = pmm_alloc();
|
|
if (fr == NULL) return NULL;
|
|
pag_map((uint32_t)ptr + i * PAGE_SIZE, (uint32_t)fr, 0);
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
void vmm_free(void* ptr, size_t pages) {
|
|
for (size_t i = 0; i < pages; i++) {
|
|
uint32_t vaddr = (uint32_t)ptr + i * PAGE_SIZE;
|
|
uint32_t paddr = (uint32_t)pag_virt_to_phys((uint32_t)ptr);
|
|
|
|
pmm_free((void*)paddr);
|
|
pag_umap(vaddr);
|
|
}
|
|
|
|
size_t base = ((size_t)ptr - KERNEL_BASE) / BLOCK_SIZE;
|
|
bmap_set_area(&vmm_bitmap, base, pages, 0);
|
|
}
|
|
|