klish/lub/dblockpool/dblockpool_alloc.c

#include <stdlib.h>

#include "private.h"

/*--------------------------------------------------------- */
void *lub_dblockpool_alloc(lub_dblockpool_t * this)
{
	void *result = NULL;
	lub_dblockpool_chunk_t *chunk;
	unsigned chunk_count = 0;

	/* first find a chunk which can service this request */
	for (chunk = this->first_chunk; chunk; chunk = chunk->next) {
		chunk_count++;
		/* try and get a block from this chunk */
		result = lub_blockpool_alloc(&chunk->pool);
		if (NULL != result) {
			/* got some memory */
			break;
		}
	}
	if ((NULL == result)
	    && (!this->max_chunks || (chunk_count < this->max_chunks))) {
		/* dynamically allocate a new chunk */
		chunk =
		    malloc(sizeof(lub_dblockpool_chunk_t) +
			   (this->block_size * this->chunk_size));
		if (NULL != chunk) {
			/* configure the new chunk */
			chunk->next = this->first_chunk;
			lub_blockpool_init(&chunk->pool,
					   &chunk[1],
					   this->block_size, this->chunk_size);
			this->first_chunk = chunk;
			chunk->count = 0;

			/* now allocate the memory */
			result = lub_blockpool_alloc(&chunk->pool);
		}
	}
	if ((NULL != result) && (NULL != chunk)) {
		/* track the number of allocations */
		chunk->count++;
	}
	return result;
}

/*--------------------------------------------------------- */