diff options
Diffstat (limited to 'trie')
-rw-r--r-- | trie/secure_trie.go | 32 | ||||
-rw-r--r-- | trie/trie_test.go | 1 |
2 files changed, 32 insertions, 1 deletions
diff --git a/trie/secure_trie.go b/trie/secure_trie.go new file mode 100644 index 000000000..c6ef90351 --- /dev/null +++ b/trie/secure_trie.go @@ -0,0 +1,32 @@ +package trie + +import "github.com/ethereum/go-ethereum/crypto" + +type SecureTrie struct { + *Trie +} + +func NewSecure(root []byte, backend Backend) *SecureTrie { + return &SecureTrie{New(root, backend)} +} + +func (self *SecureTrie) Update(key, value []byte) Node { + return self.Trie.Update(crypto.Sha3(key), value) +} +func (self *SecureTrie) UpdateString(key, value string) Node { + return self.Update([]byte(key), []byte(value)) +} + +func (self *SecureTrie) Get(key []byte) []byte { + return self.Trie.Get(crypto.Sha3(key)) +} +func (self *SecureTrie) GetString(key string) []byte { + return self.Get([]byte(key)) +} + +func (self *SecureTrie) Delete(key []byte) Node { + return self.Trie.Delete(crypto.Sha3(key)) +} +func (self *SecureTrie) DeletString(key string) Node { + return self.Delete([]byte(key)) +} diff --git a/trie/trie_test.go b/trie/trie_test.go index 4b185f355..4b0243bcf 100644 --- a/trie/trie_test.go +++ b/trie/trie_test.go @@ -274,7 +274,6 @@ func TestLargeData(t *testing.T) { trie.Update(value2.k, value2.v) vals[string(value.k)] = value vals[string(value2.k)] = value2 - fmt.Println(value, "\n", value2) } it := trie.Iterator() |