From 42bf34fff5c6bb0bd1db8c939c1ea6cc29e22329 Mon Sep 17 00:00:00 2001 From: cheme Date: Thu, 23 Aug 2018 17:03:55 +0200 Subject: [PATCH 1/2] Fix for removal of a fatdb value. --- patricia_trie/src/fatdbmut.rs | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/patricia_trie/src/fatdbmut.rs b/patricia_trie/src/fatdbmut.rs index 67f4f14..c0be7cc 100644 --- a/patricia_trie/src/fatdbmut.rs +++ b/patricia_trie/src/fatdbmut.rs @@ -84,7 +84,7 @@ where let out = self.raw.insert(hash.as_ref(), value)?; let db = self.raw.db_mut(); - // don't insert if it doesn't exist. + // insert if it doesn't exist. if out.is_none() { let aux_hash = H::hash(hash.as_ref()); db.emplace(aux_hash, DBValue::from_slice(key)); @@ -96,9 +96,10 @@ where let hash = H::hash(key); let out = self.raw.remove(hash.as_ref())?; - // don't remove if it already exists. + // remove if it already exists. if out.is_some() { - self.raw.db_mut().remove(&hash); + let aux_hash = H::hash(hash.as_ref()); + self.raw.db_mut().remove(&aux_hash); } Ok(out) @@ -126,4 +127,4 @@ mod test { let t = TrieDB::new(&memdb, &root).unwrap(); assert_eq!(t.get(&keccak::keccak(&[0x01u8, 0x23])).unwrap().unwrap(), DBValue::from_slice(&[0x01u8, 0x23])); } -} \ No newline at end of file +} From a6b0ea8826d1bc8401c0e46466e0561840c86bbe Mon Sep 17 00:00:00 2001 From: cheme Date: Wed, 5 Sep 2018 18:19:51 +0200 Subject: [PATCH 2/2] Simple test case for the pr --- patricia_trie/src/fatdbmut.rs | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/patricia_trie/src/fatdbmut.rs b/patricia_trie/src/fatdbmut.rs index c0be7cc..61279c3 100644 --- a/patricia_trie/src/fatdbmut.rs +++ b/patricia_trie/src/fatdbmut.rs @@ -127,4 +127,21 @@ mod test { let t = TrieDB::new(&memdb, &root).unwrap(); assert_eq!(t.get(&keccak::keccak(&[0x01u8, 0x23])).unwrap().unwrap(), DBValue::from_slice(&[0x01u8, 0x23])); } + + #[test] + fn fatdbmut_insert_remove_key_mapping() { + let mut memdb = MemoryDB::::new(); + let mut root = H256::new(); + let key = [0x01u8, 0x23]; + let val = [0x01u8, 0x24]; + let key_hash = keccak::keccak(&key); + let aux_hash = keccak::keccak(&key_hash); + let mut t = FatDBMut::new(&mut memdb, &mut root); + t.insert(&key, &val).unwrap(); + assert_eq!(t.get(&key), Ok(Some(DBValue::from_slice(&val)))); + assert_eq!(t.db().get(&aux_hash), Some(DBValue::from_slice(&key))); + t.remove(&key).unwrap(); + assert_eq!(t.db().get(&aux_hash), None); + } + }