lib/ntdb/test/run-remap-in-read_traverse.c

   1 #include "ntdb-source.h"
   2 /* We had a bug where we marked the ntdb read-only for a ntdb_traverse_read.
   3  * If we then expanded the ntdb, we would remap read-only, and later SEGV. */
   4 #include "tap-interface.h"
   5 #include "external-agent.h"
   6 #include "logging.h"
   7
   8 static bool file_larger(int fd, ntdb_len_t size)
   9 {
  10         struct stat st;
  11
  12         fstat(fd, &st);
  13         return st.st_size != size;
  14 }
  15
  16 static unsigned add_records_to_grow(struct agent *agent, int fd, ntdb_len_t size)
  17 {
  18         unsigned int i;
  19
  20         for (i = 0; !file_larger(fd, size); i++) {
  21                 char data[50];
  22                 sprintf(data, "%i=%i", i, i);
  23                 if (external_agent_operation(agent, STORE, data) != SUCCESS)
  24                         return 0;
  25         }
  26         diag("Added %u records to grow file", i);
  27         return i;
  28 }
  29
  30 int main(int argc, char *argv[])
  31 {
  32         unsigned int i;
  33         struct agent *agent;
  34         struct ntdb_context *ntdb;
  35         NTDB_DATA d = ntdb_mkdata("hello", 5);
  36         const char filename[] = "run-remap-in-read_traverse.ntdb";
  37
  38         plan_tests(4);
  39
  40         agent = prepare_external_agent();
  41
  42         ntdb = ntdb_open(filename, MAYBE_NOSYNC,
  43                        O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
  44
  45         ok1(external_agent_operation(agent, OPEN, filename) == SUCCESS);
  46         i = add_records_to_grow(agent, ntdb->file->fd, ntdb->file->map_size);
  47
  48         /* Do a traverse. */
  49         ok1(ntdb_traverse(ntdb, NULL, NULL) == i);
  50
  51         /* Now store something! */
  52         ok1(ntdb_store(ntdb, d, d, NTDB_INSERT) == 0);
  53         ok1(tap_log_messages == 0);
  54         ntdb_close(ntdb);
  55         free_external_agent(agent);
  56         return exit_status();
  57 }