From 34c2d1658a89462e9b34210cb19fe9ab33bc2194 Mon Sep 17 00:00:00 2001 From: Rusty Russell Date: Wed, 14 Sep 2011 07:09:13 +0930 Subject: tdb2: cleanup oob handling. The tdb_oob() function can fail due to errors, as well as because the length asked for is greater than the size of the file. Clean that up: (1) If probe is true, only fail if there's an error, not if the length is too great. (2) Exit tdb_open() if it tdb_oob() probe fails; this helps cut down test time for failtest. (3) Don't set probe to true in tdb_direct() fail; a minor issue, but it means we log failure. (Imported from CCAN commit 77658070a3e4f712b94d659b2e399031ce3394c8) Signed-off-by: Rusty Russell --- lib/tdb2/io.c | 38 +++++++++++++++++++++----------------- 1 file changed, 21 insertions(+), 17 deletions(-) (limited to 'lib/tdb2/io.c') diff --git a/lib/tdb2/io.c b/lib/tdb2/io.c index 8c5f45f308..4166cd4c67 100644 --- a/lib/tdb2/io.c +++ b/lib/tdb2/io.c @@ -70,7 +70,9 @@ void tdb_mmap(struct tdb_context *tdb) /* check for an out of bounds access - if it is out of bounds then see if the database has been expanded by someone else and expand if necessary - note that "len" is the minimum length needed for the db + note that "len" is the minimum length needed for the db. + + If probe is true, len being too large isn't a failure. */ static enum TDB_ERROR tdb_oob(struct tdb_context *tdb, tdb_off_t len, bool probe) @@ -84,15 +86,16 @@ static enum TDB_ERROR tdb_oob(struct tdb_context *tdb, tdb_off_t len, || tdb_has_expansion_lock(tdb)); if (len <= tdb->file->map_size) - return 0; + return TDB_SUCCESS; if (tdb->flags & TDB_INTERNAL) { - if (!probe) { - tdb_logerr(tdb, TDB_ERR_IO, TDB_LOG_ERROR, - "tdb_oob len %lld beyond internal" - " malloc size %lld", - (long long)len, - (long long)tdb->file->map_size); - } + if (probe) + return TDB_SUCCESS; + + tdb_logerr(tdb, TDB_ERR_IO, TDB_LOG_ERROR, + "tdb_oob len %lld beyond internal" + " malloc size %lld", + (long long)len, + (long long)tdb->file->map_size); return TDB_ERR_IO; } @@ -111,11 +114,12 @@ static enum TDB_ERROR tdb_oob(struct tdb_context *tdb, tdb_off_t len, tdb_unlock_expand(tdb, F_RDLCK); if (st.st_size < (size_t)len) { - if (!probe) { - tdb_logerr(tdb, TDB_ERR_IO, TDB_LOG_ERROR, - "tdb_oob len %zu beyond eof at %zu", - (size_t)len, st.st_size); - } + if (probe) + return TDB_SUCCESS; + + tdb_logerr(tdb, TDB_ERR_IO, TDB_LOG_ERROR, + "tdb_oob len %zu beyond eof at %zu", + (size_t)len, st.st_size); return TDB_ERR_IO; } @@ -242,7 +246,7 @@ static enum TDB_ERROR tdb_write(struct tdb_context *tdb, tdb_off_t off, "Write to read-only database"); } - ecode = tdb->methods->oob(tdb, off + len, 0); + ecode = tdb->methods->oob(tdb, off + len, false); if (ecode != TDB_SUCCESS) { return ecode; } @@ -272,7 +276,7 @@ static enum TDB_ERROR tdb_read(struct tdb_context *tdb, tdb_off_t off, { enum TDB_ERROR ecode; - ecode = tdb->methods->oob(tdb, off + len, 0); + ecode = tdb->methods->oob(tdb, off + len, false); if (ecode != TDB_SUCCESS) { return ecode; } @@ -563,7 +567,7 @@ static void *tdb_direct(struct tdb_context *tdb, tdb_off_t off, size_t len, if (unlikely(!tdb->file->map_ptr)) return NULL; - ecode = tdb_oob(tdb, off + len, true); + ecode = tdb_oob(tdb, off + len, false); if (unlikely(ecode != TDB_SUCCESS)) return TDB_ERR_PTR(ecode); return (char *)tdb->file->map_ptr + off; -- cgit