fix: update frame count when allocating some
This commit is contained in:
parent
f559e71433
commit
5ebf8ac6ab
@ -27,6 +27,6 @@ void kernel_main(void)
|
||||
init_memory();
|
||||
char *str = (char *)kalloc_frame(1);
|
||||
memcpy(str, "Hello world!\n", 15);
|
||||
kprintf(KERN_INFO, "%s", str);
|
||||
kprintf(KERN_INFO, "%d: %s", str, str);
|
||||
shell_init();
|
||||
}
|
||||
|
@ -22,12 +22,14 @@ static uint8_t frame_table[CEIL(MAX_FRAMES, 8)];
|
||||
|
||||
uintptr_t kalloc_frame(uint32_t nb_frames)
|
||||
{
|
||||
if (nb_frames > MAX_FRAMES)
|
||||
static uint32_t remaining_frames = MAX_FRAMES;
|
||||
if (nb_frames > remaining_frames)
|
||||
kprintf(KERN_CRIT, "Not enough frames (max: %d)\n", MAX_FRAMES);
|
||||
size_t i = 0;
|
||||
while (i < MAX_FRAMES) {
|
||||
size_t free_frames = 1;
|
||||
while (!GET_FRAME(i + free_frames) && free_frames < nb_frames)
|
||||
while (!GET_FRAME(i + free_frames) && free_frames < nb_frames &&
|
||||
i + free_frames < remaining_frames)
|
||||
free_frames++;
|
||||
if (free_frames == nb_frames)
|
||||
goto end;
|
||||
@ -37,6 +39,7 @@ end:
|
||||
if (i != MAX_FRAMES) {
|
||||
for (size_t j = 0; j < nb_frames; j++)
|
||||
SET_FRAME(j + i, 1);
|
||||
remaining_frames -= nb_frames;
|
||||
return (uintptr_t)&end_kernel + i * PAGE_SIZE;
|
||||
}
|
||||
kprintf(KERN_WARNING, "Not enough frames available\n", MAX_FRAMES);
|
||||
|
Loading…
Reference in New Issue
Block a user