85 lines
1.9 KiB
C
85 lines
1.9 KiB
C
/*
|
|
* memory_alloc.c
|
|
*
|
|
* Created on: Aug 27, 2011
|
|
* Author: Tiberiu
|
|
*/
|
|
|
|
#include <memory-add.h>
|
|
|
|
extern uint32 mem_kernel_end;
|
|
extern uint8 mem_initialized;
|
|
|
|
// Used prior to proper initialization
|
|
uint32 _malloc_init1 (uint32 size, uint8 page_aligned)
|
|
{
|
|
uint32 ret = mem_kernel_end;
|
|
|
|
if (page_aligned && (ret & 0xfff)) ret = (ret & 0xfffff000) + 0x1000;
|
|
mem_kernel_end = size + ret;
|
|
|
|
LogMem("%#Allocated %u bytes (%spage aligned) at end of kernel (0x%x).\n", ColorLightMagenta, size, ((page_aligned) ? "" : "not "), ret);
|
|
return (ret);
|
|
}
|
|
|
|
uint32 _malloc_init2 (uint32 size, uint8 page_aligned, uint32* phys)
|
|
{
|
|
uint32 ret = MemHeapAlloc(size, page_aligned, KernelHeap, KernelDirectory);
|
|
|
|
if (phys)
|
|
{
|
|
Page *pg = PagingGetPage(ret, 0, KernelDirectory);
|
|
*phys = (*pg & PageFrame) + (ret & 0xFFF);
|
|
|
|
LogMem("%#Allocated %u bytes (%spage aligned) at address 0x%x (phys=%x).\n", ColorLightMagenta, size, ((page_aligned) ? "" : "not "), ret, *phys);
|
|
}
|
|
|
|
else {
|
|
LogMem("%#Allocated %u bytes (%spage aligned) at address 0x%x.\n", ColorLightMagenta, size, ((page_aligned) ? "" : "not "), ret);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
// Allocate 'size' bytes
|
|
void* kmalloc (uint32 size)
|
|
{
|
|
if (!mem_initialized) return 0;
|
|
if (mem_initialized == 1) return (void*)_malloc_init1(size, 0);
|
|
|
|
return (void*)_malloc_init2(size,0,0);
|
|
}
|
|
|
|
// Allocate 'size' bytes, page aligned
|
|
void* kmalloc_a(uint32 size)
|
|
{
|
|
if (!mem_initialized) return 0;
|
|
if (mem_initialized == 1) return (void*)_malloc_init1(size, 1);
|
|
|
|
return (void*)_malloc_init2(size,1,0);
|
|
}
|
|
|
|
void* kmalloc_p(uint32 size, uint32* phys)
|
|
{
|
|
if (!mem_initialized) return 0;
|
|
if (mem_initialized == 1) {
|
|
*phys = _malloc_init1(size,0);
|
|
return (void*)(*phys);
|
|
}
|
|
|
|
return (void*)_malloc_init2(size,0,phys);
|
|
}
|
|
|
|
void* kmalloc_ap(uint32 size, uint32* phys)
|
|
{
|
|
if (!mem_initialized) return 0;
|
|
|
|
uint32 ret;
|
|
if (mem_initialized == 1) {
|
|
*phys = ret = _malloc_init1(size,1);
|
|
}
|
|
|
|
else ret = _malloc_init2(size,1,phys);
|
|
return (void*)ret;
|
|
}
|