make paging more stable

This commit is contained in:
Freya Murphy 2024-01-30 12:16:22 -05:00
parent 6b8f33c22d
commit e76cbbcb32
Signed by: freya
GPG key ID: 744AB800E383AE52
6 changed files with 474 additions and 185 deletions

View file

@ -1,11 +1,16 @@
#include "bindings.h" #include <lib.h>
#include <memory.h>
#include <serial.h>
#include <stdint.h> #include <stdint.h>
#include <stddef.h> #include <stddef.h>
#include "bindings.h"
#define INDEX 0x1CE #define INDEX 0x1CE
#define DATA 0x1CF #define DATA 0x1CF
#define FB_ADDR 0xE0000000 #define FB_ADDR 0xE0000000
#define FB_MAX 0xFF000000
#define FB_LEN (FB_MAX - FB_ADDR)
#define PREFERRED_VY 4096 #define PREFERRED_VY 4096
#define PREFERRED_B 32 #define PREFERRED_B 32
@ -44,29 +49,35 @@ int fb_init(uint16_t res_x, uint16_t res_y) {
outw(INDEX, 0x04); outw(INDEX, 0x04);
outw(DATA, 0x41); outw(DATA, 0x41);
uint32_t * text_vid_mem = (uint32_t *)0xA0000; // uint32_t * text_vid_mem = (uint32_t *)0xA0000;
text_vid_mem[0] = 0xA5ADFACE; // text_vid_mem[0] = 0xA5ADFACE;
//
for (uintptr_t fb_offset = FB_ADDR; fb_offset < 0xFF000000; fb_offset += 0x01000000) { // void *temp = mmap((void *) FB_ADDR, 0xFF1000);
/* Enable the higher memory */ //
for (uintptr_t i = fb_offset; i < fb_offset; i += 0x1000) { // for (uintptr_t fb_offset = FB_ADDR; fb_offset < FB_MAX; fb_offset += 0x01000000) {
// todo ident map fb // /* Enable the higher memory */
} // remap(temp, (void *)fb_offset, 0xFF1000);
//
/* Go find it */ // /* Go find it */
for (uintptr_t x = fb_offset; x < fb_offset + 0xFF0000; x += 0x1000) { // for (uintptr_t offset = 0; offset < 0xFF0000; offset += 0x1000) {
if (((uintptr_t *)x)[0] == 0xA5ADFACE) { // uintptr_t x = (uintptr_t)temp + offset;
fb_buffer = (uint8_t *) x; // if (((uintptr_t *)x)[0] == 0xA5ADFACE) {
goto mem_found; // fb_buffer = (uint8_t *) (fb_offset + offset);
} // goto mem_found;
} // }
} // }
// }
mem_found: //
// unmap(temp);
fb_res_x = res_x; //
fb_res_y = res_y; //mem_found:
fb_res_b = PREFERRED_B; //
// fb_res_x = res_x;
// fb_res_y = res_y;
// fb_res_b = PREFERRED_B;
//
// fb_buffer = mmap(fb_buffer, 0xFF0000);
// memset(fb_buffer, 7, 0xFF0000);
//
return 0; return 0;
} }

View file

@ -35,13 +35,18 @@ static uint32_t *read_tag(struct mboot_info *info, uint32_t *data) {
case MBOOT_MEMORYMAP: case MBOOT_MEMORYMAP:
read_memorymap(&tag, tag.size, data + 2); read_memorymap(&tag, tag.size, data + 2);
break; break;
case MBOOT_SYMBOLS:
// TODO:
goto done;
case MBOOT_XSDP: case MBOOT_XSDP:
read_xsdp(&tag, (char *) (data + 2)); read_xsdp(&tag, (char *) (data + 2));
break; break;
default: default:
break; goto done;
} }
done:
info->tags[tag.type] = tag; info->tags[tag.type] = tag;
if(tag.size % 8 != 0) { if(tag.size % 8 != 0) {

View file

@ -19,6 +19,7 @@ struct mboot_tag {
enum mboot_tag_type { enum mboot_tag_type {
MBOOT_CMDLINE = 0, MBOOT_CMDLINE = 0,
MBOOT_MEMORYMAP = 6, MBOOT_MEMORYMAP = 6,
MBOOT_SYMBOLS = 9,
MBOOT_XSDP = 14 MBOOT_XSDP = 14
}; };

View file

@ -61,170 +61,435 @@ extern struct pte paging_pt[512]; // paging_pt should NEVER be outside of this f
// paged address to read page tables // paged address to read page tables
// the structures are not gurenteed to be ident mapped // the structures are not gurenteed to be ident mapped
// map them here with map_<type>(phys_addr) before useing structures // map them here with map_<type>(phys_addr) before useing structures
void *addr_mapped = (void *) (uintptr_t) 0x204000;
static struct pml4e *pml4_mapped = (void *) (uintptr_t) 0x200000;
static struct pdpte *pdpt_mapped = (void *) (uintptr_t) 0x201000; static struct pdpte *pdpt_mapped = (void *) (uintptr_t) 0x201000;
static struct pdpte *pd_mapped = (void *) (uintptr_t) 0x202000; static struct pde *pd_mapped = (void *) (uintptr_t) 0x202000;
static struct pdpte *pt_mapped = (void *) (uintptr_t) 0x203000; static struct pte *pt_mapped = (void *) (uintptr_t) 0x203000;
void *addr_mapped = (void *) (uintptr_t) 0x204000;
static inline void invlpg(void *addr) { static inline void
invlpg(void *addr)
{
__asm volatile("invlpg (%0)" ::"r" (addr) : "memory"); __asm volatile("invlpg (%0)" ::"r" (addr) : "memory");
} }
static void map_pdpt(struct pdpte *phys_addr) { static void
paging_pt[1].address = (((uint64_t)phys_addr) >> 12); load_addr(void *phys_addr)
paging_pt[1].flags = F_PRESENT | F_WRITEABLE; {
invlpg(pdpt_mapped); static struct pte *pt = &paging_pt[4];
} pt->address = (uint64_t)phys_addr >> 12;
pt->flags = F_PRESENT | F_WRITEABLE;
static void map_pd(struct pde *phys_addr) {
paging_pt[2].address = (((uint64_t)phys_addr) >> 12);
paging_pt[2].flags = F_PRESENT | F_WRITEABLE;
invlpg(pd_mapped);
}
static void map_pt(struct pte *phys_addr) {
paging_pt[3].address = (((uint64_t)phys_addr) >> 12);
paging_pt[3].flags = F_PRESENT | F_WRITEABLE;
invlpg(pt_mapped);
}
static void map_addr(void *phys_addr) {
paging_pt[4].address = (((uint64_t)phys_addr) >> 12);
paging_pt[4].flags = F_PRESENT | F_WRITEABLE;
invlpg(addr_mapped); invlpg(addr_mapped);
} }
//static int get_maxphysaddr() { static void
// uint32_t eax, ebx, ecx, edx; load_pml4(
// __cpuid(0x80000008, eax, ebx, ecx, edx); void *phys
// return eax & 0xFF; ) {
//} static struct pte *pt = &paging_pt[0];
pt->address = (uint64_t)phys >> 12;
pt->flags = F_PRESENT | F_WRITEABLE;
invlpg(pml4_mapped);
}
//int find_phys_addr(struct pml4e *pml4, void *virt_addr, void **phys_addr) { static void
// uint64_t pml4_offset = ((uint64_t)virt_addr) >> 39; load_pdpt(
// uint64_t pdpt_offset = (((uint64_t)virt_addr) >> 30) & 0x1FF; void *phys
// uint64_t pd_offset = (((uint64_t)virt_addr) >> 21) & 0x1FF; ) {
// uint64_t pt_offset = (((uint64_t)virt_addr) >> 12) & 0x1FF; static struct pte *pt = &paging_pt[1];
// uint64_t page_offset = (((uint64_t)virt_addr) ) & 0xFFF; pt->address = (uint64_t)phys >> 12;
// pt->flags = F_PRESENT | F_WRITEABLE;
// if (!(pml4[pml4_offset].flags & F_PRESENT)) invlpg(pdpt_mapped);
// return -1; }
// struct pdpte *pdpt = (struct pdpte *)(pml4[pml4_offset].address << 12);
// if (!(pdpt[pdpt_offset].flags & F_PRESENT))
// return -1;
// struct pde *pd = (struct pde *)(pdpt[pdpt_offset].address << 12);
// if (!(pd[pd_offset].flags & F_PRESENT))
// return -1;
// struct pte *pt = (struct pte *)(pd[pd_offset].address << 12);
// if (!(pt[pt_offset].flags & F_PRESENT))
// return -1;
// *phys_addr = (void *)((pt[pt_offset].address << 12) + page_offset);
// return 0;
//}
char *curr_alloc = (void *) 0x5000; static void
load_pd(
void *phys
) {
static struct pte *pt = &paging_pt[2];
pt->address = (uint64_t)phys >> 12;
pt->flags = F_PRESENT | F_WRITEABLE;
invlpg(pdpt_mapped);
}
int map_page(struct pml4e *pml4, void *virt_addr, void *phys_addr, unsigned int flags) { static void
uint64_t pml4_offset = ((uint64_t)virt_addr) >> 39; load_pt(
uint64_t pdpt_offset = (((uint64_t)virt_addr) >> 30) & 0x1FF; void *phys
uint64_t pd_offset = (((uint64_t)virt_addr) >> 21) & 0x1FF; ) {
uint64_t pt_offset = (((uint64_t)virt_addr) >> 12) & 0x1FF; static struct pte *pt = &paging_pt[3];
//uint64_t page_offset = (((uint64_t)virt_addr) ) & 0xFFF; pt->address = (uint64_t)phys >> 12;
pt->flags = F_PRESENT | F_WRITEABLE;
invlpg(pt_mapped);
}
if (!(pml4[pml4_offset].flags & F_PRESENT)) { #define PAG_SUCCESS 0
#define PAG_CANNOT_ALLOC 1
#define PAG_NOT_PRESENT 2
static int
select_pdpt(
void *virt,
unsigned int flags,
struct pml4e *root,
struct pdpte **res,
bool create
) {
load_pml4(root);
uint64_t offset = (uint64_t)virt >> 39;
struct pml4e *pml4e = &pml4_mapped[offset];
if (!(pml4e->flags & F_PRESENT)) {
if (!create) {
return PAG_NOT_PRESENT;
}
void *new_page = alloc_phys_page(); void *new_page = alloc_phys_page();
map_addr(new_page); if (new_page == NULL) {
memset(addr_mapped, 0, 4096); return PAG_CANNOT_ALLOC;
pml4[pml4_offset].address = ((uint64_t)new_page) >> 12; }
load_addr(new_page);
memset(addr_mapped, 0, PAGE_SIZE);
pml4e->address = ((uint64_t)new_page) >> 12;
pml4e->flags = F_PRESENT;
} }
pml4[pml4_offset].flags = F_PRESENT | flags; if (flags)
pml4e->flags = F_PRESENT | flags;
*res = (struct pdpte *)(uintptr_t)(pml4e->address << 12);
return PAG_SUCCESS;
}
struct pdpte *__pdpt = (struct pdpte *)(uintptr_t)(pml4[pml4_offset].address << 12); static int
map_pdpt(__pdpt); select_pd(
if (!(pdpt_mapped[pdpt_offset].flags & F_PRESENT)) { void *virt,
unsigned int flags,
struct pdpte *pdpt,
struct pde **res,
bool create
) {
load_pdpt(pdpt);
uint64_t offset = ((uint64_t)virt >> 30) & 0x1ff;
struct pdpte *pdpte = &pdpt_mapped[offset];
if (!(pdpte->flags & F_PRESENT)) {
if (!create) {
return PAG_NOT_PRESENT;
}
void *new_page = alloc_phys_page(); void *new_page = alloc_phys_page();
map_addr(new_page); if (new_page == NULL) {
memset(addr_mapped, 0, 4096); return PAG_CANNOT_ALLOC;
pdpt_mapped[pdpt_offset].address = ((uint64_t)new_page) >> 12; }
load_addr(new_page);
memset(addr_mapped, 0, PAGE_SIZE);
pdpte->address = ((uint64_t)new_page) >> 12;
pdpte->flags = F_PRESENT;
} }
pdpt_mapped[pdpt_offset].flags = F_PRESENT | flags; if (flags)
pdpte->flags = F_PRESENT | flags;
*res = (struct pde *)(uintptr_t)(pdpte->address << 12);
return PAG_SUCCESS;
}
struct pde *__pd = (struct pde *)(uintptr_t)(pdpt_mapped[pdpt_offset].address << 12); static int
map_pd(__pd); select_pt(
if (!(pd_mapped[pd_offset].flags & F_PRESENT)) { void *virt,
unsigned int flags,
struct pde *pd,
struct pte **res,
bool create
) {
load_pd(pd);
uint64_t offset = ((uint64_t)virt >> 21) & 0x1ff;
struct pde *pde = &pd_mapped[offset];
if (!(pde->flags & F_PRESENT)) {
if (!create) {
return PAG_NOT_PRESENT;
}
void *new_page = alloc_phys_page(); void *new_page = alloc_phys_page();
map_addr(new_page); if (new_page == NULL) {
memset(addr_mapped, 0, 4096); return PAG_CANNOT_ALLOC;
pd_mapped[pd_offset].address = ((uint64_t)new_page) >> 12; }
load_addr(new_page);
memset(addr_mapped, 0, PAGE_SIZE);
pde->address = ((uint64_t)new_page) >> 12;
pde->flags = F_PRESENT;
} }
pd_mapped[pd_offset].flags = F_PRESENT | flags; if (flags)
pde->flags = F_PRESENT | flags;
*res = (struct pte *)(uintptr_t)(pde->address << 12);
return PAG_SUCCESS;
}
struct pte *__pt = (struct pte *)(uintptr_t)(pd_mapped[pd_offset].address << 12); static void
map_pt(__pt); select_page(
pt_mapped[pt_offset].flags = F_PRESENT | flags; void *virt,
pt_mapped[pt_offset].address = (((uint64_t)phys_addr) >> 12); struct pte *pt,
struct pte **res
) {
load_pt(pt);
uint64_t offset = ((uint64_t)virt >> 12) & 0x1ff;
struct pte *page = &pt_mapped[offset];
*res = page;
return;
}
static inline void
try_unmap_pml4(void) {
for (int i = 0; i < 512; i++) {
if (pml4_mapped[i].flags & F_PRESENT)
return;
}
for (int i = 0; i < 512; i++) {
if (pml4_mapped[i].address) {
void *addr = (void *)(uintptr_t)(pml4_mapped[i].address << 12);
free_phys_page(addr);
}
}
}
static inline void
try_unmap_pdpt(void) {
for (int i = 0; i < 512; i++) {
if (pdpt_mapped[i].flags & F_PRESENT)
return;
}
for (int i = 0; i < 512; i++) {
if (pdpt_mapped[i].address) {
void *addr = (void *)(uintptr_t)(pdpt_mapped[i].address << 12);
free_phys_page(addr);
}
}
try_unmap_pml4();
}
static inline void
try_unmap_pd(void) {
for (int i = 0; i < 512; i++) {
if (pd_mapped[i].flags & F_PRESENT)
return;
}
for (int i = 0; i < 512; i++) {
if (pd_mapped[i].address) {
void *addr = (void *)(uintptr_t)(pd_mapped[i].address << 12);
free_phys_page(addr);
}
}
try_unmap_pdpt();
}
static inline void
try_unmap_pt(void) {
for (int i = 0; i < 512; i++) {
if (pt_mapped[i].flags & F_PRESENT)
return;
}
for (int i = 0; i < 512; i++) {
if (pt_mapped[i].address) {
void *addr = (void *)(uintptr_t)(pt_mapped[i].address << 12);
free_phys_page(addr);
}
}
try_unmap_pd();
}
static void
unmap_page(
struct pml4e *root,
void *virt
) {
struct pdpte *pdpt;
struct pde *pd;
struct pte *pt;
struct pte *page;
unsigned int df = 0;
if (select_pdpt(virt, df, root, &pdpt, false))
return;
if (select_pd(virt, df, pdpt, &pd, false))
return;
if (select_pt(virt, df, pd, &pt, false))
return;
select_page(virt, pt, &page);
page->address = 0;
page->flags = 0;
try_unmap_pt();
invlpg(virt);
return;
}
static void
unmap_pages(
struct pml4e *root,
void *virt_start,
long page_count
) {
uint64_t pml4_o = -1,
pdpt_o = -1,
pd_o = -1;
uint64_t pml4_n,
pdpt_n,
pd_n;
struct pdpte *pdpt = NULL;
struct pde *pd = NULL;
struct pte *pt = NULL;
struct pte *page = NULL;
unsigned int df = 0;
void *virt;
for (long i = 0; i < page_count; i++) {
virt = (char *)virt_start + (i * PAGE_SIZE);
pml4_n = (uint64_t) virt >> 39;
pdpt_n = ((uint64_t) virt >> 30) & 0x1ff;
pd_n = ((uint64_t) virt >> 21) & 0x1ff;
if (pdpt == NULL || pml4_o != pml4_n) {
if (select_pdpt(virt, df, root, &pdpt, false))
continue;
pml4_o = pml4_n;
}
if (pd == NULL || pdpt_o != pdpt_n) {
if (select_pd(virt, df, pdpt, &pd, false))
continue;
pdpt_o = pdpt_n;
}
if (pt == NULL || pd_o != pd_n) {
if (pt) {
try_unmap_pt();
}
if (select_pt(virt, df, pd, &pt, false))
continue;
pd_o = pd_n;
}
select_page(virt, pt, &page);
page->address = 0;
page->flags = 0;
}
if (pt != NULL)
try_unmap_pt();
return;
}
static int
map_page(
struct pml4e *root,
void *virt,
void *phys,
unsigned int flags
) {
struct pdpte *pdpt;
struct pde *pd;
struct pte *pt;
struct pte *page;
unsigned int df = F_WRITEABLE;
if (select_pdpt(virt, df, root, &pdpt, true))
return 1;
if (select_pd(virt, df, pdpt, &pd, true))
return 1;
if (select_pt(virt, df, pd, &pt, true))
return 1;
select_page(virt, pt, &page);
page->address = (uint64_t)phys >> 12;
page->flags = F_PRESENT | flags;
invlpg(virt);
invlpg(virt_addr);
return 0; return 0;
} }
int unmap_page(struct pml4e *pml4, void *virt_addr) { static int
uint64_t pml4_offset = ((uint64_t)virt_addr) >> 39; map_pages(
uint64_t pdpt_offset = (((uint64_t)virt_addr) >> 30) & 0x1FF; struct pml4e *root,
uint64_t pd_offset = (((uint64_t)virt_addr) >> 21) & 0x1FF; void *virt_start,
uint64_t pt_offset = (((uint64_t)virt_addr) >> 12) & 0x1FF; void *phys_start,
//uint64_t page_offset = (((uint64_t)virt_addr) ) & 0xFFF; unsigned int flags,
long page_count
) {
if (!(pml4[pml4_offset].flags & F_PRESENT)) uint64_t pml4_o = -1,
return -1; pdpt_o = -1,
struct pdpte *__pdpt = (struct pdpte *)(uintptr_t)(pml4[pml4_offset].address << 12); pd_o = -1;
map_pdpt(__pdpt);
if (!(pdpt_mapped[pdpt_offset].flags & F_PRESENT))
return -1;
struct pde *__pd = (struct pde *)(uintptr_t)(pdpt_mapped[pdpt_offset].address << 12);
map_pd(__pd);
if (!(pd_mapped[pd_offset].flags & F_PRESENT))
return -1;
struct pte *__pt = (struct pte *)(uintptr_t)(pd_mapped[pd_offset].address << 12);
map_pt(__pt);
if (!(pt_mapped[pt_offset].flags & F_PRESENT))
return -1;
pt_mapped[pt_offset].flags = 0; uint64_t pml4_n,
pdpt_n,
pd_n;
struct pdpte *pdpt = NULL;
struct pde *pd = NULL;
struct pte *pt = NULL;
struct pte *page = NULL;
void *virt, *phys;
unsigned int df = F_WRITEABLE;
long i;
for (i = 0; i < page_count; i++) {
virt = (char *)virt_start + (i * PAGE_SIZE);
phys = (char *)phys_start + (i * PAGE_SIZE);
pml4_n = (uint64_t) virt >> 39;
pdpt_n = ((uint64_t) virt >> 30) & 0x1ff;
pd_n = ((uint64_t) virt >> 21) & 0x1ff;
if (pdpt == NULL || pml4_o != pml4_n) {
if (select_pdpt(virt, df, root, &pdpt, true))
goto failed;
pml4_o = pml4_n;
}
if (pd == NULL || pdpt_o != pdpt_n) {
if (select_pd(virt, df, pdpt, &pd, true))
goto failed;
pdpt_o = pdpt_n;
}
if (pt == NULL || pd_o != pd_n) {
if (select_pt(virt, df, pd, &pt, true))
goto failed;
pd_o = pd_n;
}
select_page(virt, pt, &page);
page->address = (uint64_t)phys >> 12;
page->flags = F_PRESENT | flags;
invlpg(virt);
int i = 0;
for(; i < 512; i++) {
if (pt_mapped[i].flags & F_PRESENT)
break;
} }
if (i == 512)
goto done;
pd_mapped[pd_offset].flags = 0;
for(i = 0; i < 512; i++) {
if (pd_mapped[i].flags & F_PRESENT)
break;
}
if (i == 512)
goto done;
pdpt_mapped[pdpt_offset].flags = 0;
for(i = 0; i < 512; i++) {
if(pdpt_mapped[i].flags & F_PRESENT)
break;
}
if (i == 512)
goto done;
pml4[pml4_offset].flags = 0;
//TODO: Return memory used for page structures
done:
invlpg(virt_addr);
return 0; return 0;
failed:
unmap_pages(root, virt, i);
return 1;
} }
void paging_init(void) { void paging_init(void) {
@ -260,21 +525,24 @@ void *mmap(void *addr, size_t len) {
return NULL; return NULL;
} }
void *phys = page_align(addr); void *phys = page_align(addr);
for (long i = 0; i < pages; i++) { if (map_pages(
void *virt_temp = (char *)virt + (i * PAGE_SIZE); kernel_pml4,
void *phys_temp = (char *)phys + (i * PAGE_SIZE); virt,
map_page(kernel_pml4, virt_temp, phys_temp, F_WRITEABLE); phys,
F_WRITEABLE,
pages
)) {
virtaddr_free(virt);
return NULL;
} }
map_page(kernel_pml4, virt, (void*)0x23443, F_WRITEABLE);
return virt; return virt;
} }
void unmap(void *addr) { void unmap(void *addr) {
long pages = virtaddr_free(addr); long pages = virtaddr_free(addr);
for (long i = 0; i < pages; i++) { if (pages < 1)
void *virt = (char *)addr + (i * PAGE_SIZE); return;
unmap_page(kernel_pml4, virt); unmap_pages(kernel_pml4, addr, pages);
}
} }
void *alloc_pages(int count) { void *alloc_pages(int count) {
@ -286,10 +554,15 @@ void *alloc_pages(int count) {
virtaddr_free(virt); virtaddr_free(virt);
return NULL; return NULL;
} }
for (int i = 0; i < count; i++) { if (map_pages(
void *virt_temp = (char *)virt + (i * PAGE_SIZE); kernel_pml4,
void *phys_temp = (char *)phys + (i * PAGE_SIZE); virt,
map_page(kernel_pml4, virt_temp, phys_temp, F_WRITEABLE); phys,
F_WRITEABLE,
count
)) {
virtaddr_free(virt);
return NULL;
} }
return virt; return virt;
} }
@ -298,10 +571,7 @@ void free_page(void *virt) {
long pages = virtaddr_free(virt); long pages = virtaddr_free(virt);
if (pages < 1) if (pages < 1)
return; return;
for (long i = 0; i < pages; i++) { unmap_pages(kernel_pml4, virt, pages);
void *virt_temp = (char *)virt + (i * PAGE_SIZE);
unmap_page(kernel_pml4, virt_temp);
}
} }
void memory_lock(void) { void memory_lock(void) {
@ -317,5 +587,6 @@ int kmap_page(void *virt_addr, void *phys_addr, unsigned int flags) {
} }
int kunmap_page(void *virt_addr) { int kunmap_page(void *virt_addr) {
return unmap_page(kernel_pml4, virt_addr); unmap_page(kernel_pml4, virt_addr);
return 0;
} }

View file

@ -10,7 +10,6 @@
#define F_MEGABYTE 0x080 #define F_MEGABYTE 0x080
#define F_GLOBAL 0x100 #define F_GLOBAL 0x100
void paging_init(void);
int kmap_page(void *virt_addr, void *phys_addr, unsigned int flags); int kmap_page(void *virt_addr, void *phys_addr, unsigned int flags);
int kunmap_page(void *virt_addr); int kunmap_page(void *virt_addr);
void paging_init(void);

View file

@ -26,6 +26,7 @@ static uint64_t total_memory;
static uint64_t free_memory; static uint64_t free_memory;
static uint64_t page_count; static uint64_t page_count;
static uint64_t page_free_start; static uint64_t page_free_start;
static uint64_t segment_count;
static struct memory_area *page_start; static struct memory_area *page_start;
static int n_pages(const struct memory_area *m) { static int n_pages(const struct memory_area *m) {
@ -34,7 +35,8 @@ static int n_pages(const struct memory_area *m) {
static void *page_at(int i) { static void *page_at(int i) {
int cur_page = 0; int cur_page = 0;
for (struct memory_area *m = page_start; m != NULL; m++) { for (uint64_t idx = 0; idx < segment_count; idx++) {
struct memory_area *m = &page_start[idx];
int pages = n_pages(m); int pages = n_pages(m);
if (i - cur_page < pages) { if (i - cur_page < pages) {
return (void *) (m->addr + (PAGE_SIZE * (i - cur_page))); return (void *) (m->addr + (PAGE_SIZE * (i - cur_page)));
@ -165,7 +167,7 @@ void memory_init(struct memory_map *map) {
end += map->size; end += map->size;
struct memory_segment *segment = &map->entries[0]; struct memory_segment *segment = &map->entries[0];
int segment_count = 0; segment_count = 0;
for(; (uintptr_t) segment < end; segment++) { for(; (uintptr_t) segment < end; segment++) {
if (segment_invalid(segment)) if (segment_invalid(segment))