blocklist peer on retrieval timeout (#685)

pkg/node/node.go

@@ -238,7 +238,7 @@ func NewBee(addr string, swarmAddress swarm.Address, keystore keystore.Service,
 
 	chunkvalidator := swarm.NewChunkValidator(soc.NewValidator(), content.NewValidator())
 
-	retrieve := retrieval.New(p2ps, kad, logger, acc, accounting.NewFixedPricer(swarmAddress, 10), chunkvalidator)
+	retrieve := retrieval.New(p2ps, storer, kad, logger, acc, accounting.NewFixedPricer(swarmAddress, 10), chunkvalidator)
 	tagg := tags.NewTags(stateStore, logger)
 	b.tagsCloser = tagg
 
@@ -257,7 +257,6 @@ func NewBee(addr string, swarmAddress swarm.Address, keystore keystore.Service,
 	} else {
 		ns = netstore.New(storer, nil, retrieve, logger, chunkvalidator)
 	}
-	retrieve.SetStorer(ns)
 
 	pushSyncProtocol := pushsync.New(p2ps, storer, kad, tagg, psss.TryUnwrap, logger, acc, accounting.NewFixedPricer(swarmAddress, 10))
 

pkg/p2p/p2p.go

@@ -20,13 +20,17 @@ type Service interface {
 	AddProtocol(ProtocolSpec) error
 	// Connect to a peer but do not notify topology about the established connection.
 	Connect(ctx context.Context, addr ma.Multiaddr) (address *bzz.Address, err error)
+	Disconnecter
+	Peers() []Peer
+	AddNotifier(topology.Notifier)
+	Addresses() ([]ma.Multiaddr, error)
+}
+
+type Disconnecter interface {
 	Disconnect(overlay swarm.Address) error
 	// Blocklist will disconnect a peer and put it on a blocklist (blocking in & out connections) for provided duration
 	// duration 0 is treated as an infinite duration
 	Blocklist(overlay swarm.Address, duration time.Duration) error
-	Peers() []Peer
-	AddNotifier(topology.Notifier)
-	Addresses() ([]ma.Multiaddr, error)
 }
 
 // DebugService extends the Service with method used for debugging.
@@ -41,6 +45,11 @@ type Streamer interface {
 	NewStream(ctx context.Context, address swarm.Address, h Headers, protocol, version, stream string) (Stream, error)
 }
 
+type StreamerDisconnecter interface {
+	Streamer
+	Disconnecter
+}
+
 // Stream represent a bidirectional data Stream.
 type Stream interface {
 	io.ReadWriter

pkg/p2p/streamtest/streamtest.go

@@ -319,3 +319,52 @@ type Option interface {
 type optionFunc func(*Recorder)
 
 func (f optionFunc) apply(r *Recorder) { f(r) }
+
+var _ p2p.StreamerDisconnecter = (*RecorderDisconnecter)(nil)
+
+type RecorderDisconnecter struct {
+	*Recorder
+	disconnected map[string]struct{}
+	blocklisted  map[string]time.Duration
+	mu           sync.RWMutex
+}
+
+func NewRecorderDisconnecter(r *Recorder) *RecorderDisconnecter {
+	return &RecorderDisconnecter{
+		Recorder:     r,
+		disconnected: make(map[string]struct{}),
+		blocklisted:  make(map[string]time.Duration),
+	}
+}
+
+func (r *RecorderDisconnecter) Disconnect(overlay swarm.Address) error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
+	r.disconnected[overlay.String()] = struct{}{}
+	return nil
+}
+
+func (r *RecorderDisconnecter) Blocklist(overlay swarm.Address, d time.Duration) error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
+	r.blocklisted[overlay.String()] = d
+	return nil
+}
+
+func (r *RecorderDisconnecter) IsDisconnected(overlay swarm.Address) bool {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+
+	_, yes := r.disconnected[overlay.String()]
+	return yes
+}
+
+func (r *RecorderDisconnecter) IsBlocklisted(overlay swarm.Address) (bool, time.Duration) {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+
+	d, yes := r.blocklisted[overlay.String()]
+	return yes, d
+}

pkg/recovery/repair_test.go

@@ -269,13 +269,11 @@ func newTestNetStore(t *testing.T, recoveryFunc recovery.RecoveryHook) storage.S
 		_, _, _ = f(peerID, 0)
 		return nil
 	}}
-	server := retrieval.New(nil, nil, logger, serverMockAccounting, nil, nil)
-	server.SetStorer(mockStorer)
-	recorder := streamtest.New(
+	server := retrieval.New(nil, mockStorer, ps, logger, serverMockAccounting, nil, nil)
+	recorder := streamtest.NewRecorderDisconnecter(streamtest.New(
 		streamtest.WithProtocols(server.Protocol()),
-	)
-	retrieve := retrieval.New(recorder, ps, logger, serverMockAccounting, pricerMock, nil)
-	retrieve.SetStorer(mockStorer)
+	))
+	retrieve := retrieval.New(recorder, mockStorer, ps, logger, serverMockAccounting, pricerMock, nil)
 	ns := netstore.New(storer, recoveryFunc, retrieve, logger, nil)
 	return ns
 }

pkg/retrieval/retrieval.go

@@ -6,6 +6,7 @@ package retrieval
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"time"
 
@@ -35,7 +36,7 @@ type Interface interface {
 }
 
 type Service struct {
-	streamer      p2p.Streamer
+	streamer      p2p.StreamerDisconnecter
 	peerSuggester topology.EachPeerer
 	storer        storage.Storer
 	singleflight  singleflight.Group
@@ -45,10 +46,11 @@ type Service struct {
 	validator     swarm.Validator
 }
 
-func New(streamer p2p.Streamer, chunkPeerer topology.EachPeerer, logger logging.Logger, accounting accounting.Interface, pricer accounting.Pricer, validator swarm.Validator) *Service {
+func New(streamer p2p.StreamerDisconnecter, storer storage.Storer, chunkPeerer topology.EachPeerer, logger logging.Logger, accounting accounting.Interface, pricer accounting.Pricer, validator swarm.Validator) *Service {
 	return &Service{
 		streamer:      streamer,
 		peerSuggester: chunkPeerer,
+		storer:        storer,
 		logger:        logger,
 		accounting:    accounting,
 		pricer:        pricer,
@@ -72,11 +74,10 @@ func (s *Service) Protocol() p2p.ProtocolSpec {
 const (
 	maxPeers             = 5
 	retrieveChunkTimeout = 10 * time.Second
+	blocklistDuration    = time.Minute
 )
 
 func (s *Service) RetrieveChunk(ctx context.Context, addr swarm.Address) (swarm.Chunk, error) {
-	ctx, cancel := context.WithTimeout(ctx, maxPeers*retrieveChunkTimeout)
-	defer cancel()
 
 	v, err, _ := s.singleflight.Do(addr.String(), func() (interface{}, error) {
 		var skipPeers []swarm.Address
@@ -89,6 +90,14 @@ func (s *Service) RetrieveChunk(ctx context.Context, addr swarm.Address) (swarm.
 				}
 				s.logger.Debugf("retrieval: failed to get chunk %s from peer %s: %v", addr, peer, err)
 				skipPeers = append(skipPeers, peer)
+				if errors.Is(err, context.DeadlineExceeded) {
+					if err := s.streamer.Blocklist(peer, blocklistDuration); err != nil {
+						s.logger.Errorf("retrieval: unable to block peer %s", peer)
+						s.logger.Debugf("retrieval: blocking peer %s: %v", peer, err)
+					} else {
+						s.logger.Warningf("retrieval: peer %s blocked as unresponsive", peer)
+					}
+				}
 				continue
 			}
 			s.logger.Tracef("retrieval: got chunk %s from peer %s", addr, peer)
@@ -124,7 +133,7 @@ func (s *Service) retrieveChunk(ctx context.Context, addr swarm.Address, skipPee
 	chunkPrice := s.pricer.PeerPrice(peer, addr)
 	err = s.accounting.Reserve(peer, chunkPrice)
 	if err != nil {
-		return nil, peer, err
+		return nil, peer, fmt.Errorf("accounting retrieve: %w", err)
 	}
 	defer s.accounting.Release(peer, chunkPrice)
 
@@ -146,26 +155,26 @@ func (s *Service) retrieveChunk(ctx context.Context, addr swarm.Address, skipPee
 	if err := w.WriteMsgWithContext(ctx, &pb.Request{
 		Addr: addr.Bytes(),
 	}); err != nil {
-		return nil, peer, fmt.Errorf("write request: %w peer %s", err, peer.String())
+		return nil, peer, fmt.Errorf("write request: %w", err)
 	}
 
 	var d pb.Delivery
 	if err := r.ReadMsgWithContext(ctx, &d); err != nil {
-		return nil, peer, fmt.Errorf("read delivery: %w peer %s", err, peer.String())
+		return nil, peer, fmt.Errorf("read delivery: %w", err)
 	}
 
 	// credit the peer after successful delivery
 	chunk = swarm.NewChunk(addr, d.Data)
 	if !s.validator.Validate(chunk) {
-		return nil, peer, err
+		return nil, peer, fmt.Errorf("new chunk: %w", err)
 	}
 
 	err = s.accounting.Credit(peer, chunkPrice)
 	if err != nil {
-		return nil, peer, err
+		return nil, peer, fmt.Errorf("accounting credit: %w", err)
 	}
 
-	return chunk, peer, err
+	return chunk, peer, nil
 }
 
 func (s *Service) closestPeer(addr swarm.Address, skipPeers []swarm.Address) (swarm.Address, error) {
@@ -219,18 +228,27 @@ func (s *Service) handler(ctx context.Context, p p2p.Peer, stream p2p.Stream) (e
 	}()
 	var req pb.Request
 	if err := r.ReadMsg(&req); err != nil {
-		return fmt.Errorf("read request: %w peer %s", err, p.Address.String())
+		return fmt.Errorf("read request: %w", err)
 	}
 	ctx = context.WithValue(ctx, requestSourceContextKey{}, p.Address.String())
-	chunk, err := s.storer.Get(ctx, storage.ModeGetRequest, swarm.NewAddress(req.Addr))
+	addr := swarm.NewAddress(req.Addr)
+	chunk, err := s.storer.Get(ctx, storage.ModeGetRequest, addr)
+	if err != nil {
+		if errors.Is(err, storage.ErrNotFound) {
+			// forward the request
+			chunk, err = s.RetrieveChunk(ctx, addr)
 			if err != nil {
-		return fmt.Errorf("get from store: %w peer %s", err, p.Address.String())
+				return fmt.Errorf("retrieve chunk: %w", err)
+			}
+		} else {
+			return fmt.Errorf("get from store: %w", err)
+		}
 	}
 
 	if err := w.WriteMsgWithContext(ctx, &pb.Delivery{
 		Data: chunk.Data(),
 	}); err != nil {
-		return fmt.Errorf("write delivery: %w peer %s", err, p.Address.String())
+		return fmt.Errorf("write delivery: %w", err)
 	}
 
 	// compute the price we charge for this chunk and debit it from p's balance
@@ -242,8 +260,3 @@ func (s *Service) handler(ctx context.Context, p p2p.Peer, stream p2p.Stream) (e
 
 	return nil
 }
-
-// SetStorer sets the storer. This call is not goroutine safe.
-func (s *Service) SetStorer(storer storage.Storer) {
-	s.storer = storer
-}

pkg/retrieval/retrieval_test.go

@@ -51,11 +51,10 @@ func TestDelivery(t *testing.T) {
 	pricerMock := accountingmock.NewPricer(price, price)
 
 	// create the server that will handle the request and will serve the response
-	server := retrieval.New(nil, nil, logger, serverMockAccounting, pricerMock, mockValidator)
-	server.SetStorer(mockStorer)
-	recorder := streamtest.New(
+	server := retrieval.New(nil, mockStorer, nil, logger, serverMockAccounting, pricerMock, mockValidator)
+	recorder := streamtest.NewRecorderDisconnecter(streamtest.New(
 		streamtest.WithProtocols(server.Protocol()),
-	)
+	))
 
 	clientMockAccounting := accountingmock.NewAccounting()
 
@@ -70,8 +69,7 @@ func TestDelivery(t *testing.T) {
 		_, _, _ = f(peerID, 0)
 		return nil
 	}}
-	client := retrieval.New(recorder, ps, logger, clientMockAccounting, pricerMock, mockValidator)
-	client.SetStorer(clientMockStorer)
+	client := retrieval.New(recorder, clientMockStorer, ps, logger, clientMockAccounting, pricerMock, mockValidator)
 	ctx, cancel := context.WithTimeout(context.Background(), testTimeout)
 	defer cancel()
 	v, err := client.RetrieveChunk(ctx, reqAddr)