Updated block to use state instead of trie directly

1 files changed, 8 insertions, 0 deletions

diff --git a/ethchain/state.go b/ethchain/state.go
index 1a18ea1d7..cff192b54 100644
--- a/ethchain/state.go
+++ b/ethchain/state.go
@@ -13,6 +13,10 @@ func NewState(trie *ethutil.Trie) *State {
 	return &State{trie: trie}
 }
 
+func (s *State) Reset() {
+	s.trie.Undo()
+}
+
 func (s *State) GetContract(addr []byte) *Contract {
 	data := s.trie.Get(string(addr))
 	if data == "" {
@@ -54,3 +58,7 @@ func (s *State) GetAccount(addr []byte) (account *Address) {
 func (s *State) UpdateAccount(addr []byte, account *Address) {
 	s.trie.Update(string(addr), string(account.RlpEncode()))
 }
+
+func (s *State) Cmp(other *State) bool {
+	return s.trie.Cmp(other.trie)
+}