tb_pointer_t tb_allocator_nalloc_(tb_allocator_ref_t allocator, tb_size_t item, tb_size_t size __tb_debug_decl__)
{
// check
tb_assert_and_check_return_val(allocator, tb_null);
// nalloc it
return tb_allocator_malloc_(allocator, item * size __tb_debug_args__);
}
static tb_pointer_t tb_default_allocator_malloc(tb_allocator_ref_t self, tb_size_t size __tb_debug_decl__)
{
// check
tb_default_allocator_ref_t allocator = (tb_default_allocator_ref_t)self;
tb_assert_and_check_return_val(allocator, tb_null);
// check
tb_assert_and_check_return_val(allocator->large_allocator && allocator->small_allocator && size, tb_null);
// done
return size <= TB_SMALL_ALLOCATOR_DATA_MAXN? tb_allocator_malloc_(allocator->small_allocator, size __tb_debug_args__) : tb_allocator_large_malloc_(allocator->large_allocator, size, tb_null __tb_debug_args__);
}
tb_pointer_t tb_allocator_malloc0_(tb_allocator_ref_t allocator, tb_size_t size __tb_debug_decl__)
{
// check
tb_assert_and_check_return_val(allocator, tb_null);
// malloc it
tb_pointer_t data = tb_allocator_malloc_(allocator, size __tb_debug_args__);
// clear it
if (data) tb_memset_(data, 0, size);
// ok?
return data;
}
tb_pointer_t tb_allocator_malloc0_(tb_allocator_ref_t allocator, tb_size_t size __tb_debug_decl__)
{
// check
tb_assert_and_check_return_val(allocator, tb_null);
#ifdef __tb_alloc_trace__
tb_trace_w("tb_allocator_malloc0_(): %lu", size, func_, line_, file_);
#endif
// malloc it
tb_pointer_t data = tb_allocator_malloc_(allocator, size __tb_debug_args__);
#ifdef __tb_alloc_trace__
tb_trace_w("tb_allocator_malloc0_(%p): %lu", data, size, func_, line_, file_);
#endif
// clear it
if (data) tb_memset_(data, 0, size);
// ok?
return data;
}
tb_pointer_t tb_pool_malloc_(tb_pool_ref_t pool, tb_size_t size __tb_debug_decl__)
{
// check
tb_pool_impl_t* impl = (tb_pool_impl_t*)pool;
tb_assert_and_check_return_val(impl, tb_null);
// uses allocator?
if (impl->allocator) return tb_allocator_malloc_(impl->allocator, size __tb_debug_args__);
// check
tb_assert_and_check_return_val(impl->large_pool && impl->small_pool && size, tb_null);
// enter
tb_spinlock_enter(&impl->lock);
// done
tb_pointer_t data = size <= TB_SMALL_POOL_DATA_SIZE_MAXN? tb_small_pool_malloc_(impl->small_pool, size __tb_debug_args__) : tb_large_pool_malloc_(impl->large_pool, size, tb_null __tb_debug_args__);
// leave
tb_spinlock_leave(&impl->lock);
// ok?
return data;
}
static tb_pointer_t tb_default_allocator_ralloc(tb_allocator_ref_t self, tb_pointer_t data, tb_size_t size __tb_debug_decl__)
{
// check
tb_default_allocator_ref_t allocator = (tb_default_allocator_ref_t)self;
tb_assert_and_check_return_val(allocator, tb_null);
// check
tb_assert_and_check_return_val(allocator && allocator->large_allocator && allocator->small_allocator && size, tb_null);
// done
tb_pointer_t data_new = tb_null;
do
{
// no data?
if (!data)
{
// malloc it directly
data_new = size <= TB_SMALL_ALLOCATOR_DATA_MAXN? tb_allocator_malloc_(allocator->small_allocator, size __tb_debug_args__) : tb_allocator_large_malloc_(allocator->large_allocator, size, tb_null __tb_debug_args__);
break;
}
// the data head
tb_pool_data_head_t* data_head = &(((tb_pool_data_head_t*)data)[-1]);
tb_assertf(data_head->debug.magic == TB_POOL_DATA_MAGIC, "ralloc invalid data: %p", data);
tb_assert_and_check_break(data_head->size);
// small => small
if (data_head->size <= TB_SMALL_ALLOCATOR_DATA_MAXN && size <= TB_SMALL_ALLOCATOR_DATA_MAXN)
data_new = tb_allocator_ralloc_(allocator->small_allocator, data, size __tb_debug_args__);
// small => large
else if (data_head->size <= TB_SMALL_ALLOCATOR_DATA_MAXN)
{
// make the new data
data_new = tb_allocator_large_malloc_(allocator->large_allocator, size, tb_null __tb_debug_args__);
tb_assert_and_check_break(data_new);
// copy the old data
tb_memcpy_(data_new, data, tb_min(data_head->size, size));
// free the old data
tb_allocator_free_(allocator->small_allocator, data __tb_debug_args__);
}
// large => small
else if (size <= TB_SMALL_ALLOCATOR_DATA_MAXN)
{
// make the new data
data_new = tb_allocator_malloc_(allocator->small_allocator, size __tb_debug_args__);
tb_assert_and_check_break(data_new);
// copy the old data
tb_memcpy_(data_new, data, tb_min(data_head->size, size));
// free the old data
tb_allocator_large_free_(allocator->large_allocator, data __tb_debug_args__);
}
// large => large
else data_new = tb_allocator_large_ralloc_(allocator->large_allocator, data, size, tb_null __tb_debug_args__);
} while (0);
// ok?
return data_new;
}
