diff --git a/st.c b/st.c index f26443f99c..1d78ffebad 100644 --- a/st.c +++ b/st.c @@ -834,6 +834,7 @@ st_update(st_table *table, st_data_t key, st_update_callback_func *func, st_data existing = 1; } { + const st_data_t old_key = key; retval = (*func)(&key, &value, arg, existing); if (!table->entries_packed) { FIND_ENTRY(table, ptr, hash_val, bin_pos); @@ -845,6 +846,14 @@ st_update(st_table *table, st_data_t key, st_update_callback_func *func, st_data add_packed_direct(table, key, value, hash_val); break; } + if (old_key != PKEY(table, i)) return -1; + if (old_key != key) { + if (do_hash(key, table) != hash_val && + !EQUAL(table, key, old_key)) { + return -1; + } + PKEY(table, i) = key; + } PVAL_SET(table, i, value); break; case ST_DELETE: @@ -863,6 +872,7 @@ st_update(st_table *table, st_data_t key, st_update_callback_func *func, st_data existing = 1; } { + const st_data_t old_key = key; retval = (*func)(&key, &value, arg, existing); unpacked: switch (retval) { @@ -871,6 +881,14 @@ st_update(st_table *table, st_data_t key, st_update_callback_func *func, st_data add_direct(table, key, value, hash_val, hash_pos(hash_val, table->num_bins)); break; } + if (old_key != ptr->key) return -1; + if (old_key != key) { + if (do_hash(key, table) != hash_val && + !EQUAL(table, key, old_key)) { + return -1; + } + ptr->key = key; + } ptr->record = value; break; case ST_DELETE: