trie: fix error in node decoding

This commit is contained in:
Martin Holst Swende 2019-02-16 15:04:09 +01:00
parent 5b8ae7885e
commit d8224029b7
No known key found for this signature in database
GPG key ID: 683B438C05A5DDF0
2 changed files with 16 additions and 0 deletions

View file

@ -52,6 +52,9 @@ func hexToCompact(hex []byte) []byte {
} }
func compactToHex(compact []byte) []byte { func compactToHex(compact []byte) []byte {
if len(compact) == 0 {
return compact
}
base := keybytesToHex(compact) base := keybytesToHex(compact)
// delete terminator flag // delete terminator flag
if base[0] < 2 { if base[0] < 2 {

View file

@ -614,3 +614,16 @@ func updateString(trie *Trie, k, v string) {
func deleteString(trie *Trie, k string) { func deleteString(trie *Trie, k string) {
trie.Delete([]byte(k)) trie.Delete([]byte(k))
} }
func TestDecodeNode(t *testing.T) {
t.Parallel()
var (
hash = make([]byte, 20)
elems = make([]byte, 20)
)
for i := 0; i < 5000000; i++ {
rand.Read(hash)
rand.Read(elems)
decodeNode(hash, elems, 1)
}
}