tdb2: extend test/layout to be able to place in file.
authorRusty Russell <rusty@rustcorp.com.au>
Mon, 25 Oct 2010 00:08:55 +0000 (10:38 +1030)
committerRusty Russell <rusty@rustcorp.com.au>
Tue, 26 Oct 2010 08:48:42 +0000 (19:18 +1030)
This was for lockcheck, but that didn't work very well.  Seems like a useful
addition nonetheless.

ccan/tdb2/test/layout.c
ccan/tdb2/test/layout.h
ccan/tdb2/test/run-03-coalesce.c

index ab3837465477943a48be4b52887673331d0d26a1..27c2d0459d1439fdbe9af7cd4e8bc1866023b70c 100644 (file)
@@ -3,11 +3,13 @@
 #include <stdlib.h>
 #include <string.h>
 #include <assert.h>
+#include <err.h>
 #include "logging.h"
 
-struct tdb_layout *new_tdb_layout(void)
+struct tdb_layout *new_tdb_layout(const char *filename)
 {
        struct tdb_layout *layout = malloc(sizeof(*layout));
+       layout->filename = filename;
        layout->num_elems = 0;
        layout->elem = NULL;
        return layout;
@@ -302,5 +304,19 @@ struct tdb_context *tdb_layout_get(struct tdb_layout *layout)
 
        /* Write tailer. */
        ((uint8_t *)tdb->map_ptr)[tdb->map_size-1] = last_zone->zone_bits;
+
+       /* Get physical if they asked for it. */
+       if (layout->filename) {
+               int fd = open(layout->filename, O_WRONLY|O_TRUNC|O_CREAT,
+                             0600);
+               if (fd < 0)
+                       err(1, "opening %s for writing", layout->filename);
+               write(fd, tdb->map_ptr, tdb->map_size);
+               close(fd);
+               tdb_close(tdb);
+               /* NOMMAP is for lockcheck. */
+               tdb = tdb_open(layout->filename, TDB_NOMMAP, O_RDWR, 0,
+                              &tap_log_attr);
+       }
        return tdb;
 }
index 25edd8b777a4cd18e8c50522d8fbe69719df17b0..6cbf3d0d2f727bc0392a5f599bb0fedb9cea1834 100644 (file)
@@ -2,7 +2,7 @@
 #define TDB2_TEST_LAYOUT_H
 #include <ccan/tdb2/private.h>
 
-struct tdb_layout *new_tdb_layout(void);
+struct tdb_layout *new_tdb_layout(const char *filename);
 void tdb_layout_add_zone(struct tdb_layout *layout,
                         unsigned int zone_bits,
                         bool fill_prev);
@@ -61,6 +61,7 @@ union tdb_layout_elem {
 };
 
 struct tdb_layout {
+       const char *filename;
        unsigned int num_elems;
        union tdb_layout_elem *elem;
 };
index 529b6c7bf9625dfe980ef95a790d6fb686b16d6c..dc97137f3c62976392fd6025217b80b41d7db377 100644 (file)
@@ -37,7 +37,7 @@ int main(int argc, char *argv[])
        key.dsize = 5;
 
        /* No coalescing can be done due to EOF */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb = tdb_layout_get(layout);
        len = layout->elem[1].free.len;
@@ -57,7 +57,7 @@ int main(int argc, char *argv[])
        tdb_close(tdb);
 
        /* No coalescing can be done due to used record */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb_layout_add_free(layout, 1024);
        tdb_layout_add_used(layout, key, data, 6);
@@ -78,7 +78,7 @@ int main(int argc, char *argv[])
        tdb_close(tdb);
 
        /* Coalescing can be done due to two free records, then EOF */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb_layout_add_free(layout, 1024);
        tdb = tdb_layout_get(layout);
@@ -101,7 +101,7 @@ int main(int argc, char *argv[])
        tdb_close(tdb);
 
        /* Coalescing can be done due to two free records, then data */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb_layout_add_free(layout, 1024);
        tdb_layout_add_free(layout, 512);
@@ -125,7 +125,7 @@ int main(int argc, char *argv[])
        tdb_close(tdb);
 
        /* Coalescing can be done due to three free records, then EOF */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb_layout_add_free(layout, 1024);
        tdb_layout_add_free(layout, 512);
@@ -151,7 +151,7 @@ int main(int argc, char *argv[])
        tdb_close(tdb);
 
        /* Coalescing across two zones isn't possible. */
-       layout = new_tdb_layout();
+       layout = new_tdb_layout(NULL);
        tdb_layout_add_zone(layout, zone_bits, false);
        tdb_layout_add_zone(layout, zone_bits, true);
        tdb = tdb_layout_get(layout);