1 #define _XOPEN_SOURCE 500
3 #include "lock-tracking.h"
5 #define fcntl fcntl_with_lockcheck
7 #include <ccan/tdb/tdb.h>
8 #include <ccan/tdb/io.c>
9 #include <ccan/tdb/tdb.c>
10 #include <ccan/tdb/lock.c>
11 #include <ccan/tdb/freelist.c>
12 #include <ccan/tdb/traverse.c>
13 #include <ccan/tdb/transaction.c>
14 #include <ccan/tdb/error.c>
15 #include <ccan/tdb/open.c>
16 #include <ccan/tdb/check.c>
17 #include <ccan/tap/tap.h>
24 #define NUM_ENTRIES 10
26 static bool prepare_entries(struct tdb_context *tdb)
31 for (i = 0; i < NUM_ENTRIES; i++) {
32 key.dsize = sizeof(i);
33 key.dptr = (void *)&i;
34 data.dsize = strlen("world");
35 data.dptr = (void *)"world";
37 if (tdb_store(tdb, key, data, 0) != 0)
43 static void delete_entries(struct tdb_context *tdb)
48 for (i = 0; i < NUM_ENTRIES; i++) {
49 key.dsize = sizeof(i);
50 key.dptr = (void *)&i;
52 ok1(tdb_delete(tdb, key) == 0);
56 /* We don't know how many times this will run. */
57 static int delete_other(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data,
61 memcpy(&i, key.dptr, 4);
62 i = (i + 1) % NUM_ENTRIES;
63 key.dptr = (void *)&i;
64 if (tdb_delete(tdb, key) != 0)
65 (*(int *)private_data)++;
69 static int delete_self(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data,
72 ok1(tdb_delete(tdb, key) == 0);
76 int main(int argc, char *argv[])
78 struct tdb_context *tdb;
82 tdb = tdb_open_ex("run-no-lock-during-traverse.tdb",
83 1024, TDB_CLEAR_IF_FIRST, O_CREAT|O_TRUNC|O_RDWR,
84 0600, &taplogctx, NULL);
87 ok1(prepare_entries(tdb));
88 ok1(locking_errors == 0);
89 ok1(tdb_lockall(tdb) == 0);
90 ok1(locking_errors == 0);
91 tdb_traverse(tdb, delete_other, &errors);
93 ok1(locking_errors == 0);
94 ok1(tdb_unlockall(tdb) == 0);
96 ok1(prepare_entries(tdb));
97 ok1(locking_errors == 0);
98 ok1(tdb_lockall(tdb) == 0);
99 ok1(locking_errors == 0);
100 tdb_traverse(tdb, delete_self, NULL);
101 ok1(locking_errors == 0);
102 ok1(tdb_unlockall(tdb) == 0);
104 ok1(prepare_entries(tdb));
105 ok1(locking_errors == 0);
106 ok1(tdb_lockall(tdb) == 0);
107 ok1(locking_errors == 0);
109 ok1(locking_errors == 0);
110 ok1(tdb_unlockall(tdb) == 0);
112 ok1(tdb_close(tdb) == 0);
114 return exit_status();