klish/lub/heap/cache_bucket.c

#include "cache.h"

/*--------------------------------------------------------- */
size_t
lub_heap_cache_bucket__get_block_overhead(lub_heap_cache_bucket_t *this,
                                          const char              *ptr)
{
    return sizeof(lub_heap_cache_bucket_t*);
}
/*--------------------------------------------------------- */
size_t
lub_heap_cache_bucket__get_block_size(lub_heap_cache_bucket_t *this,
                                      const char              *ptr)
{
    size_t size = 0;

    /* get the size from the cache */
    lub_blockpool_stats_t stats;
    lub_blockpool__get_stats(&this->m_blockpool,&stats);
    size = stats.block_size - sizeof(lub_heap_cache_bucket_t*);

    return size;
}
/*--------------------------------------------------------- */
void
lub_heap_cache_bucket_init(lub_heap_cache_bucket_t *this,
                           lub_heap_cache_t        *cache,
                           size_t                   block_size,
                           size_t                   bucket_size)
{
    /* initialise the blockpool */
    size_t num_blocks = bucket_size/block_size;
    
    lub_blockpool_init(&this->m_blockpool,
                       this->m_memory_start,
                       block_size,
                       num_blocks);
    this->m_memory_end = this->m_memory_start + bucket_size;
    this->m_cache      = cache;
}
/*--------------------------------------------------------- */
void *
lub_heap_cache_bucket_alloc(lub_heap_cache_bucket_t *this)
{
    void                     *ptr        = 0;
    lub_heap_cache_bucket_t **bucket_ptr = lub_blockpool_alloc(&this->m_blockpool);
    if(bucket_ptr)
    {
        *bucket_ptr = this;
        ptr = ++bucket_ptr;
        /* make sure that released memory is tainted */
        lub_heap_taint_memory(ptr,
                              LUB_HEAP_TAINT_ALLOC,
                              this->m_blockpool.m_block_size-sizeof(lub_heap_cache_bucket_t*));
    }
    return ptr;
}
/*--------------------------------------------------------- */
lub_heap_status_t
lub_heap_cache_bucket_free(lub_heap_cache_bucket_t *this,
                           void                    *ptr)
{
    lub_heap_status_t         status     = LUB_HEAP_CORRUPTED;
    lub_heap_cache_bucket_t **bucket_ptr = ptr;
    --bucket_ptr;
    if(*bucket_ptr == this)
    {
        lub_blockpool_free(&this->m_blockpool,bucket_ptr);
        /* make sure that released memory is tainted */
        lub_heap_taint_memory((char*)bucket_ptr,
                              LUB_HEAP_TAINT_FREE,
                              this->m_blockpool.m_block_size);
        status = LUB_HEAP_OK;
    }
    return status;
}
/*--------------------------------------------------------- */