github/prysm
1
1
Fork 0
mirror of https://github.com/OffchainLabs/prysm.git synced 2026-01-08 23:18:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

p2p: Return error if no ChainState found for a peer (#9502)

Browse Source 
* p2p: Return error if no ChainState found for a peer

* add commentary
This commit is contained in:
Preston Van Loon
2021-08-31 17:25:54 -05:00
committed by GitHub
parent 6ee0a7e811
commit dad03ade77
3 changed files with 42 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
beacon-chain/p2p/peers/status.go
View File

@@ -33,6 +33,7 @@ import (
"github.com/libp2p/go-libp2p-core/peer"
ma "github.com/multiformats/go-multiaddr"
manet "github.com/multiformats/go-multiaddr/net"
"github.com/pkg/errors"
types "github.com/prysmaticlabs/eth2-types"
"github.com/prysmaticlabs/go-bitfield"
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
@@ -74,6 +75,11 @@ const (
MaxBackOffDuration = 5000
)
// ErrNoPeerStatus is returned when there is a map entry for a given peer but there is no chain
// status for that peer. This should happen in rare circumstances only, but is a very possible
// scenario in a chaotic and adversarial network.
var ErrNoPeerStatus = errors.New("no chain status for peer")
// Status is the structure holding the peer status information.
type Status struct {
ctx context.Context
@@ -190,10 +196,17 @@ func (p *Status) SetChainState(pid peer.ID, chainState *pb.Status) {
}
// ChainState gets the chain state of the given remote peer.
// This can return nil if there is no known chain state for the peer.
// This will error if the peer does not exist.
// This will error if there is no known chain state for the peer.
func (p *Status) ChainState(pid peer.ID) (*pb.Status, error) {
return p.scorers.PeerStatusScorer().PeerStatus(pid)
s, err := p.scorers.PeerStatusScorer().PeerStatus(pid)
if err != nil {
return nil, err
}
if s == nil {
return nil, ErrNoPeerStatus
}
return s, nil
}
// IsActive checks if a peers is active and returns the result appropriately.