From a15c1cf175d7f29a50d7b2c1acb0f67faab1b06f Mon Sep 17 00:00:00 2001 From: Rusty Russell Date: Wed, 14 Sep 2011 08:03:13 +0930 Subject: tdb2: tdb_repack Move the tdb1_repack() code into the core, make it general, rename to tdb_repack(). It's generic code: copy database into temporary, wipe it, copy back. Signed-off-by: Rusty Russell (Imported from CCAN commit e487983a4099b6f760056ff7182f2ff543e6da71) --- lib/tdb2/test/run-93-repack.c | 76 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 76 insertions(+) create mode 100644 lib/tdb2/test/run-93-repack.c (limited to 'lib/tdb2/test') diff --git a/lib/tdb2/test/run-93-repack.c b/lib/tdb2/test/run-93-repack.c new file mode 100644 index 0000000000..906a17b4b3 --- /dev/null +++ b/lib/tdb2/test/run-93-repack.c @@ -0,0 +1,76 @@ +#include "tdb2-source.h" +#include +#include "logging.h" + +#define NUM_TESTS 50000 + +static bool store_all(struct tdb_context *tdb) +{ + unsigned int i; + struct tdb_data key = { (unsigned char *)&i, sizeof(i) }; + struct tdb_data dbuf = { (unsigned char *)&i, sizeof(i) }; + + for (i = 0; i < NUM_TESTS; i++) { + if (tdb_store(tdb, key, dbuf, TDB_INSERT) != TDB_SUCCESS) + return false; + } + return true; +} + +static int mark_entry(struct tdb_context *tdb, + TDB_DATA key, TDB_DATA data, bool found[]) +{ + unsigned int num; + + if (key.dsize != sizeof(num)) + return -1; + memcpy(&num, key.dptr, key.dsize); + if (num >= NUM_TESTS) + return -1; + if (found[num]) + return -1; + found[num] = true; + return 0; +} + +static bool is_all_set(bool found[], unsigned int num) +{ + unsigned int i; + + for (i = 0; i < num; i++) + if (!found[i]) + return false; + return true; +} + +int main(int argc, char *argv[]) +{ + unsigned int i; + bool found[NUM_TESTS]; + struct tdb_context *tdb; + int flags[] = { TDB_DEFAULT, TDB_NOMMAP, + TDB_CONVERT, TDB_NOMMAP|TDB_CONVERT, + }; + + plan_tests(sizeof(flags) / sizeof(flags[0]) * 6 + 1); + + for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) { + tdb = tdb_open("run-93-repack.tdb", flags[i], + O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr); + ok1(tdb); + if (!tdb) + break; + + ok1(store_all(tdb)); + + ok1(tdb_repack(tdb) == TDB_SUCCESS); + memset(found, 0, sizeof(found)); + ok1(tdb_check(tdb, NULL, NULL) == TDB_SUCCESS); + ok1(tdb_traverse(tdb, mark_entry, found) == NUM_TESTS); + ok1(is_all_set(found, NUM_TESTS)); + tdb_close(tdb); + } + + ok1(tap_log_messages == 0); + return exit_status(); +} -- cgit