lib/tdb2/test/api-93-repack.c

   1 #include "config.h"
   2 #include "tdb2.h"
   3 #include "tap-interface.h"
   4 #include <sys/types.h>
   5 #include <sys/stat.h>
   6 #include <fcntl.h>
   7 #include "logging.h"
   8
   9 #define NUM_TESTS 1000
  10
  11 static bool store_all(struct tdb_context *tdb)
  12 {
  13         unsigned int i;
  14         struct tdb_data key = { (unsigned char *)&i, sizeof(i) };
  15         struct tdb_data dbuf = { (unsigned char *)&i, sizeof(i) };
  16
  17         for (i = 0; i < NUM_TESTS; i++) {
  18                 if (tdb_store(tdb, key, dbuf, TDB_INSERT) != TDB_SUCCESS)
  19                         return false;
  20         }
  21         return true;
  22 }
  23
  24 static int mark_entry(struct tdb_context *tdb,
  25                       TDB_DATA key, TDB_DATA data, bool found[])
  26 {
  27         unsigned int num;
  28
  29         if (key.dsize != sizeof(num))
  30                 return -1;
  31         memcpy(&num, key.dptr, key.dsize);
  32         if (num >= NUM_TESTS)
  33                 return -1;
  34         if (found[num])
  35                 return -1;
  36         found[num] = true;
  37         return 0;
  38 }
  39
  40 static bool is_all_set(bool found[], unsigned int num)
  41 {
  42         unsigned int i;
  43
  44         for (i = 0; i < num; i++)
  45                 if (!found[i])
  46                         return false;
  47         return true;
  48 }
  49
  50 int main(int argc, char *argv[])
  51 {
  52         unsigned int i;
  53         bool found[NUM_TESTS];
  54         struct tdb_context *tdb;
  55         int flags[] = { TDB_DEFAULT, TDB_NOMMAP,
  56                         TDB_CONVERT, TDB_NOMMAP|TDB_CONVERT,
  57                         TDB_VERSION1, TDB_VERSION1|TDB_NOMMAP,
  58                         TDB_VERSION1|TDB_CONVERT,
  59                         TDB_VERSION1|TDB_NOMMAP|TDB_CONVERT
  60         };
  61
  62         plan_tests(sizeof(flags) / sizeof(flags[0]) * 6 + 1);
  63
  64         for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
  65                 tdb = tdb_open("run-93-repack.tdb", flags[i],
  66                                O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
  67                 ok1(tdb);
  68                 if (!tdb)
  69                         break;
  70
  71                 ok1(store_all(tdb));
  72
  73                 ok1(tdb_repack(tdb) == TDB_SUCCESS);
  74                 memset(found, 0, sizeof(found));
  75                 ok1(tdb_check(tdb, NULL, NULL) == TDB_SUCCESS);
  76                 ok1(tdb_traverse(tdb, mark_entry, found) == NUM_TESTS);
  77                 ok1(is_all_set(found, NUM_TESTS));
  78                 tdb_close(tdb);
  79         }
  80
  81         ok1(tap_log_messages == 0);
  82         return exit_status();
  83 }