28 lines
941 B
C
28 lines
941 B
C
#include <pool.h>
|
|
#include "os_macros.h"
|
|
#include "os_align.h"
|
|
|
|
os_bool_t pool_add_block(pool_t* self, void* block, os_size_t block_size){
|
|
// 对齐
|
|
os_uintptr_t init_start = (os_uintptr_t)block; // 初始的位置
|
|
os_uintptr_t start = OS_ALIGN_UP(init_start, OS_ALIGN_SIZE); // 向上对齐后的位置
|
|
block_size -= (start - init_start); // 实际使用的块大小
|
|
|
|
if(block_size < self->obj_size){
|
|
return OS_FALSE;
|
|
}
|
|
|
|
// 将内存块拆成 obj_size 大小
|
|
os_size_t count = block_size/self->obj_size;
|
|
uint8_t * last = &(((uint8_t*)start)[(count - 1) * self->obj_size]);
|
|
for(uint8_t * p = (uint8_t*)start; p<last; p+=self->obj_size){
|
|
((pool_link_t*)p)->next = (pool_link_t*)(p+self->obj_size);
|
|
}
|
|
((pool_link_t*)last)->next = 0;
|
|
self->free_list_p = (pool_link_t*)start;
|
|
self->capacity += count;
|
|
|
|
return OS_TRUE;
|
|
}
|
|
|