diff options
author | Sonic <sonic@dexon.org> | 2018-11-09 17:24:54 +0800 |
---|---|---|
committer | Wei-Ning Huang <w@byzantine-lab.io> | 2019-06-12 17:27:18 +0800 |
commit | 638816ae9be2dab6f48a367689563121c85d7786 (patch) | |
tree | bb0179cf42b998e5aea51f66fd9f80e7c33bec20 /core | |
parent | 5aaccf466773e60c93ab150d89abd950bd43ca28 (diff) | |
download | go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar.gz go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar.bz2 go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar.lz go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar.xz go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.tar.zst go-tangerine-638816ae9be2dab6f48a367689563121c85d7786.zip |
core: support extracting governance state from state trie
Diffstat (limited to 'core')
-rw-r--r-- | core/blockchain.go | 27 | ||||
-rw-r--r-- | core/state/gov.go | 31 | ||||
-rw-r--r-- | core/types/govstate.go | 15 |
3 files changed, 73 insertions, 0 deletions
diff --git a/core/blockchain.go b/core/blockchain.go index e7b1c846f..8bf680c38 100644 --- a/core/blockchain.go +++ b/core/blockchain.go @@ -1740,6 +1740,33 @@ func (bc *BlockChain) GetPending() (*types.Block, *state.StateDB) { return block, s } +// GetGovStateByHash extracts the governance contract's state from state trie +// (the merkle proof of governance contract address and the whole storage) +// at the given block hash. +func (bc *BlockChain) GetGovStateByHash(hash common.Hash) (*types.GovState, error) { + header := bc.GetHeaderByHash(hash) + if header == nil { + return nil, fmt.Errorf("header not found") + } + statedb, err := bc.StateAt(header.Root) + if err != nil { + return nil, err + } + return state.GetGovState(statedb, header, vm.GovernanceContractAddress) +} + +func (bc *BlockChain) GetGovStateByNumber(number uint64) (*types.GovState, error) { + header := bc.GetHeaderByNumber(number) + if header == nil { + return nil, fmt.Errorf("header not found") + } + statedb, err := bc.StateAt(header.Root) + if err != nil { + return nil, err + } + return state.GetGovState(statedb, header, vm.GovernanceContractAddress) +} + // reorg takes two blocks, an old chain and a new chain and will reconstruct the // blocks and inserts them to be part of the new canonical chain and accumulates // potential missing transactions and post an event about them. diff --git a/core/state/gov.go b/core/state/gov.go new file mode 100644 index 000000000..8b94cfab2 --- /dev/null +++ b/core/state/gov.go @@ -0,0 +1,31 @@ +package state + +import ( + "github.com/dexon-foundation/dexon/common" + "github.com/dexon-foundation/dexon/core/types" + "github.com/dexon-foundation/dexon/trie" +) + +func GetGovState(statedb *StateDB, header *types.Header, + addr common.Address) (*types.GovState, error) { + proof, err := statedb.GetProof(addr) + if err != nil { + return nil, err + } + + govState := &types.GovState{ + BlockHash: header.Hash(), + Number: header.Number, + Root: header.Root, + Proof: proof, + } + + if t := statedb.StorageTrie(addr); t != nil { + it := trie.NewIterator(t.NodeIterator(nil)) + for it.Next() { + govState.Storage = append(govState.Storage, + [2][]byte{it.Key, it.Value}) + } + } + return govState, nil +} diff --git a/core/types/govstate.go b/core/types/govstate.go new file mode 100644 index 000000000..75f77cb0d --- /dev/null +++ b/core/types/govstate.go @@ -0,0 +1,15 @@ +package types + +import ( + "math/big" + + "github.com/dexon-foundation/dexon/common" +) + +type GovState struct { + BlockHash common.Hash + Number *big.Int + Root common.Hash + Proof [][]byte + Storage [][2][]byte +} |