Banning peers sending bad gossip

cl/cltypes/ssz/ssz.go

@@ -99,7 +99,9 @@ func DecodeDynamicList[T Unmarshaler](bytes []byte, start, end uint32, max uint6
 			return nil, ErrBadOffset
 		}
 		objs[i] = objs[i].Clone().(T)
-		objs[i].DecodeSSZ(buf[currentOffset:endOffset])
+		if err := objs[i].DecodeSSZ(buf[currentOffset:endOffset]); err != nil {
+			return nil, err
+		}
 		currentOffset = endOffset
 	}
 	return objs, nil
@@ -121,7 +123,9 @@ func DecodeStaticList[T Unmarshaler](bytes []byte, start, end, bytesPerElement u
 	objs := make([]T, elementsNum)
 	for i := range objs {
 		objs[i] = objs[i].Clone().(T)
-		objs[i].DecodeSSZ(buf[i*int(bytesPerElement):])
+		if err := objs[i].DecodeSSZ(buf[i*int(bytesPerElement):]); err != nil {
+			return nil, err
+		}
 	}
 	return objs, nil
 }

cmd/lightclient/lightclient/lightclient.go

@@ -145,7 +145,7 @@ func (l *LightClient) Start() {
 		for _, update := range updates {
 			err := l.processLightClientUpdate(update)
 			if err != nil {
-				log.Warn("Could not validate update", "err", err)
+				log.Debug("Could not validate update", "err", err)
 				updates = []*cltypes.LightClientUpdate{}
 				break
 			}

cmd/lightclient/lightclient/subscriber.go

@@ -60,7 +60,7 @@ Retry:
 				continue Retry
 			}
 
-			log.Warn("[Lightclient] could not read gossip :/", "reason", err)
+			log.Debug("[Lightclient] could not read gossip :/", "reason", err)
 			time.Sleep(time.Second)
 		}
 	}
@@ -79,8 +79,9 @@ func (c *ChainTipSubscriber) subscribeGossip() error {
 			return err
 		}
 		if err := c.handleGossipData(data); err != nil {
-			log.Warn("could not process new gossip",
+			log.Debug("could not process new gossip",
 				"gossipType", data.Type, "reason", err)
+
 		}
 	}
 }
@@ -96,6 +97,9 @@ func (c *ChainTipSubscriber) handleGossipData(data *sentinel.GossipData) error {
 	case sentinel.GossipType_BeaconBlockGossipType:
 		block := &cltypes.SignedBeaconBlock{}
 		if err := block.DecodeSSZWithVersion(data.Data, int(version)); err != nil {
+			if _, err := c.sentinel.BanPeer(c.ctx, data.Peer); err != nil {
+				return err
+			}
 			return fmt.Errorf("could not unmarshall block: %s", err)
 		}
 
@@ -104,6 +108,9 @@ func (c *ChainTipSubscriber) handleGossipData(data *sentinel.GossipData) error {
 	case sentinel.GossipType_LightClientFinalityUpdateGossipType:
 		finalityUpdate := &cltypes.LightClientFinalityUpdate{}
 		if err := finalityUpdate.DecodeSSZWithVersion(data.Data, int(version)); err != nil {
+			if _, err := c.sentinel.BanPeer(c.ctx, data.Peer); err != nil {
+				return err
+			}
 			return fmt.Errorf("could not unmarshall finality update: %s", err)
 		}
 		c.lastUpdate = &cltypes.LightClientUpdate{
@@ -123,6 +130,9 @@ func (c *ChainTipSubscriber) handleGossipData(data *sentinel.GossipData) error {
 
 		optimisticUpdate := &cltypes.LightClientOptimisticUpdate{}
 		if err := optimisticUpdate.DecodeSSZWithVersion(data.Data, int(version)); err != nil {
+			if _, err := c.sentinel.BanPeer(c.ctx, data.Peer); err != nil {
+				return err
+			}
 			return fmt.Errorf("could not unmarshall optimistic update: %s", err)
 		}
 		c.lastUpdate = &cltypes.LightClientUpdate{

cmd/sentinel/sentinel/peers/peers.go

@@ -24,7 +24,7 @@ import (
 )
 
 const (
-	maxBadPeers       = 50
+	maxBadPeers       = 50000
 	maxPeerRecordSize = 1000
 	DefaultMaxPeers   = 33
 	MaxBadResponses   = 50

cmd/sentinel/sentinel/service/notifiers.go

@@ -14,6 +14,7 @@ const (
 type gossipObject struct {
 	data []byte              // gossip data
 	t    sentinel.GossipType // determine which gossip message we are notifying of
+	pid  string
 }
 
 type gossipNotifier struct {
@@ -28,14 +29,15 @@ func newGossipNotifier() *gossipNotifier {
 	}
 }
 
-func (g *gossipNotifier) notify(t sentinel.GossipType, data []byte) {
+func (g *gossipNotifier) notify(t sentinel.GossipType, data []byte, pid string) {
 	g.mu.Lock()
 	defer g.mu.Unlock()
 
 	for _, ch := range g.notifiers {
 		ch <- gossipObject{
 			data: data,
 			t:    t,
+			pid:  pid,
 		}
 	}
 }

cmd/sentinel/sentinel/service/service.go

@@ -14,6 +14,7 @@ import (
 	"github.com/ledgerwatch/erigon/cmd/sentinel/sentinel/communication"
 	"github.com/ledgerwatch/log/v3"
 	pubsub "github.com/libp2p/go-libp2p-pubsub"
+	"github.com/libp2p/go-libp2p/core/peer"
 )
 
 type SentinelServer struct {
@@ -32,6 +33,17 @@ func NewSentinelServer(ctx context.Context, sentinel *sentinel.Sentinel) *Sentin
 	}
 }
 
+//BanPeer(context.Context, *Peer) (*EmptyMessage, error)
+
+func (s *SentinelServer) BanPeer(_ context.Context, p *sentinelrpc.Peer) (*sentinelrpc.EmptyMessage, error) {
+	var pid peer.ID
+	if err := pid.UnmarshalText([]byte(p.Pid)); err != nil {
+		return nil, err
+	}
+	s.sentinel.Peers().BanBadPeer(pid)
+	return &sentinelrpc.EmptyMessage{}, nil
+}
+
 func (s *SentinelServer) SubscribeGossip(_ *sentinelrpc.EmptyMessage, stream sentinelrpc.Sentinel_SubscribeGossipServer) error {
 	// first of all subscribe
 	ch, subId, err := s.gossipNotifier.addSubscriber()
@@ -49,6 +61,9 @@ func (s *SentinelServer) SubscribeGossip(_ *sentinelrpc.EmptyMessage, stream sen
 			if err := stream.Send(&sentinelrpc.GossipData{
 				Data: packet.data,
 				Type: packet.t,
+				Peer: &sentinelrpc.Peer{
+					Pid: packet.pid,
+				},
 			}); err != nil {
 				log.Warn("[Sentinel] Could not relay gossip packet", "reason", err)
 			}
@@ -131,28 +146,33 @@ func (s *SentinelServer) handleGossipPacket(pkt *pubsub.Message) error {
 	var err error
 	log.Trace("[Sentinel Gossip] Received Packet", "topic", pkt.Topic)
 	data := pkt.GetData()
+
 	// If we use snappy codec then decompress it accordingly.
 	if strings.Contains(*pkt.Topic, sentinel.SSZSnappyCodec) {
 		data, err = utils.DecompressSnappy(data)
 		if err != nil {
 			return err
 		}
 	}
+	textPid, err := pkt.ReceivedFrom.MarshalText()
+	if err != nil {
+		return err
+	}
 	// Check to which gossip it belongs to.
 	if strings.Contains(*pkt.Topic, string(sentinel.BeaconBlockTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_BeaconBlockGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_BeaconBlockGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.BeaconAggregateAndProofTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_AggregateAndProofGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_AggregateAndProofGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.VoluntaryExitTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_VoluntaryExitGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_VoluntaryExitGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.ProposerSlashingTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_ProposerSlashingGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_ProposerSlashingGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.AttesterSlashingTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_AttesterSlashingGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_AttesterSlashingGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.LightClientFinalityUpdateTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_LightClientFinalityUpdateGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_LightClientFinalityUpdateGossipType, data, string(textPid))
 	} else if strings.Contains(*pkt.Topic, string(sentinel.LightClientOptimisticUpdateTopic)) {
-		s.gossipNotifier.notify(sentinelrpc.GossipType_LightClientOptimisticUpdateGossipType, data)
+		s.gossipNotifier.notify(sentinelrpc.GossipType_LightClientOptimisticUpdateGossipType, data, string(textPid))
 	}
 	return nil
 }

go.mod

@@ -3,7 +3,7 @@ module github.com/ledgerwatch/erigon
 go 1.18
 
 require (
-	github.com/ledgerwatch/erigon-lib v0.0.0-20230317035131-699c78496921
+	github.com/ledgerwatch/erigon-lib v0.0.0-20230318162133-325f5be53671
 	github.com/ledgerwatch/erigon-snapshot v1.1.1-0.20230306083105-1391330d62a3
 	github.com/ledgerwatch/log/v3 v3.7.0
 	github.com/ledgerwatch/secp256k1 v1.0.0

go.sum

@@ -519,8 +519,8 @@ github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 github.com/kylelemons/godebug v0.0.0-20170224010052-a616ab194758 h1:0D5M2HQSGD3PYPwICLl+/9oulQauOuETfgFvhBDffs0=
 github.com/leanovate/gopter v0.2.9 h1:fQjYxZaynp97ozCzfOyOuAGOU4aU/z37zf/tOujFk7c=
 github.com/leanovate/gopter v0.2.9/go.mod h1:U2L/78B+KVFIx2VmW6onHJQzXtFb+p5y3y2Sh+Jxxv8=
-github.com/ledgerwatch/erigon-lib v0.0.0-20230317035131-699c78496921 h1:OiH6pX3jhXY5JqYGpaw1ieY32Q4WIDaxkWFYcRpr+Xg=
-github.com/ledgerwatch/erigon-lib v0.0.0-20230317035131-699c78496921/go.mod h1:I+1Oys00tH9C8Ow1u5atjh3XasMPE4VoAG5lq8Bi1Zs=
+github.com/ledgerwatch/erigon-lib v0.0.0-20230318162133-325f5be53671 h1:Upu6Rah5e9iQtg4YlkUBttEakRcRmFpa09B/A67rEyo=
+github.com/ledgerwatch/erigon-lib v0.0.0-20230318162133-325f5be53671/go.mod h1:I+1Oys00tH9C8Ow1u5atjh3XasMPE4VoAG5lq8Bi1Zs=
 github.com/ledgerwatch/erigon-snapshot v1.1.1-0.20230306083105-1391330d62a3 h1:tfzawK1gIIgRjVZeANXOr0Ziu+kqCIBuKMe0TXfl5Aw=
 github.com/ledgerwatch/erigon-snapshot v1.1.1-0.20230306083105-1391330d62a3/go.mod h1:3AuPxZc85jkehh/HA9h8gabv5MSi3kb/ddtzBsTVJFo=
 github.com/ledgerwatch/log/v3 v3.7.0 h1:aFPEZdwZx4jzA3+/Pf8wNDN5tCI0cIolq/kfvgcM+og=