git.ozlabs.org Git - ccan/blob - ccan/tdb/test/run-no-lock-during-traverse.c

   1 #define _XOPEN_SOURCE 500
   2 #include <unistd.h>
   3 #include "lock-tracking.h"
   4
   5 #define fcntl fcntl_with_lockcheck
   6
   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/tdb/hash.c>
  18 #include <ccan/tap/tap.h>
  19 #include <stdlib.h>
  20 #include <err.h>
  21 #include "logging.h"
  22
  23 #undef fcntl
  24
  25 #define NUM_ENTRIES 10
  26
  27 static bool prepare_entries(struct tdb_context *tdb)
  28 {
  29         unsigned int i;
  30         TDB_DATA key, data;
  31
  32         for (i = 0; i < NUM_ENTRIES; i++) {
  33                 key.dsize = sizeof(i);
  34                 key.dptr = (void *)&i;
  35                 data.dsize = strlen("world");
  36                 data.dptr = (void *)"world";
  37
  38                 if (tdb_store(tdb, key, data, 0) != 0)
  39                         return false;
  40         }
  41         return true;
  42 }
  43
  44 static void delete_entries(struct tdb_context *tdb)
  45 {
  46         unsigned int i;
  47         TDB_DATA key;
  48
  49         for (i = 0; i < NUM_ENTRIES; i++) {
  50                 key.dsize = sizeof(i);
  51                 key.dptr = (void *)&i;
  52
  53                 ok1(tdb_delete(tdb, key) == 0);
  54         }
  55 }
  56
  57 /* We don't know how many times this will run. */
  58 static int delete_other(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data,
  59                         void *private_data)
  60 {
  61         unsigned int i;
  62         memcpy(&i, key.dptr, 4);
  63         i = (i + 1) % NUM_ENTRIES;
  64         key.dptr = (void *)&i;
  65         if (tdb_delete(tdb, key) != 0)
  66                 (*(int *)private_data)++;
  67         return 0;
  68 }
  69
  70 static int delete_self(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data,
  71                         void *private_data)
  72 {
  73         ok1(tdb_delete(tdb, key) == 0);
  74         return 0;
  75 }
  76
  77 int main(int argc, char *argv[])
  78 {
  79         struct tdb_context *tdb;
  80         int errors = 0;
  81
  82         plan_tests(41);
  83         tdb = tdb_open_ex("run-no-lock-during-traverse.tdb",
  84                           1024, TDB_CLEAR_IF_FIRST, O_CREAT|O_TRUNC|O_RDWR,
  85                           0600, &taplogctx, NULL);
  86
  87         ok1(tdb);
  88         ok1(prepare_entries(tdb));
  89         ok1(locking_errors == 0);
  90         ok1(tdb_lockall(tdb) == 0);
  91         ok1(locking_errors == 0);
  92         tdb_traverse(tdb, delete_other, &errors);
  93         ok1(errors == 0);
  94         ok1(locking_errors == 0);
  95         ok1(tdb_unlockall(tdb) == 0);
  96
  97         ok1(prepare_entries(tdb));
  98         ok1(locking_errors == 0);
  99         ok1(tdb_lockall(tdb) == 0);
 100         ok1(locking_errors == 0);
 101         tdb_traverse(tdb, delete_self, NULL);
 102         ok1(locking_errors == 0);
 103         ok1(tdb_unlockall(tdb) == 0);
 104
 105         ok1(prepare_entries(tdb));
 106         ok1(locking_errors == 0);
 107         ok1(tdb_lockall(tdb) == 0);
 108         ok1(locking_errors == 0);
 109         delete_entries(tdb);
 110         ok1(locking_errors == 0);
 111         ok1(tdb_unlockall(tdb) == 0);
 112
 113         ok1(tdb_close(tdb) == 0);
 114
 115         return exit_status();
 116 }