tdb2: fix bucket search
[ccan] / ccan / tdb2 / test / run-01-zones.c
1 #include <ccan/tdb2/tdb.c>
2 #include <ccan/tdb2/free.c>
3 #include <ccan/tdb2/lock.c>
4 #include <ccan/tdb2/io.c>
5 #include <ccan/tdb2/hash.c>
6 #include <ccan/tdb2/check.c>
7 #include <ccan/tap/tap.h>
8 #include "logging.h"
9 #include "layout.h"
10
11 /* Calculate start of zone offset from layout directly. */
12 static tdb_off_t layout_zone_off(tdb_off_t off, struct tdb_layout *layout)
13 {
14         unsigned int i;
15
16         /* Every second one is a free entry, so divide by 2 to get zone */
17         for (i = 0; i < layout->num_elems; i++) {
18                 if (layout->elem[i].base.type != ZONE)
19                         continue;
20                 if (layout->elem[i].base.off
21                     + (1ULL << layout->elem[i].zone.zone_bits) > off)
22                         return layout->elem[i].base.off;
23         }
24         abort();
25 }
26
27 int main(int argc, char *argv[])
28 {
29         struct tdb_context *tdb;
30         struct tdb_layout *layout;
31         struct free_zone_header zhdr;
32         tdb_off_t off, step;
33         unsigned int i;
34
35         /* FIXME: Test TDB_CONVERT */
36
37         plan_tests(3 + 100);
38
39         /* No coalescing can be done due to EOF */
40         layout = new_tdb_layout(NULL);
41         tdb_layout_add_zone(layout, INITIAL_ZONE_BITS, false);
42         tdb_layout_add_zone(layout, INITIAL_ZONE_BITS, true);
43         tdb_layout_add_zone(layout, INITIAL_ZONE_BITS+1, true);
44         tdb_layout_add_zone(layout, INITIAL_ZONE_BITS+2, true);
45         tdb_layout_add_zone(layout, INITIAL_ZONE_BITS+2, true);
46         tdb = tdb_layout_get(layout);
47
48         ok1(tdb_check(tdb, NULL, NULL) == 0);
49
50         /* Last zone should get right zone. */
51         ok1(last_zone(tdb, &zhdr)
52             == layout->elem[layout->num_elems-1].base.off);
53         ok1(zhdr.zone_bits == INITIAL_ZONE_BITS+2);
54
55         off = sizeof(struct tdb_header);
56         step = (tdb->map_size - 1 - off) / 100;
57         for (i = 0; i < 100; i++, off += step) {
58                 ok1(off_to_zone(tdb, off, &zhdr) == layout_zone_off(off, layout));
59         }
60
61         return exit_status();
62 }