roysc/plugeth
1
0
Fork 0
forked from cerc-io/plugeth
Code Pull Requests Activity

les, light: verbose errors on state retrieval issues

Browse Source
This commit is contained in:
Péter Szilágyi 2019-03-18 13:19:40 +02:00
parent 54cd3e89a4
commit 211ec46284
No known key found for this signature in database
GPG Key ID: E9AE538CEDF8293D
2 changed files with 116 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

185
les/handler.go
View File

@ -72,7 +72,7 @@ type BlockChain interface {
GetHeaderByHash(hash common.Hash) *types.Header GetHeaderByHash(hash common.Hash) *types.Header
CurrentHeader() *types.Header CurrentHeader() *types.Header
GetTd(hash common.Hash, number uint64) *big.Int GetTd(hash common.Hash, number uint64) *big.Int
State() (*state.StateDB, error) StateCache() state.Database
InsertHeaderChain(chain []*types.Header, checkFreq int) (int, error) InsertHeaderChain(chain []*types.Header, checkFreq int) (int, error)
Rollback(chain []common.Hash) Rollback(chain []common.Hash)
GetHeaderByNumber(number uint64) *types.Header GetHeaderByNumber(number uint64) *types.Header
@ -642,24 +642,33 @@ func (pm *ProtocolManager) handleMsg(p *peer) error {
if i != 0 && !task.waitOrStop() { if i != 0 && !task.waitOrStop() {
return return
} }
// Retrieve the requested state entry, stopping if enough was found // Look up the root hash belonging to the request
if number := rawdb.ReadHeaderNumber(pm.chainDb, req.BHash); number != nil { number := rawdb.ReadHeaderNumber(pm.chainDb, req.BHash)
if header := rawdb.ReadHeader(pm.chainDb, req.BHash, *number); header != nil { if number == nil {
statedb, err := pm.blockchain.State() p.Log().Warn("Failed to retrieve block num for code", "hash", req.BHash)
if err != nil { continue
continue }
} header := rawdb.ReadHeader(pm.chainDb, req.BHash, *number)
account, err := pm.getAccount(statedb, header.Root, common.BytesToHash(req.AccKey)) if header == nil {
if err != nil { p.Log().Warn("Failed to retrieve header for code", "block", *number, "hash", req.BHash)
continue continue
} }
code, _ := statedb.Database().TrieDB().Node(common.BytesToHash(account.CodeHash)) triedb := pm.blockchain.StateCache().TrieDB()
data = append(data, code) account, err := pm.getAccount(triedb, header.Root, common.BytesToHash(req.AccKey))
if bytes += len(code); bytes >= softResponseLimit { if err != nil {
break p.Log().Warn("Failed to retrieve account for code", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "err", err)
} continue
} }
code, err := triedb.Node(common.BytesToHash(account.CodeHash))
if err != nil {
p.Log().Warn("Failed to retrieve account code", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "codehash", common.BytesToHash(account.CodeHash), "err", err)
continue
}
// Accumulate the code and abort if enough data was retrieved
data = append(data, code)
if bytes += len(code); bytes >= softResponseLimit {
break
} }
} }
sendResponse(req.ReqID, uint64(reqCnt), p.ReplyCode(req.ReqID, data), task.done()) sendResponse(req.ReqID, uint64(reqCnt), p.ReplyCode(req.ReqID, data), task.done())
@ -779,33 +788,51 @@ func (pm *ProtocolManager) handleMsg(p *peer) error {
if i != 0 && !task.waitOrStop() { if i != 0 && !task.waitOrStop() {
return return
} }
// Retrieve the requested state entry, stopping if enough was found // Look up the root hash belonging to the request
if number := rawdb.ReadHeaderNumber(pm.chainDb, req.BHash); number != nil { number := rawdb.ReadHeaderNumber(pm.chainDb, req.BHash)
if header := rawdb.ReadHeader(pm.chainDb, req.BHash, *number); header != nil { if number == nil {
statedb, err := pm.blockchain.State() p.Log().Warn("Failed to retrieve block num for proof", "hash", req.BHash)
if err != nil { continue
continue }
} header := rawdb.ReadHeader(pm.chainDb, req.BHash, *number)
var trie state.Trie if header == nil {
if len(req.AccKey) > 0 { p.Log().Warn("Failed to retrieve header for proof", "block", *number, "hash", req.BHash)
account, err := pm.getAccount(statedb, header.Root, common.BytesToHash(req.AccKey)) continue
if err != nil { }
continue // Open the account or storage trie for the request
} statedb := pm.blockchain.StateCache()
trie, _ = statedb.Database().OpenStorageTrie(common.BytesToHash(req.AccKey), account.Root)
} else {
trie, _ = statedb.Database().OpenTrie(header.Root)
}
if trie != nil {
var proof light.NodeList
trie.Prove(req.Key, 0, &proof)
proofs = append(proofs, proof) var trie state.Trie
if bytes += proof.DataSize(); bytes >= softResponseLimit { switch len(req.AccKey) {
break case 0:
} // No account key specified, open an account trie
} trie, err = statedb.OpenTrie(header.Root)
if trie == nil || err != nil {
p.Log().Warn("Failed to open storage trie for proof", "block", header.Number, "hash", header.Hash(), "root", header.Root, "err", err)
continue
} }
default:
// Account key specified, open a storage trie
account, err := pm.getAccount(statedb.TrieDB(), header.Root, common.BytesToHash(req.AccKey))
if err != nil {
p.Log().Warn("Failed to retrieve account for proof", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "err", err)
continue
}
trie, err = statedb.OpenStorageTrie(common.BytesToHash(req.AccKey), account.Root)
if trie == nil || err != nil {
p.Log().Warn("Failed to open storage trie for proof", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "root", account.Root, "err", err)
continue
}
}
// Prove the user's request from the account or stroage trie
var proof light.NodeList
if err := trie.Prove(req.Key, 0, &proof); err != nil {
p.Log().Warn("Failed to prove state request", "block", header.Number, "hash", header.Hash(), "err", err)
continue
}
proofs = append(proofs, proof)
if bytes += proof.DataSize(); bytes >= softResponseLimit {
break
} }
} }
sendResponse(req.ReqID, uint64(reqCnt), p.ReplyProofs(req.ReqID, proofs), task.done()) sendResponse(req.ReqID, uint64(reqCnt), p.ReplyProofs(req.ReqID, proofs), task.done())
@ -824,7 +851,6 @@ func (pm *ProtocolManager) handleMsg(p *peer) error {
// Gather state data until the fetch or network limits is reached // Gather state data until the fetch or network limits is reached
var ( var (
lastBHash common.Hash lastBHash common.Hash
statedb *state.StateDB
root common.Hash root common.Hash
) )
reqCnt := len(req.Reqs) reqCnt := len(req.Reqs)
@ -832,43 +858,60 @@ func (pm *ProtocolManager) handleMsg(p *peer) error {
return errResp(ErrRequestRejected, "") return errResp(ErrRequestRejected, "")
} }
go func() { go func() {
nodes := light.NewNodeSet() nodes := light.NewNodeSet()
for i, req := range req.Reqs { for i, req := range req.Reqs {
if i != 0 && !task.waitOrStop() { if i != 0 && !task.waitOrStop() {
return return
} }
// Look up the state belonging to the request // Look up the root hash belonging to the request
if statedb == nil || req.BHash != lastBHash { var (
statedb, root, lastBHash = nil, common.Hash{}, req.BHash number *uint64
header *types.Header
trie state.Trie
)
if req.BHash != lastBHash {
root, lastBHash = common.Hash{}, req.BHash
if number := rawdb.ReadHeaderNumber(pm.chainDb, req.BHash); number != nil { if number = rawdb.ReadHeaderNumber(pm.chainDb, req.BHash); number == nil {
if header := rawdb.ReadHeader(pm.chainDb, req.BHash, *number); header != nil { p.Log().Warn("Failed to retrieve block num for proof", "hash", req.BHash)
statedb, _ = pm.blockchain.State()
root = header.Root
}
}
}
if statedb == nil {
continue
}
// Pull the account or storage trie of the request
var trie state.Trie
if len(req.AccKey) > 0 {
account, err := pm.getAccount(statedb, root, common.BytesToHash(req.AccKey))
if err != nil {
continue continue
} }
trie, _ = statedb.Database().OpenStorageTrie(common.BytesToHash(req.AccKey), account.Root) if header = rawdb.ReadHeader(pm.chainDb, req.BHash, *number); header == nil {
} else { p.Log().Warn("Failed to retrieve header for proof", "block", *number, "hash", req.BHash)
trie, _ = statedb.Database().OpenTrie(root) continue
}
root = header.Root
} }
if trie == nil { // Open the account or storage trie for the request
continue statedb := pm.blockchain.StateCache()
switch len(req.AccKey) {
case 0:
// No account key specified, open an account trie
trie, err = statedb.OpenTrie(root)
if trie == nil || err != nil {
p.Log().Warn("Failed to open storage trie for proof", "block", header.Number, "hash", header.Hash(), "root", root, "err", err)
continue
}
default:
// Account key specified, open a storage trie
account, err := pm.getAccount(statedb.TrieDB(), root, common.BytesToHash(req.AccKey))
if err != nil {
p.Log().Warn("Failed to retrieve account for proof", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "err", err)
continue
}
trie, err = statedb.OpenStorageTrie(common.BytesToHash(req.AccKey), account.Root)
if trie == nil || err != nil {
p.Log().Warn("Failed to open storage trie for proof", "block", header.Number, "hash", header.Hash(), "account", common.BytesToHash(req.AccKey), "root", account.Root, "err", err)
continue
}
} }
// Prove the user's request from the account or stroage trie // Prove the user's request from the account or stroage trie
trie.Prove(req.Key, req.FromLevel, nodes) if err := trie.Prove(req.Key, req.FromLevel, nodes); err != nil {
p.Log().Warn("Failed to prove state request", "block", header.Number, "hash", header.Hash(), "err", err)
continue
}
if nodes.DataSize() >= softResponseLimit { if nodes.DataSize() >= softResponseLimit {
break break
} }
@ -1190,8 +1233,8 @@ func (pm *ProtocolManager) handleMsg(p *peer) error {
} }
// getAccount retrieves an account from the state based at root. // getAccount retrieves an account from the state based at root.
func (pm *ProtocolManager) getAccount(statedb *state.StateDB, root, hash common.Hash) (state.Account, error) { func (pm *ProtocolManager) getAccount(triedb *trie.Database, root, hash common.Hash) (state.Account, error) {
trie, err := trie.New(root, statedb.Database().TrieDB()) trie, err := trie.New(root, triedb)
if err != nil { if err != nil {
return state.Account{}, err return state.Account{}, err
} }

5
light/lightchain.go
View File

@ -211,9 +211,8 @@ func (lc *LightChain) Genesis() *types.Block {
return lc.genesisBlock return lc.genesisBlock
} }
// State returns a new mutable state based on the current HEAD block. func (lc *LightChain) StateCache() state.Database {
func (lc *LightChain) State() (*state.StateDB, error) { panic("not implemented")
return nil, errors.New("not implemented, needs client/server interface split")
} }
// GetBody retrieves a block body (transactions and uncles) from the database // GetBody retrieves a block body (transactions and uncles) from the database