tdb2: make tdb_check typesafe.

[ccan] / ccan / tdb2 / tdb.c

diff --git a/ccan/tdb2/tdb.c b/ccan/tdb2/tdb.c
index f6ca01657518af29c1111700c530abda29ba7d43..ce431208bf4cb4fa28d09f9158c7fb02c93d7d37 100644 (file)
--- a/ccan/tdb2/tdb.c
+++ b/ccan/tdb2/tdb.c
@@ -1,4 +1,5 @@
 #include "private.h"
+#include <ccan/asprintf/asprintf.h>
 #include <ccan/tdb2/tdb2.h>
 #include <assert.h>
 #include <stdarg.h>
@@ -236,6 +237,13 @@ struct tdb_context *tdb_open(const char *name, int tdb_flags,
                attr = attr->base.next;
        }
 
+       if (tdb_flags & ~(TDB_INTERNAL | TDB_NOLOCK | TDB_NOMMAP | TDB_CONVERT
+                         | TDB_NOSYNC)) {
+               ecode = tdb_logerr(tdb, TDB_ERR_EINVAL, TDB_LOG_USE_ERROR,
+                                  "tdb_open: unknown flags %u", tdb_flags);
+               goto fail;
+       }
+
        if ((open_flags & O_ACCMODE) == O_WRONLY) {
                ecode = tdb_logerr(tdb, TDB_ERR_EINVAL, TDB_LOG_USE_ERROR,
                                   "tdb_open: can't open tdb %s write-only",
@@ -756,23 +764,18 @@ enum TDB_ERROR COLD tdb_logerr(struct tdb_context *tdb,
        if (!tdb->logfn)
                return ecode;
 
-       /* FIXME: Doesn't assume asprintf. */
        va_start(ap, fmt);
-       len = vsnprintf(NULL, 0, fmt, ap);
+       len = vasprintf(&message, fmt, ap);
        va_end(ap);
 
-       message = malloc(len + 1);
-       if (!message) {
+       if (len < 0) {
                tdb->logfn(tdb, TDB_LOG_ERROR, tdb->log_private,
                           "out of memory formatting message:");
                tdb->logfn(tdb, level, tdb->log_private, fmt);
-               return ecode;
+       } else {
+               tdb->logfn(tdb, level, tdb->log_private, message);
+               free(message);
        }
-       va_start(ap, fmt);
-       len = vsprintf(message, fmt, ap);
-       va_end(ap);
-       tdb->logfn(tdb, level, tdb->log_private, message);
-       free(message);
        errno = saved_errno;
        return ecode;
 }