aboutsummaryrefslogtreecommitdiffstats
path: root/core
diff options
context:
space:
mode:
authorobscuren <geffobscura@gmail.com>2015-04-08 19:04:23 +0800
committerobscuren <geffobscura@gmail.com>2015-04-08 19:06:56 +0800
commit31b086f5111e4bd506a4ae468a9247bff24dd5cb (patch)
tree793e3bf0bd88d963d7c7e9fe2255326a5747114a /core
parent09147a50ede8c85022d115e23bb2ce067a50c8de (diff)
downloaddexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar.gz
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar.bz2
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar.lz
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar.xz
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.tar.zst
dexon-31b086f5111e4bd506a4ae468a9247bff24dd5cb.zip
Added additional methods to the managed state
* GetNonce Returns the canonical nonce * SetNonce Set the managed account's nonce
Diffstat (limited to 'core')
-rw-r--r--core/state/managed_state.go28
-rw-r--r--core/state/managed_state_test.go18
2 files changed, 44 insertions, 2 deletions
diff --git a/core/state/managed_state.go b/core/state/managed_state.go
index 0fcc1be67..ddf337af3 100644
--- a/core/state/managed_state.go
+++ b/core/state/managed_state.go
@@ -20,6 +20,7 @@ type ManagedState struct {
accounts map[string]*account
}
+// ManagedState returns a new managed state with the statedb as it's backing layer
func ManageState(statedb *StateDB) *ManagedState {
return &ManagedState{
StateDB: statedb,
@@ -27,14 +28,16 @@ func ManageState(statedb *StateDB) *ManagedState {
}
}
+// SetState sets the backing layer of the managed state
func (ms *ManagedState) SetState(statedb *StateDB) {
ms.mu.Lock()
defer ms.mu.Unlock()
ms.StateDB = statedb
}
+// RemoveNonce removed the nonce from the managed state and all future pending nonces
func (ms *ManagedState) RemoveNonce(addr common.Address, n uint64) {
- if ms.hasAccount(addr) {
+ if ms.HasAccount(addr) {
ms.mu.Lock()
defer ms.mu.Unlock()
@@ -47,6 +50,7 @@ func (ms *ManagedState) RemoveNonce(addr common.Address, n uint64) {
}
}
+// NewNonce returns the new canonical nonce for the managed account
func (ms *ManagedState) NewNonce(addr common.Address) uint64 {
ms.mu.RLock()
defer ms.mu.RUnlock()
@@ -61,11 +65,31 @@ func (ms *ManagedState) NewNonce(addr common.Address) uint64 {
return uint64(len(account.nonces)) + account.nstart
}
-func (ms *ManagedState) hasAccount(addr common.Address) bool {
+// GetNonce returns the canonical nonce for the managed or unmanged account
+func (ms *ManagedState) GetNonce(addr common.Address) uint64 {
+ if ms.HasAccount(addr) {
+ account := ms.getAccount(addr)
+ return uint64(len(account.nonces)) + account.nstart
+ } else {
+ return ms.StateDB.GetNonce(addr)
+ }
+}
+
+// SetNonce sets the new canonical nonce for the managed state
+func (ms *ManagedState) SetNonce(addr common.Address, nonce uint64) {
+ so := ms.GetOrNewStateObject(addr)
+ so.SetNonce(nonce)
+
+ ms.accounts[addr.Str()] = newAccount(so)
+}
+
+// HasAccount returns whether the given address is managed or not
+func (ms *ManagedState) HasAccount(addr common.Address) bool {
_, ok := ms.accounts[addr.Str()]
return ok
}
+// populate the managed state
func (ms *ManagedState) getAccount(addr common.Address) *account {
straddr := addr.Str()
if account, ok := ms.accounts[straddr]; !ok {
diff --git a/core/state/managed_state_test.go b/core/state/managed_state_test.go
index b61f59e6d..766231d21 100644
--- a/core/state/managed_state_test.go
+++ b/core/state/managed_state_test.go
@@ -87,3 +87,21 @@ func TestRemoteNonceChange(t *testing.T) {
t.Error("expected nonce after remote update to be", 201, "got", nonce)
}
}
+
+func TestSetNonce(t *testing.T) {
+ ms, _ := create()
+
+ var addr common.Address
+ ms.SetNonce(addr, 10)
+
+ if ms.GetNonce(addr) != 10 {
+ t.Errorf("Expected nonce of 10, got", ms.GetNonce(addr))
+ }
+
+ addr[0] = 1
+ ms.StateDB.SetNonce(addr, 1)
+
+ if ms.GetNonce(addr) != 1 {
+ t.Errorf("Expected nonce of 1, got", ms.GetNonce(addr))
+ }
+}