]> git.ozlabs.org Git - ccan/commitdiff
tdb2: enforce TDB_CONVERT
authorRusty Russell <rusty@rustcorp.com.au>
Wed, 31 Aug 2011 01:14:12 +0000 (10:44 +0930)
committerRusty Russell <rusty@rustcorp.com.au>
Wed, 31 Aug 2011 01:14:12 +0000 (10:44 +0930)
If the caller actually specifies TDB_CONVERT, fail if the TDB does not
need conversion.  It's unusual for the user to specify TDB_CONVERT
(it's auto-detected) but better to be strict.

ccan/tdb2/open.c
ccan/tdb2/test/logging.c
ccan/tdb2/test/logging.h
ccan/tdb2/test/run-35-convert.c [new file with mode: 0644]

index fec3ef1596a562c90c6c40588265b0e05c93ed20..96b5a5ea3f6a2b73b265e0f69ef7826aff27d207 100644 (file)
@@ -532,6 +532,12 @@ struct tdb_context *tdb_open(const char *name, int tdb_flags,
                                           name, (long long)hdr.version);
                        goto fail;
                }
                                           name, (long long)hdr.version);
                        goto fail;
                }
+       } else if (tdb->flags & TDB_CONVERT) {
+               ecode = tdb_logerr(tdb, TDB_ERR_IO, TDB_LOG_ERROR,
+                                  "tdb_open:"
+                                  " %s does not need TDB_CONVERT",
+                                  name);
+               goto fail;
        }
 
        tdb_convert(tdb, &hdr, sizeof(hdr));
        }
 
        tdb_convert(tdb, &hdr, sizeof(hdr));
index e36e2b8818d5ce41ffe9a9e458a0d5b60f6e17d1..4857a4bfad7896e50272e52ab3ff0ab84e0d12a7 100644 (file)
@@ -5,6 +5,7 @@
 
 unsigned tap_log_messages;
 const char *log_prefix = "";
 
 unsigned tap_log_messages;
 const char *log_prefix = "";
+char *log_last = NULL;
 bool suppress_logging;
 
 union tdb_attribute tap_log_attr = {
 bool suppress_logging;
 
 union tdb_attribute tap_log_attr = {
@@ -20,6 +21,9 @@ void tap_log_fn(struct tdb_context *tdb,
                return;
 
        diag("tdb log level %u: %s%s", level, log_prefix, message);
                return;
 
        diag("tdb log level %u: %s%s", level, log_prefix, message);
+       if (log_last)
+               free(log_last);
+       log_last = strdup(message);
        tap_log_messages++;
 }
 
        tap_log_messages++;
 }
 
index d172f867fd218f4984d5061631bda9c3cb57ce4c..0fd59ba6bb921ed0a4e70fddda2c00aa69d24de3 100644 (file)
@@ -8,6 +8,7 @@ extern bool suppress_logging;
 extern const char *log_prefix;
 extern unsigned tap_log_messages;
 extern union tdb_attribute tap_log_attr;
 extern const char *log_prefix;
 extern unsigned tap_log_messages;
 extern union tdb_attribute tap_log_attr;
+extern char *log_last;
 
 void tap_log_fn(struct tdb_context *tdb,
                enum tdb_log_level level,
 
 void tap_log_fn(struct tdb_context *tdb,
                enum tdb_log_level level,
diff --git a/ccan/tdb2/test/run-35-convert.c b/ccan/tdb2/test/run-35-convert.c
new file mode 100644 (file)
index 0000000..0300369
--- /dev/null
@@ -0,0 +1,61 @@
+#include <ccan/tdb2/private.h>
+#include <ccan/failtest/failtest_override.h>
+#include <ccan/tdb2/tdb.c>
+#include <ccan/tdb2/open.c>
+#include <ccan/tdb2/free.c>
+#include <ccan/tdb2/lock.c>
+#include <ccan/tdb2/io.c>
+#include <ccan/tdb2/hash.c>
+#include <ccan/tdb2/transaction.c>
+#include <ccan/tdb2/check.c>
+#include <ccan/tap/tap.h>
+#include <ccan/failtest/failtest.h>
+#include "logging.h"
+#include "failtest_helper.h"
+
+int main(int argc, char *argv[])
+{
+       unsigned int i, messages = 0;
+       struct tdb_context *tdb;
+       int flags[] = { TDB_DEFAULT, TDB_NOMMAP,
+                       TDB_CONVERT, TDB_NOMMAP|TDB_CONVERT };
+
+       failtest_init(argc, argv);
+       failtest_hook = block_repeat_failures;
+       failtest_exit_check = exit_check_log;
+       plan_tests(sizeof(flags) / sizeof(flags[0]) * 4);
+       for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
+               tdb = tdb_open("run-35-convert.tdb", flags[i],
+                              O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
+               if (!ok1(tdb))
+                       failtest_exit(exit_status());
+
+               tdb_close(tdb);
+               /* If we say TDB_CONVERT, it must be converted */
+               tdb = tdb_open("run-35-convert.tdb",
+                              flags[i]|TDB_CONVERT,
+                              O_RDWR, 0600, &tap_log_attr);
+               if (flags[i] & TDB_CONVERT) {
+                       if (!tdb)
+                               failtest_exit(exit_status());
+                       ok1(tdb_get_flags(tdb) & TDB_CONVERT);
+                       tdb_close(tdb);
+               } else {
+                       if (!ok1(!tdb && errno == EIO))
+                               failtest_exit(exit_status());
+                       ok1(tap_log_messages == ++messages);
+                       if (!ok1(log_last && strstr(log_last, "TDB_CONVERT")))
+                               failtest_exit(exit_status());
+               }
+
+               /* If don't say TDB_CONVERT, it *may* be converted */
+               tdb = tdb_open("run-35-convert.tdb",
+                              flags[i] & ~TDB_CONVERT,
+                              O_RDWR, 0600, &tap_log_attr);
+               if (!tdb)
+                       failtest_exit(exit_status());
+               ok1(tdb_get_flags(tdb) == flags[i]);
+               tdb_close(tdb);
+       }
+       failtest_exit(exit_status());
+}