tdb2: handle non-transaction-page-aligned sizes in recovery.

[ccan] / ccan / tdb2 / transaction.c

diff --git a/ccan/tdb2/transaction.c b/ccan/tdb2/transaction.c
index f1414391abe42caa7b81504dede1ea3700291c1b..73ceb9620265ee70b86320871c57aba0488c429f 100644 (file)
--- a/ccan/tdb2/transaction.c
+++ b/ccan/tdb2/transaction.c
@@ -615,7 +615,7 @@ static tdb_len_t tdb_recovery_size(struct tdb_context *tdb)
        tdb_len_t recovery_size = 0;
        int i;
 
-       recovery_size = sizeof(tdb_len_t);
+       recovery_size = 0;
        for (i=0;i<tdb->transaction->num_blocks;i++) {
                if (i * PAGESIZE >= tdb->transaction->old_map_size) {
                        break;
@@ -770,7 +770,7 @@ static enum TDB_ERROR transaction_setup_recovery(struct tdb_context *tdb,
        const struct tdb_methods *methods = tdb->transaction->io_methods;
        struct tdb_recovery_record *rec;
        tdb_off_t old_map_size = tdb->transaction->old_map_size;
-       uint64_t magic, tailer;
+       uint64_t magic;
        int i;
        enum TDB_ERROR ecode;
 
@@ -815,6 +815,7 @@ static enum TDB_ERROR transaction_setup_recovery(struct tdb_context *tdb,
                if (offset >= old_map_size) {
                        continue;
                }
+
                if (offset + length > tdb->file->map_size) {
                        free(data);
                        return tdb_logerr(tdb, TDB_ERR_CORRUPT, TDB_LOG_ERROR,
@@ -829,9 +830,19 @@ static enum TDB_ERROR transaction_setup_recovery(struct tdb_context *tdb,
                /* the recovery area contains the old data, not the
                   new data, so we have to call the original tdb_read
                   method to get it */
-               ecode = methods->tread(tdb, offset,
-                                      p + sizeof(offset) + sizeof(length),
-                                      length);
+               if (offset + length > old_map_size) {
+                       /* Short read at EOF, and zero fill. */
+                       unsigned int len = old_map_size - offset;
+                       ecode = methods->tread(tdb, offset,
+                                              p + sizeof(offset) + sizeof(length),
+                                              len);
+                       memset(p + sizeof(offset) + sizeof(length) + len, 0,
+                              length - len);
+               } else {
+                       ecode = methods->tread(tdb, offset,
+                                              p + sizeof(offset) + sizeof(length),
+                                              length);
+               }
                if (ecode != TDB_SUCCESS) {
                        free(data);
                        return ecode;
@@ -839,11 +850,6 @@ static enum TDB_ERROR transaction_setup_recovery(struct tdb_context *tdb,
                p += sizeof(offset) + sizeof(length) + length;
        }
 
-       /* and the tailer */
-       tailer = sizeof(*rec) + recovery_max_size;
-       memcpy(p, &tailer, sizeof(tailer));
-       tdb_convert(tdb, p, sizeof(tailer));
-
        /* write the recovery data to the recovery area */
        ecode = methods->twrite(tdb, recovery_offset, data,
                                sizeof(*rec) + recovery_size);