Remove partial key size limit from trie codec (#12566)

* remove size limit from trie codec

* test previous upper limit is not enforced anymore

* fmt

* restore test
This commit is contained in:
cheme
2022-11-08 21:24:54 +01:00
committed by GitHub
parent 78cddd7631
commit 8b2ca711b3
4 changed files with 17 additions and 9 deletions
+15 -1
View File
@@ -527,7 +527,6 @@ where
/// Constants used into trie simplification codec.
mod trie_constants {
const FIRST_PREFIX: u8 = 0b_00 << 6;
pub const NIBBLE_SIZE_BOUND: usize = u16::max_value() as usize;
pub const LEAF_PREFIX_MASK: u8 = 0b_01 << 6;
pub const BRANCH_WITHOUT_MASK: u8 = 0b_10 << 6;
pub const BRANCH_WITH_MASK: u8 = 0b_11 << 6;
@@ -987,6 +986,21 @@ mod tests {
assert_eq!(first_storage_root, second_storage_root);
}
#[test]
fn big_key() {
let check = |keysize: usize| {
let mut memdb = PrefixedMemoryDB::<Blake2Hasher>::default();
let mut root = Default::default();
let mut t = TrieDBMutBuilder::<LayoutV1>::new(&mut memdb, &mut root).build();
t.insert(&vec![0x01u8; keysize][..], &[0x01u8, 0x23]).unwrap();
std::mem::drop(t);
let t = TrieDBBuilder::<LayoutV1>::new(&memdb, &root).build();
assert_eq!(t.get(&vec![0x01u8; keysize][..]).unwrap(), Some(vec![0x01u8, 0x23]));
};
check(u16::MAX as usize / 2); // old limit
check(u16::MAX as usize / 2 + 1); // value over old limit still works
}
#[test]
fn node_with_no_children_fail_decoding() {
let branch = NodeCodec::<Blake2Hasher>::branch_node_nibbled(