From 93a7f324c06321023e2702f7989fdfad32573793 Mon Sep 17 00:00:00 2001 From: Determinant Date: Sat, 19 Sep 2020 22:27:11 -0400 Subject: apply mulit-coin universally --- core/state/statedb.go | 57 +++++++++++++++++++++++++++------------------------ 1 file changed, 30 insertions(+), 27 deletions(-) (limited to 'core/state/statedb.go') diff --git a/core/state/statedb.go b/core/state/statedb.go index dbf0870..c4d926d 100644 --- a/core/state/statedb.go +++ b/core/state/statedb.go @@ -267,29 +267,30 @@ func (self *StateDB) GetBalanceMultiCoin(addr common.Address, coinID common.Hash return common.Big0 } -func (self *StateDB) EnableMultiCoin(addr common.Address) error { - stateObject := self.GetOrNewStateObject(addr) - if stateObject.data.Root != emptyRoot && stateObject.data.Root != zeroRoot { - return errors.New(fmt.Sprintf("not a fresh account: %s", stateObject.data.Root.Hex())) - } - if !stateObject.EnableMultiCoin() { - return errors.New("multi-coin mode already enabled") - } - log.Debug(fmt.Sprintf("enabled MC for %s", addr.Hex())) - return nil -} - -func (self *StateDB) ForceEnableMultiCoin(addr common.Address) { - stateObject := self.GetOrNewStateObject(addr) - stateObject.EnableMultiCoin() -} +//func (self *StateDB) EnableMultiCoin(addr common.Address) error { +// stateObject := self.GetOrNewStateObject(addr) +// if stateObject.data.Root != emptyRoot && stateObject.data.Root != zeroRoot { +// return errors.New(fmt.Sprintf("not a fresh account: %s", stateObject.data.Root.Hex())) +// } +// if !stateObject.EnableMultiCoin() { +// return errors.New("multi-coin mode already enabled") +// } +// log.Debug(fmt.Sprintf("enabled MC for %s", addr.Hex())) +// return nil +//} +// +//func (self *StateDB) ForceEnableMultiCoin(addr common.Address) { +// stateObject := self.GetOrNewStateObject(addr) +// stateObject.EnableMultiCoin() +//} func (self *StateDB) IsMultiCoin(addr common.Address) bool { - stateObject := self.getStateObject(addr) - if stateObject != nil { - return stateObject.IsMultiCoin() - } - return false + return true + //stateObject := self.getStateObject(addr) + //if stateObject != nil { + // return stateObject.IsMultiCoin() + //} + //return false } func (s *StateDB) GetNonce(addr common.Address) uint64 { @@ -339,6 +340,10 @@ func (s *StateDB) GetCodeHash(addr common.Address) common.Hash { func (s *StateDB) GetState(addr common.Address, hash common.Hash) common.Hash { stateObject := s.getStateObject(addr) if stateObject != nil { + // NOTE: last-minute fix: just universally enable MC + //if stateObject.data.IsMultiCoin { + NormalizeStateKey(&hash) + //} return stateObject.GetState(s.db, hash) } return common.Hash{} @@ -425,9 +430,6 @@ func (s *StateDB) SetBalance(addr common.Address, amount *big.Int) { // AddBalance adds amount to the account associated with addr. func (s *StateDB) AddBalanceMultiCoin(addr common.Address, coinID common.Hash, amount *big.Int) { - if !s.IsMultiCoin(addr) { - s.EnableMultiCoin(addr) - } stateObject := s.GetOrNewStateObject(addr) if stateObject != nil { stateObject.AddBalanceMultiCoin(coinID, amount, s.db) @@ -467,9 +469,10 @@ func (s *StateDB) SetState(addr common.Address, key, value common.Hash) (res err res = nil stateObject := s.GetOrNewStateObject(addr) if stateObject != nil { - if stateObject.data.IsMultiCoin { - NormalizeStateKey(&key) - } + // NOTE: last-minute fix: just universally enable MC + //if stateObject.data.IsMultiCoin { + NormalizeStateKey(&key) + //} stateObject.SetState(s.db, key, value) } return -- cgit v1.2.3-70-g09d2