fix: simplify code and start frame at 1 instead of 0 for the NULL
This commit is contained in:
parent
9dc9002f00
commit
5f075a9090
@ -6,13 +6,24 @@
|
||||
#include "memory.h"
|
||||
#include "utils.h"
|
||||
|
||||
#define MAX_FRAMES 1048319
|
||||
|
||||
static uint32_t end_kernel; // WIP but it's useless now
|
||||
#define MAX_FRAMES (0xC0000000 / PAGE_SIZE)
|
||||
|
||||
static uint8_t frame_table[MAX_FRAMES];
|
||||
static uint32_t remaining_frames = MAX_FRAMES;
|
||||
|
||||
static int32_t find_next_block(size_t nb_frames)
|
||||
{
|
||||
for (uint32_t i = 1; i < MAX_FRAMES; i++) {
|
||||
uint32_t j;
|
||||
for (j = 0; frame_table[i + j] == 0 && j < nb_frames; j++)
|
||||
;
|
||||
if (j == nb_frames)
|
||||
return i;
|
||||
i += j;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
void *alloc_frames(size_t size)
|
||||
{
|
||||
const uint32_t nb_frames = CEIL(size, PAGE_SIZE);
|
||||
@ -20,35 +31,24 @@ void *alloc_frames(size_t size)
|
||||
kprintf(KERN_CRIT "Not enough frames (max: %d)\n", MAX_FRAMES);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
size_t i = 0;
|
||||
while (i < MAX_FRAMES) {
|
||||
size_t free_frames = 1;
|
||||
while (!frame_table[i + free_frames] && free_frames < nb_frames)
|
||||
free_frames++;
|
||||
if (free_frames == nb_frames)
|
||||
goto end;
|
||||
if (!free_frames)
|
||||
i++;
|
||||
i += free_frames;
|
||||
size_t i = find_next_block(nb_frames);
|
||||
if (i == -1) {
|
||||
kprintf(KERN_WARNING "Not enough frames available\n",
|
||||
MAX_FRAMES);
|
||||
return NULL;
|
||||
}
|
||||
end:
|
||||
if (i != MAX_FRAMES) {
|
||||
for (size_t j = 0; j < nb_frames; j++)
|
||||
frame_table[j + i] = 1;
|
||||
remaining_frames -= nb_frames;
|
||||
return &end_kernel + i * PAGE_SIZE;
|
||||
}
|
||||
kprintf(KERN_WARNING "Not enough frames available\n", MAX_FRAMES);
|
||||
return NULL;
|
||||
for (size_t j = 0; j < nb_frames; j++)
|
||||
frame_table[j + i] = 1;
|
||||
remaining_frames -= nb_frames;
|
||||
return (void*)(i * PAGE_SIZE);
|
||||
}
|
||||
|
||||
int free_frames(void *frame_ptr, size_t size)
|
||||
{
|
||||
const uint32_t nb_frames = CEIL(size, PAGE_SIZE);
|
||||
const uint32_t start = (frame_ptr - (void *)&end_kernel) / PAGE_SIZE;
|
||||
const uint32_t start = (uint32_t)frame_ptr / PAGE_SIZE;
|
||||
|
||||
if (start > MAX_FRAMES || frame_ptr < (void *)&end_kernel) {
|
||||
if (start > MAX_FRAMES) {
|
||||
kprintf(KERN_WARNING "Address out of range\n");
|
||||
return -1;
|
||||
} else if ((uint32_t)frame_ptr % PAGE_SIZE) {
|
||||
|
Loading…
Reference in New Issue
Block a user