mykola/cutil
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add prev and next block checking in heap_free

Browse source
This commit is contained in:
mykola2312 2022-05-24 21:54:00 +03:00
parent dae46bb74d
commit 0570f9fdd9
4 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cutil.c
View file

@ -20,7 +20,7 @@ void* _cu_realloc(void* mem, size_t size)
void _cu_free(void* mem)
{
return heap_free((mblock_t*)((u8*)mem - sizeof(mblock_t)));
return heap_free(&cu_heap, (mblock_t*)((u8*)mem - sizeof(mblock_t)));
}
void* (*cu_malloc)(size_t) = _cu_malloc;

9
heap.c
View file

@ -55,14 +55,15 @@ mblock_t* heap_alloc(mheap_t* heap, uword _size)
return block;
}
void heap_free(mblock_t* block)
void heap_free(mheap_t* heap, mblock_t* block)
{
block->size &= ~MBLOCK_ALLOCATED;
mblock_t* prev = block->prev;
mblock_t* next = (mblock_t*)((u8*)block + block->size);
if (!(prev->size & MBLOCK_ALLOCATED))
mblock_t* end = (mblock_t*)((u8*)heap->start + heap->size);
if (prev && !(prev->size & MBLOCK_ALLOCATED))
heap_join(block, prev);
if (!(next->size & MBLOCK_ALLOCATED))
if (next < end && !(next->size & MBLOCK_ALLOCATED))
heap_join(block, next);
}
@ -81,7 +82,7 @@ mblock_t* heap_realloc(mheap_t* heap, mblock_t* block, uword _size)
}
else
{
heap_free(block);
heap_free(heap, block);
new = heap_alloc(heap, size);
if (new)
{

2
heap.h
View file

@ -26,7 +26,7 @@ void heap_init(mheap_t* heap, void* start, uword size);
void heap_split(mblock_t* block, uword size);
void heap_join(mblock_t* block, mblock_t* other);
mblock_t* heap_alloc(mheap_t* heap, uword _size);
void heap_free(mblock_t* block);
void heap_free(mheap_t* heap, mblock_t* block);
mblock_t* heap_realloc(mheap_t* heap, mblock_t* block, uword _size);
#endif

2
test.c
View file

@ -207,7 +207,7 @@ int main()
void* m1 = heap_alloc(&heap, 8);
void* m2 = heap_alloc(&heap, 32);
void* m3 = heap_alloc(&heap, 9);
heap_free(m2);
heap_free(&heap, m2);
void* m4 = heap_alloc(&heap, 8);
printf("heap_alloc\t%p\n", m1);
printf("heap_alloc\t%p\n", m2);