ccan/ccan/talloc/test/run-external-alloc.c

#include <ccan/talloc/talloc.c>
#include <ccan/tap/tap.h>
#include <assert.h>

/* Much testing already done in run.c */

static int ext_alloc_count, ext_free_count, ext_realloc_count, lock_count, unlock_count;
static void *expected_parent;

static void *ext_realloc(const void *parent, void *ptr, size_t size)
{
	ok1(parent == expected_parent);
	if (ptr == NULL)
		ext_alloc_count++;
	if (size == 0)
		ext_free_count++;
	if (ptr && size)
		ext_realloc_count++;
	return realloc(ptr, size);
}

static void ext_lock(const void *ctx)
{
	lock_count++;
}

static void ext_unlock(void)
{
	unlock_count++;
}

int main(void)
{
	char *p, *p2, *head;
	plan_tests(15);

	expected_parent = NULL;
	head = talloc_add_external(NULL, ext_realloc, ext_lock, ext_unlock);
	assert(head);
	ok1(ext_alloc_count == 1);

	expected_parent = head;
	p = talloc_array(head, char, 1);
	ok1(ext_alloc_count == 2);
	assert(p);

	/* Child is also externally allocated */
	expected_parent = p;
	p2 = talloc(p, char);
	ok1(ext_alloc_count == 3);

	expected_parent = head;
	p = talloc_realloc(NULL, p, char, 1000);
	ok1(ext_realloc_count == 1);
	assert(p);

	expected_parent = p;
	talloc_free(p2);
	ok1(ext_free_count == 1);

	expected_parent = head;
	talloc_free(p);
	ok1(ext_free_count == 2);

	expected_parent = NULL;
	talloc_free(head);
	ok1(ext_free_count == 3);

	ok1(lock_count == unlock_count);

	return exit_status();
}