-
Notifications
You must be signed in to change notification settings - Fork 1.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
83a9b13
commit 41ab497
Showing
6 changed files
with
443 additions
and
35 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,297 @@ | ||
| package state | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "sync/atomic" | ||
|
|
||
| versa "github.com/bnb-chain/versioned-state-database" | ||
| "github.com/ethereum/go-ethereum/common" | ||
| "github.com/ethereum/go-ethereum/common/lru" | ||
| "github.com/ethereum/go-ethereum/core/rawdb" | ||
| "github.com/ethereum/go-ethereum/core/types" | ||
| "github.com/ethereum/go-ethereum/crypto" | ||
| "github.com/ethereum/go-ethereum/ethdb" | ||
| "github.com/ethereum/go-ethereum/log" | ||
| "github.com/ethereum/go-ethereum/rlp" | ||
| "github.com/ethereum/go-ethereum/trie" | ||
| "github.com/ethereum/go-ethereum/trie/trienode" | ||
| "github.com/ethereum/go-ethereum/triedb" | ||
| ) | ||
|
|
||
| type cachingVersaDB struct { | ||
| triedb *triedb.Database | ||
| versionDB versa.Database | ||
| codeDB ethdb.KeyValueStore | ||
| codeSizeCache *lru.Cache[common.Hash, int] | ||
| codeCache *lru.SizeConstrainedCache[common.Hash, []byte] | ||
|
|
||
| accTree *VersaTree | ||
| state versa.StateHandler | ||
| root common.Hash | ||
| mode versa.StateMode | ||
| hasState atomic.Bool | ||
| } | ||
|
|
||
| // NewVersaDatabase should be call by NewDatabaseWithNodeDB | ||
| // TODO:: NewDatabaseWithNodeDB should add mode param. | ||
| func NewVersaDatabase(db ethdb.Database, triedb *triedb.Database, mode versa.StateMode) Database { | ||
| return &cachingVersaDB{ | ||
| triedb: triedb, | ||
| versionDB: triedb.VersaDB(), | ||
| codeDB: db, | ||
| codeSizeCache: lru.NewCache[common.Hash, int](codeSizeCacheSize), | ||
| codeCache: lru.NewSizeConstrainedCache[common.Hash, []byte](codeCacheSize), | ||
| } | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) OpenTrie(root common.Hash) (Trie, error) { | ||
| if cv.hasState.Load() { | ||
| //TODO:: will change to log.Error after stabilization | ||
| panic("account tree has open") | ||
| } | ||
|
|
||
| // TODO:: if root tree, versa db shouldb ignore check version | ||
| state, err := cv.versionDB.OpenState(0, root, cv.mode) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| handler, err := cv.versionDB.OpenTree(state, 0, common.Hash{}, root) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| tree := &VersaTree{ | ||
| db: cv.versionDB, | ||
| handler: handler, | ||
| accountTree: true, | ||
| root: root, | ||
| } | ||
|
|
||
| cv.state = state | ||
| cv.hasState.Store(true) // if set, can't change | ||
| cv.accTree = tree | ||
| cv.root = root | ||
|
|
||
| return tree, nil | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) OpenStorageTrie(stateRoot common.Hash, address common.Address, root common.Hash, _ Trie) (Trie, error) { | ||
| if !cv.hasState.Load() { | ||
| //TODO:: will change to log.Error after stabilization | ||
| panic("open account tree, before open storage tree") | ||
| } | ||
| if cv.root.Cmp(root) != 0 { | ||
| panic("account root mismatch, on open storage tree") | ||
| } | ||
|
|
||
| version, account, err := cv.accTree.getAccountWithVersion(address) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| if account.Root.Cmp(stateRoot) != 0 { | ||
| return nil, fmt.Errorf("state root mismatch") | ||
| } | ||
|
|
||
| handler, err := cv.versionDB.OpenTree(cv.state, version, crypto.Keccak256Hash(address.Bytes()), stateRoot) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| tree := &VersaTree{ | ||
| db: cv.versionDB, | ||
| handler: handler, | ||
| version: version, | ||
| root: root, | ||
| stateRoot: stateRoot, | ||
| address: address, | ||
| } | ||
| return tree, nil | ||
| } | ||
|
|
||
| // Flush unique to versa | ||
| func (cv *cachingVersaDB) Flush() error { | ||
| return cv.versionDB.Flush(cv.state) | ||
| } | ||
|
|
||
| // Release unique to versa | ||
| func (cv *cachingVersaDB) Release() error { | ||
| if err := cv.versionDB.CloseState(cv.state); err != nil { | ||
| return nil | ||
| } | ||
| cv.release() | ||
| return nil | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) release() { | ||
| cv.hasState.Store(false) | ||
| cv.accTree = nil | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) CopyTrie(Trie) Trie { | ||
| //TODO:: support in the future | ||
| return nil | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) ContractCode(addr common.Address, codeHash common.Hash) ([]byte, error) { | ||
| code, _ := cv.codeCache.Get(codeHash) | ||
| if len(code) > 0 { | ||
| return code, nil | ||
| } | ||
| code = rawdb.ReadCodeWithPrefix(cv.codeDB, codeHash) | ||
| if len(code) > 0 { | ||
| cv.codeCache.Add(codeHash, code) | ||
| cv.codeSizeCache.Add(codeHash, len(code)) | ||
| return code, nil | ||
| } | ||
| return nil, errors.New("not found") | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) ContractCodeSize(addr common.Address, codeHash common.Hash) (int, error) { | ||
| if cached, ok := cv.codeSizeCache.Get(codeHash); ok { | ||
| return cached, nil | ||
| } | ||
| code, err := cv.ContractCode(addr, codeHash) | ||
| return len(code), err | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) ContractCodeWithPrefix(address common.Address, codeHash common.Hash) ([]byte, error) { | ||
| code, _ := cv.codeCache.Get(codeHash) | ||
| if len(code) > 0 { | ||
| return code, nil | ||
| } | ||
| code = rawdb.ReadCodeWithPrefix(cv.codeDB, codeHash) | ||
| if len(code) > 0 { | ||
| cv.codeCache.Add(codeHash, code) | ||
| cv.codeSizeCache.Add(codeHash, len(code)) | ||
| return code, nil | ||
| } | ||
| return nil, errors.New("not found") | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) DiskDB() ethdb.KeyValueStore { | ||
| return cv.codeDB | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) TrieDB() *triedb.Database { | ||
| return cv.triedb | ||
| } | ||
|
|
||
| func (cv *cachingVersaDB) NoTries() bool { | ||
| // TODO:: not support fastnode | ||
| return false | ||
| } | ||
|
|
||
| type VersaTree struct { | ||
| db versa.Database | ||
| handler versa.TreeHandler | ||
| version int64 | ||
| accountTree bool | ||
|
|
||
| // TODO:: debugging, used for logging | ||
| root common.Hash | ||
| stateRoot common.Hash | ||
| address common.Address | ||
| } | ||
|
|
||
| func (vt *VersaTree) GetKey(key []byte) []byte { | ||
| _, val, err := vt.db.Get(vt.handler, key) | ||
| if err != nil { | ||
| log.Warn("failed to get key from version db") | ||
| } | ||
| return val | ||
| } | ||
|
|
||
| func (vt *VersaTree) GetAccount(address common.Address) (*types.StateAccount, error) { | ||
| _, res, err := vt.getAccountWithVersion(address) | ||
| return res, err | ||
| } | ||
|
|
||
| func (vt *VersaTree) getAccountWithVersion(address common.Address) (int64, *types.StateAccount, error) { | ||
| vt.CheckAccountTree() | ||
| ver, res, err := vt.db.Get(vt.handler, address.Bytes()) | ||
| if res == nil || err != nil { | ||
| return ver, nil, err | ||
| } | ||
| ret := new(types.StateAccount) | ||
| err = rlp.DecodeBytes(res, ret) | ||
| return ver, ret, err | ||
| } | ||
|
|
||
| func (vt *VersaTree) GetStorage(_ common.Address, key []byte) ([]byte, error) { | ||
| vt.CheckStorageTree() | ||
| _, res, err := vt.db.Get(vt.handler, key) | ||
| if res == nil || err != nil { | ||
| return nil, err | ||
| } | ||
| return res, err | ||
| } | ||
|
|
||
| func (vt *VersaTree) UpdateAccount(address common.Address, account *types.StateAccount) error { | ||
| vt.CheckAccountTree() | ||
| data, err := rlp.EncodeToBytes(account) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| return vt.db.Put(vt.handler, address.Bytes(), data) | ||
| } | ||
|
|
||
| func (vt *VersaTree) UpdateStorage(_ common.Address, key, value []byte) error { | ||
| vt.CheckStorageTree() | ||
| return vt.db.Put(vt.handler, key, value) | ||
| } | ||
|
|
||
| func (vt *VersaTree) DeleteAccount(address common.Address) error { | ||
| vt.CheckAccountTree() | ||
| return vt.db.Delete(vt.handler, address.Bytes()) | ||
| } | ||
|
|
||
| func (vt *VersaTree) DeleteStorage(_ common.Address, key []byte) error { | ||
| vt.CheckStorageTree() | ||
| return vt.db.Delete(vt.handler, key) | ||
| } | ||
|
|
||
| func (vt *VersaTree) UpdateContractCode(address common.Address, codeHash common.Hash, code []byte) error { | ||
| return nil | ||
| } | ||
|
|
||
| func (vt *VersaTree) Hash() common.Hash { | ||
| hash, err := vt.db.CalcRootHash(vt.handler) | ||
| if err != nil { | ||
| log.Warn("failed to cacl versa tree hash", "error", err) | ||
| } | ||
| return hash | ||
| } | ||
|
|
||
| func (vt *VersaTree) Commit(_ bool) (common.Hash, *trienode.NodeSet, error) { | ||
| hash, err := vt.db.Commit(vt.handler) | ||
| if err != nil { | ||
| log.Warn("failed to commit versa tree", "error", err) | ||
| } | ||
| return hash, nil, nil | ||
| } | ||
|
|
||
| func (vt *VersaTree) NodeIterator(startKey []byte) (trie.NodeIterator, error) { | ||
| panic("versa tree not support iterate") | ||
| return nil, nil | ||
| } | ||
|
|
||
| func (vt *VersaTree) Prove(key []byte, proofDb ethdb.KeyValueWriter) error { | ||
| panic("versa tree not support prove") | ||
| return nil | ||
| } | ||
|
|
||
| // TODO:: test code, will be deleted after stabilization | ||
| func (vt *VersaTree) CheckAccountTree() { | ||
| if !vt.accountTree { | ||
| panic("sub tree can't operate account") | ||
| } | ||
| } | ||
|
|
||
| // TODO:: test code, will be deleted after stabilization | ||
| func (vt *VersaTree) CheckStorageTree() { | ||
| if vt.accountTree { | ||
| panic("root tree can't operate storage") | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.