klish/lub/heap/heap_pre_realloc.c

#include <string.h>

#include "private.h"
#include "context.h"
#include "node.h"

/*
 * This function
 */
void lub_heap_pre_realloc(lub_heap_t * this, char **ptr, size_t * size)
{
	if (*ptr) {
		/* is this a pointer to a node in the "leak" trees? */
		lub_heap_node_t *node =
		    lub_heap_node_from_start_of_block_ptr(*ptr);
		if (NULL != node) {
			lub_heap_node_fini(node);
			/* move the pointer to the start of the block */
			*ptr = (char *)node;
		}
	}
	if ((0 < lub_heap_frame_count)) {
		size_t old_size = *size;
		if (old_size) {
			/* allocate enough bytes for a node */
			*size += lub_heap_node__get_instanceSize();
			if (*size < old_size) {
				/* we've wrapped the size variable
				 * make sure we fail the allocation
				 */
				*size = (size_t) - 1;
			}
		}
	}
}