git.ozlabs.org Git - ccan/blob - junkcode/rusty@rustcorp.com.au-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 #include "helprun-external-agent.h"
   8
   9 static bool file_larger(int fd, ntdb_len_t size)
  10 {
  11         struct stat st;
  12
  13         fstat(fd, &st);
  14         return st.st_size != size;
  15 }
  16
  17 static unsigned add_records_to_grow(struct agent *agent, int fd, ntdb_len_t size)
  18 {
  19         unsigned int i;
  20
  21         for (i = 0; !file_larger(fd, size); i++) {
  22                 char data[50];
  23                 sprintf(data, "%i=%i", i, i);
  24                 if (external_agent_operation(agent, STORE, data) != SUCCESS)
  25                         return 0;
  26         }
  27         diag("Added %u records to grow file", i);
  28         return i;
  29 }
  30
  31 int main(int argc, char *argv[])
  32 {
  33         unsigned int i;
  34         struct agent *agent;
  35         struct ntdb_context *ntdb;
  36         NTDB_DATA d = ntdb_mkdata("hello", 5);
  37         const char filename[] = "run-remap-in-read_traverse.ntdb";
  38
  39         plan_tests(4);
  40
  41         agent = prepare_external_agent();
  42
  43         ntdb = ntdb_open(filename, MAYBE_NOSYNC,
  44                        O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
  45
  46         ok1(external_agent_operation(agent, OPEN, filename) == SUCCESS);
  47         i = add_records_to_grow(agent, ntdb->file->fd, ntdb->file->map_size);
  48
  49         /* Do a traverse. */
  50         ok1(ntdb_traverse(ntdb, NULL, NULL) == i);
  51
  52         /* Now store something! */
  53         ok1(ntdb_store(ntdb, d, d, NTDB_INSERT) == 0);
  54         ok1(tap_log_messages == 0);
  55         ntdb_close(ntdb);
  56         free_external_agent(agent);
  57         return exit_status();
  58 }