X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Ftdb2%2Fhash.c;h=dec7aae554acbd0f21fd10bf44c886ef72aae9ae;hb=06a5b1a852c207a5172922e39d8effab0dd17b91;hp=ae9442d934d149717cd13200bea98b8424466cfa;hpb=cd1c68c0d989e6aed401260f92c63db8e8132db5;p=ccan diff --git a/ccan/tdb2/hash.c b/ccan/tdb2/hash.c index ae9442d9..dec7aae5 100644 --- a/ccan/tdb2/hash.c +++ b/ccan/tdb2/hash.c @@ -537,7 +537,7 @@ again: /* Return 1 if we find something, 0 if not, -1 on error. */ int next_in_hash(struct tdb_context *tdb, int ltype, struct traverse_info *tinfo, - TDB_DATA *kbuf, unsigned int *dlen) + TDB_DATA *kbuf, size_t *dlen) { const unsigned group_bits = TDB_TOPLEVEL_HASH_BITS-TDB_HASH_GROUP_BITS; tdb_off_t hlock_start, hlock_range, off; @@ -560,6 +560,14 @@ int next_in_hash(struct tdb_context *tdb, int ltype, ltype); return -1; } + if (rec_magic(&rec) != TDB_MAGIC) { + tdb->log(tdb, TDB_DEBUG_FATAL, tdb->log_priv, + "next_in_hash:" + " corrupt record at %llu\n", + (long long)off); + return -1; + } + kbuf->dsize = rec_key_length(&rec); /* They want data as well? */ @@ -591,7 +599,7 @@ int next_in_hash(struct tdb_context *tdb, int ltype, /* Return 1 if we find something, 0 if not, -1 on error. */ int first_in_hash(struct tdb_context *tdb, int ltype, struct traverse_info *tinfo, - TDB_DATA *kbuf, unsigned int *dlen) + TDB_DATA *kbuf, size_t *dlen) { tinfo->prev = 0; tinfo->toplevel_group = 0;