summaryrefslogtreecommitdiff
path: root/ccan/tal/test/run-test-backend.c
blob: 8fdfc064abb7eca68d06a18d0f4b00f7b9927df3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include <stdlib.h>
#include <stdbool.h>

/* Make sure it always uses our allocation/resize/free fns! */
static bool my_alloc_called;

static void *my_alloc(size_t len)
{
	my_alloc_called = true;
	return (char *)malloc(len + 16) + 16;
}

static void my_free(void *p)
{
	if (p)
		free((char *)p - 16);
}

static void *my_realloc(void *old, size_t new_size)
{
	return (char *)realloc((char *)old - 16, new_size + 16) + 16;
}

#define free ((void (*)(void *))abort)
#define malloc ((void *(*)(size_t))abort)
#define realloc ((void *(*)(void *, size_t))abort)

#include <ccan/tal/tal.h>
#include <ccan/tal/tal.c>
#include <ccan/tap/tap.h>

#define NUM_ALLOCS 1000

static void destroy_p(void *p)
{
}

int main(void)
{
	void *p, *c[NUM_ALLOCS];
	int i;
	char *name;

	/* Mostly we rely on the allocator (or valgrind) crashing. */
	plan_tests(1);

	tal_set_backend(my_alloc, my_realloc, my_free, NULL);

	p = tal(NULL, char);
	ok1(my_alloc_called);

	/* Adding properties makes us allocated. */
	tal_add_destructor(p, destroy_p);

	tal_set_name(p, "test");
	name = tal_arr(NULL, char, 6);
	strcpy(name, "test2");
	tal_set_name(p, name);
	/* makes us free old name */
	tal_set_name(p, name);
	tal_free(name);

	/* Add lots of children. */
	for (i = 0; i < NUM_ALLOCS; i++)
		c[i] = tal(p, char);

	/* Now steal a few. */
	for (i = 1; i < NUM_ALLOCS / 2; i++)
		tal_steal(c[0], c[i]);

	/* Now free individual ones.. */
	for (i = NUM_ALLOCS / 2; i < NUM_ALLOCS; i++)
		tal_free(c[i]);

	/* Finally, free the parent. */
	tal_free(p);

	tal_cleanup();
	return exit_status();
}