tal: tal_dup()
[ccan] / ccan / tal / test / run-overflow.c
1 #include <ccan/tal/tal.h>
2 #include <ccan/tal/tal.c>
3 #include <ccan/tap/tap.h>
4
5 static int error_count;
6
7 static void my_error(const char *msg)
8 {
9         error_count++;
10 }
11
12 int main(void)
13 {
14         void *p;
15         int *pi, *origpi;
16
17         plan_tests(26);
18
19         tal_set_backend(NULL, NULL, NULL, my_error);
20
21         p = tal_arr(NULL, int, (size_t)-1);
22         ok1(!p);
23         ok1(error_count == 1);
24
25         p = tal_arr(NULL, char, (size_t)-2);
26         ok1(!p);
27         ok1(error_count == 2);
28
29         /* Now try overflow cases for tal_dup. */
30         error_count = 0;
31         pi = origpi = tal_arr(NULL, int, 100);
32         ok1(pi);
33         ok1(error_count == 0);
34         pi = tal_dup(NULL, int, pi, (size_t)-1, 0);
35         ok1(!pi);
36         ok1(error_count == 1);
37         pi = tal_dup(NULL, int, pi, 0, (size_t)-1);
38         ok1(!pi);
39         ok1(error_count == 2);
40
41         pi = tal_dup(NULL, int, pi, (size_t)-1UL / sizeof(int),
42                      (size_t)-1UL / sizeof(int));
43         ok1(!pi);
44         ok1(error_count == 3);
45         /* This will still overflow when tal_hdr is added. */
46         pi = tal_dup(NULL, int, pi, (size_t)-1UL / sizeof(int) / 2,
47                      (size_t)-1UL / sizeof(int) / 2);
48         ok1(!pi);
49         ok1(error_count == 4);
50
51         /* Now, check that with TAL_TAKE we free old one on failure. */
52         pi = tal_arr(NULL, int, 100);
53         error_count = 0;
54         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1, 0);
55         ok1(!pi);
56         ok1(error_count == 1);
57         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
58
59         pi = tal_arr(NULL, int, 100);
60         error_count = 0;
61         pi = tal_dup(TAL_TAKE, int, pi, 0, (size_t)-1);
62         ok1(!pi);
63         ok1(error_count == 1);
64         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
65
66         pi = tal_arr(NULL, int, 100);
67         error_count = 0;
68         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1UL / sizeof(int),
69                      (size_t)-1UL / sizeof(int));
70         ok1(!pi);
71         ok1(error_count == 1);
72         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
73
74         pi = tal_arr(NULL, int, 100);
75         error_count = 0;
76         /* This will still overflow when tal_hdr is added. */
77         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1UL / sizeof(int) / 2,
78                      (size_t)-1UL / sizeof(int) / 2);
79         ok1(!pi);
80         ok1(error_count == 1);
81         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
82
83         return exit_status();
84 }