talloc: spelling fix.
[ccan] / ccan / talloc / test / run-external-alloc.c
1 #include <ccan/talloc/talloc.c>
2 #include <ccan/tap/tap.h>
3 #include <assert.h>
4
5 /* Much testing already done in run.c */
6
7 static int ext_alloc_count, ext_free_count, ext_realloc_count, lock_count, unlock_count;
8 static void *expected_parent;
9
10 static void *ext_realloc(const void *parent, void *ptr, size_t size)
11 {
12         ok1(parent == expected_parent);
13         if (ptr == NULL)
14                 ext_alloc_count++;
15         if (size == 0)
16                 ext_free_count++;
17         if (ptr && size)
18                 ext_realloc_count++;
19         return realloc(ptr, size);
20 }
21
22 static void ext_lock(const void *ctx)
23 {
24         lock_count++;
25 }
26
27 static void ext_unlock(void)
28 {
29         unlock_count++;
30 }
31
32 int main(void)
33 {
34         char *p, *p2, *head;
35         plan_tests(15);
36
37         expected_parent = NULL;
38         head = talloc_add_external(NULL, ext_realloc, ext_lock, ext_unlock);
39         assert(head);
40         ok1(ext_alloc_count == 1);
41
42         expected_parent = head;
43         p = talloc_array(head, char, 1);
44         ok1(ext_alloc_count == 2);
45         assert(p);
46
47         /* Child is also externally allocated */
48         expected_parent = p;
49         p2 = talloc(p, char);
50         ok1(ext_alloc_count == 3);
51
52         expected_parent = head;
53         p = talloc_realloc(NULL, p, char, 1000);
54         ok1(ext_realloc_count == 1);
55         assert(p);
56
57         expected_parent = p;
58         talloc_free(p2);
59         ok1(ext_free_count == 1);
60
61         expected_parent = head;
62         talloc_free(p);
63         ok1(ext_free_count == 2);
64
65         expected_parent = NULL;
66         talloc_free(head);
67         ok1(ext_free_count == 3);
68
69         ok1(lock_count == unlock_count);
70
71         return exit_status();
72 }