tdb2: don't start again when we coalesce a record.
[ccan] / ccan / tdb2 / test / run-03-coalesce.c
1 #include <ccan/tdb2/tdb.c>
2 #include <ccan/tdb2/open.c>
3 #include <ccan/tdb2/free.c>
4 #include <ccan/tdb2/lock.c>
5 #include <ccan/tdb2/io.c>
6 #include <ccan/tdb2/hash.c>
7 #include <ccan/tdb2/check.c>
8 #include <ccan/tdb2/transaction.c>
9 #include <ccan/tap/tap.h>
10 #include "logging.h"
11 #include "layout.h"
12
13 static tdb_len_t free_record_length(struct tdb_context *tdb, tdb_off_t off)
14 {
15         struct tdb_free_record f;
16         enum TDB_ERROR ecode;
17
18         ecode = tdb_read_convert(tdb, off, &f, sizeof(f));
19         if (ecode != TDB_SUCCESS)
20                 return ecode;
21         if (frec_magic(&f) != TDB_FREE_MAGIC)
22                 return TDB_ERR_CORRUPT;
23         return frec_len(&f);
24 }
25
26 int main(int argc, char *argv[])
27 {
28         tdb_off_t b_off, test;
29         struct tdb_context *tdb;
30         struct tdb_layout *layout;
31         struct tdb_data data, key;
32         tdb_len_t len;
33
34         /* FIXME: Test TDB_CONVERT */
35         /* FIXME: Test lock order fail. */
36
37         plan_tests(42);
38         data = tdb_mkdata("world", 5);
39         key = tdb_mkdata("hello", 5);
40
41         /* No coalescing can be done due to EOF */
42         layout = new_tdb_layout("run-03-coalesce.tdb");
43         tdb_layout_add_freetable(layout);
44         len = 1024;
45         tdb_layout_add_free(layout, len, 0);
46         tdb = tdb_layout_get(layout);
47         ok1(tdb_check(tdb, NULL, NULL) == 0);
48         ok1(free_record_length(tdb, layout->elem[1].base.off) == len);
49
50         /* Figure out which bucket free entry is. */
51         b_off = bucket_off(tdb->ftable_off, size_to_bucket(len));
52         /* Lock and fail to coalesce. */
53         ok1(tdb_lock_free_bucket(tdb, b_off, TDB_LOCK_WAIT) == 0);
54         test = layout->elem[1].base.off;
55         ok1(coalesce(tdb, layout->elem[1].base.off, b_off, len, &test, &test)
56             == 0);
57         tdb_unlock_free_bucket(tdb, b_off);
58         ok1(free_record_length(tdb, layout->elem[1].base.off) == len);
59         ok1(test == layout->elem[1].base.off);
60         ok1(tdb_check(tdb, NULL, NULL) == 0);
61         tdb_close(tdb);
62         tdb_layout_free(layout);
63
64         /* No coalescing can be done due to used record */
65         layout = new_tdb_layout("run-03-coalesce.tdb");
66         tdb_layout_add_freetable(layout);
67         tdb_layout_add_free(layout, 1024, 0);
68         tdb_layout_add_used(layout, key, data, 6);
69         tdb = tdb_layout_get(layout);
70         ok1(free_record_length(tdb, layout->elem[1].base.off) == 1024);
71         ok1(tdb_check(tdb, NULL, NULL) == 0);
72
73         /* Figure out which bucket free entry is. */
74         b_off = bucket_off(tdb->ftable_off, size_to_bucket(1024));
75         /* Lock and fail to coalesce. */
76         ok1(tdb_lock_free_bucket(tdb, b_off, TDB_LOCK_WAIT) == 0);
77         test = layout->elem[1].base.off;
78         ok1(coalesce(tdb, layout->elem[1].base.off, b_off, 1024, &test, &test)
79             == 0);
80         tdb_unlock_free_bucket(tdb, b_off);
81         ok1(free_record_length(tdb, layout->elem[1].base.off) == 1024);
82         ok1(test == layout->elem[1].base.off);
83         ok1(tdb_check(tdb, NULL, NULL) == 0);
84         tdb_close(tdb);
85         tdb_layout_free(layout);
86
87         /* Coalescing can be done due to two free records, then EOF */
88         layout = new_tdb_layout("run-03-coalesce.tdb");
89         tdb_layout_add_freetable(layout);
90         tdb_layout_add_free(layout, 1024, 0);
91         tdb_layout_add_free(layout, 2048, 0);
92         tdb = tdb_layout_get(layout);
93         ok1(free_record_length(tdb, layout->elem[1].base.off) == 1024);
94         ok1(free_record_length(tdb, layout->elem[2].base.off) == 2048);
95         ok1(tdb_check(tdb, NULL, NULL) == 0);
96
97         /* Figure out which bucket (first) free entry is. */
98         b_off = bucket_off(tdb->ftable_off, size_to_bucket(1024));
99         /* Lock and coalesce. */
100         ok1(tdb_lock_free_bucket(tdb, b_off, TDB_LOCK_WAIT) == 0);
101         test = layout->elem[2].base.off;
102         ok1(coalesce(tdb, layout->elem[1].base.off, b_off, 1024, &test, &test)
103             == 1024 + sizeof(struct tdb_used_record) + 2048);
104         /* Should tell us it's erased this one... */
105         ok1(test == TDB_ERR_NOEXIST);
106         ok1(tdb->file->allrecord_lock.count == 0 && tdb->file->num_lockrecs == 0);
107         ok1(free_record_length(tdb, layout->elem[1].base.off)
108             == 1024 + sizeof(struct tdb_used_record) + 2048);
109         ok1(tdb_check(tdb, NULL, NULL) == 0);
110         tdb_close(tdb);
111         tdb_layout_free(layout);
112
113         /* Coalescing can be done due to two free records, then data */
114         layout = new_tdb_layout("run-03-coalesce.tdb");
115         tdb_layout_add_freetable(layout);
116         tdb_layout_add_free(layout, 1024, 0);
117         tdb_layout_add_free(layout, 512, 0);
118         tdb_layout_add_used(layout, key, data, 6);
119         tdb = tdb_layout_get(layout);
120         ok1(free_record_length(tdb, layout->elem[1].base.off) == 1024);
121         ok1(free_record_length(tdb, layout->elem[2].base.off) == 512);
122         ok1(tdb_check(tdb, NULL, NULL) == 0);
123
124         /* Figure out which bucket free entry is. */
125         b_off = bucket_off(tdb->ftable_off, size_to_bucket(1024));
126         /* Lock and coalesce. */
127         ok1(tdb_lock_free_bucket(tdb, b_off, TDB_LOCK_WAIT) == 0);
128         test = layout->elem[2].base.off;
129         ok1(coalesce(tdb, layout->elem[1].base.off, b_off, 1024, &test, &test)
130             == 1024 + sizeof(struct tdb_used_record) + 512);
131         ok1(tdb->file->allrecord_lock.count == 0 && tdb->file->num_lockrecs == 0);
132         ok1(free_record_length(tdb, layout->elem[1].base.off)
133             == 1024 + sizeof(struct tdb_used_record) + 512);
134         ok1(test == TDB_ERR_NOEXIST);
135         ok1(tdb_check(tdb, NULL, NULL) == 0);
136         tdb_close(tdb);
137         tdb_layout_free(layout);
138
139         /* Coalescing can be done due to three free records, then EOF */
140         layout = new_tdb_layout("run-03-coalesce.tdb");
141         tdb_layout_add_freetable(layout);
142         tdb_layout_add_free(layout, 1024, 0);
143         tdb_layout_add_free(layout, 512, 0);
144         tdb_layout_add_free(layout, 256, 0);
145         tdb = tdb_layout_get(layout);
146         ok1(free_record_length(tdb, layout->elem[1].base.off) == 1024);
147         ok1(free_record_length(tdb, layout->elem[2].base.off) == 512);
148         ok1(free_record_length(tdb, layout->elem[3].base.off) == 256);
149         ok1(tdb_check(tdb, NULL, NULL) == 0);
150
151         /* Figure out which bucket free entry is. */
152         b_off = bucket_off(tdb->ftable_off, size_to_bucket(1024));
153         /* Lock and coalesce. */
154         ok1(tdb_lock_free_bucket(tdb, b_off, TDB_LOCK_WAIT) == 0);
155         test = layout->elem[2].base.off;
156         ok1(coalesce(tdb, layout->elem[1].base.off, b_off, 1024, &test, &test)
157             == 1024 + sizeof(struct tdb_used_record) + 512
158             + sizeof(struct tdb_used_record) + 256);
159         ok1(tdb->file->allrecord_lock.count == 0
160             && tdb->file->num_lockrecs == 0);
161         ok1(free_record_length(tdb, layout->elem[1].base.off)
162             == 1024 + sizeof(struct tdb_used_record) + 512
163             + sizeof(struct tdb_used_record) + 256);
164         ok1(tdb_check(tdb, NULL, NULL) == 0);
165         tdb_close(tdb);
166         tdb_layout_free(layout);
167
168         ok1(tap_log_messages == 0);
169         return exit_status();
170 }