Implemented managed state

* Reimplemented nonce management for known accounts.

1 files changed, 32 insertions, 18 deletions

diff --git a/state/managed_state.go b/state/managed_state.go
index b0a129af0..1914e977a 100644
--- a/state/managed_state.go
+++ b/state/managed_state.go
@@ -2,44 +2,53 @@ package state
 
 import "sync"
 
+type account struct {
+	stateObject *StateObject
+	nstart      uint64
+	nonces      []bool
+}
+
 type ManagedState struct {
 	*StateDB
 
 	mu sync.RWMutex
 
-	accounts map[string]*StateObject
+	accounts map[string]*account
 }
 
 func ManageState(statedb *StateDB) *ManagedState {
 	return &ManagedState{
 		StateDB:  statedb,
-		accounts: make(map[string]*StateObject),
+		accounts: make(map[string]*account),
 	}
 }
 
-func (ms *ManagedState) IncrementNonce(addr []byte) {
-	ms.mu.Lock()
-	defer ms.mu.Unlock()
-
-	ms.getAccount(addr).nonce++
-}
-
-func (ms *ManagedState) DecrementNonce(addr []byte) {
-	// Decrementing a nonce does not mean we are interested in the account
-	// incrementing only happens if you control the account, therefor
-	// incrementing  behaves differently from decrementing
+func (ms *ManagedState) RemoveNonce(addr []byte, n uint64) {
 	if ms.hasAccount(addr) {
 		ms.mu.Lock()
 		defer ms.mu.Unlock()
 
-		ms.getAccount(addr).nonce--
+		account := ms.getAccount(addr)
+		if n-account.nstart < uint64(len(account.nonces)) {
+			reslice := make([]bool, n-account.nstart)
+			copy(reslice, account.nonces[:n-account.nstart])
+			account.nonces = reslice
+		}
 	}
 }
 
-func (ms *ManagedState) GetNonce(addr []byte) uint64 {
+func (ms *ManagedState) NewNonce(addr []byte) uint64 {
 	ms.mu.RLock()
 	defer ms.mu.RUnlock()
-	return ms.getAccount(addr).nonce
+
+	account := ms.getAccount(addr)
+	for i, nonce := range account.nonces {
+		if !nonce {
+			return account.nstart + uint64(i)
+		}
+	}
+	account.nonces = append(account.nonces, false)
+	return uint64(len(account.nonces)) + account.nstart
 }
 
 func (ms *ManagedState) hasAccount(addr []byte) bool {
@@ -47,10 +56,15 @@ func (ms *ManagedState) hasAccount(addr []byte) bool {
 	return ok
 }
 
-func (ms *ManagedState) getAccount(addr []byte) *StateObject {
+func (ms *ManagedState) getAccount(addr []byte) *account {
 	if _, ok := ms.accounts[string(addr)]; !ok {
-		ms.accounts[string(addr)] = ms.GetOrNewStateObject(addr)
+		so := ms.GetOrNewStateObject(addr)
+		ms.accounts[string(addr)] = newAccount(so)
 	}
 
 	return ms.accounts[string(addr)]
 }
+
+func newAccount(so *StateObject) *account {
+	return &account{so, so.nonce - 1, nil}
+}