feat(dot/network): Add warp sync request handler

dot/network/config.go

@@ -66,6 +66,7 @@ type Config struct {
 	// Service interfaces
 	BlockState         BlockState
 	Syncer             Syncer
+	warpSyncProvider   WarpSyncProvider
 	TransactionHandler TransactionHandler
 
 	// Used to specify the address broadcasted to other peers, and avoids using pubip.Get

dot/network/messages/warp_sync.go

@@ -0,0 +1,51 @@
+// Copyright 2024 ChainSafe Systems (ON)
+// SPDX-License-Identifier: LGPL-3.0-only
+
+package messages
+
+import (
+	"bytes"
+	"fmt"
+
+	"github.com/ChainSafe/gossamer/lib/common"
+	"github.com/ChainSafe/gossamer/pkg/scale"
+)
+
+// WarpProofRequest is a struct for p2p warp proof request
+type WarpProofRequest struct {
+	Begin common.Hash
+}
+
+// Decode decodes the message into a WarpProofRequest
+func (wsr *WarpProofRequest) Decode(in []byte) error {
+	reader := bytes.NewReader(in)
+	sd := scale.NewDecoder(reader)
+	err := sd.Decode(&wsr)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// Encode encodes the warp sync request
+func (wsr *WarpProofRequest) Encode() ([]byte, error) {
+	buffer := bytes.NewBuffer(nil)
+	encoder := scale.NewEncoder(buffer)
+	err := encoder.Encode(wsr)
+	if err != nil {
+		return nil, err
+	}
+	return buffer.Bytes(), nil
+}
+
+// String returns the string representation of a WarpProofRequest
+func (wsr *WarpProofRequest) String() string {
+	if wsr == nil {
+		return "WarpProofRequest=nil"
+	}
+
+	return fmt.Sprintf("WarpProofRequest begin=%v", wsr.Begin)
+}
+
+var _ P2PMessage = (*WarpProofRequest)(nil)

dot/network/mocks_generate_test.go

@@ -6,5 +6,6 @@ package network
 //go:generate mockgen -destination=mock_telemetry_test.go -package $GOPACKAGE . Telemetry
 //go:generate mockgen -destination=mock_syncer_test.go -package $GOPACKAGE . Syncer
 //go:generate mockgen -destination=mock_block_state_test.go -package $GOPACKAGE . BlockState
+//go:generate mockgen -destination=mock_warp_sync_provider_test.go -package $GOPACKAGE . WarpSyncProvider
 //go:generate mockgen -destination=mock_transaction_handler_test.go -package $GOPACKAGE . TransactionHandler
 //go:generate mockgen -destination=mock_stream_test.go -package $GOPACKAGE github.com/libp2p/go-libp2p/core/network Stream

dot/network/service.go

@@ -33,6 +33,7 @@ const (
 
 	// the following are sub-protocols used by the node
 	SyncID          = "/sync/2"
+	WarpSyncID      = "/sync/warp"
 	lightID         = "/light/2"
 	blockAnnounceID = "/block-announces/1"
 	transactionsID  = "/transactions/1"
@@ -129,6 +130,7 @@ type Service struct {
 	blockState         BlockState
 	syncer             Syncer
 	transactionHandler TransactionHandler
+	warpSyncProvider   WarpSyncProvider
 
 	// Configuration options
 	noBootstrap bool
@@ -215,6 +217,7 @@ func NewService(cfg *Config) (*Service, error) {
 		noBootstrap:            cfg.NoBootstrap,
 		noMDNS:                 cfg.NoMDNS,
 		syncer:                 cfg.Syncer,
+		warpSyncProvider:       cfg.warpSyncProvider,
 		notificationsProtocols: make(map[MessageType]*notificationsProtocol),
 		lightRequest:           make(map[peer.ID]struct{}),
 		telemetryInterval:      cfg.telemetryInterval,
@@ -253,6 +256,7 @@ func (s *Service) Start() error {
 	}
 
 	s.host.registerStreamHandler(s.host.protocolID+SyncID, s.handleSyncStream)
+	s.host.registerStreamHandler(s.host.protocolID+WarpSyncID, s.handleWarpSyncStream)
 	s.host.registerStreamHandler(s.host.protocolID+lightID, s.handleLightStream)
 
 	// register block announce protocol

dot/network/warp_sync.go

@@ -0,0 +1,70 @@
+// Copyright 2024 ChainSafe Systems (ON)
+// SPDX-License-Identifier: LGPL-3.0-only
+
+package network
+
+import (
+	"github.com/ChainSafe/gossamer/dot/network/messages"
+	"github.com/ChainSafe/gossamer/lib/common"
+	libp2pnetwork "github.com/libp2p/go-libp2p/core/network"
+	"github.com/libp2p/go-libp2p/core/peer"
+)
+
+// WarpSyncProvider is an interface for generating warp sync proofs
+type WarpSyncProvider interface {
+	// Generate proof starting at given block hash. The proof is accumulated until maximum proof
+	// size is reached.
+	generate(start common.Hash) (encodedProof []byte, err error)
+}
+
+func (s *Service) handleWarpSyncRequest(req messages.WarpProofRequest) ([]byte, error) {
+	// use the backend to generate the warp proof
+	proof, err := s.warpSyncProvider.generate(req.Begin)
+	if err != nil {
+		return nil, err
+	}
+	// send the response through pendingResponse channel
+	return proof, nil
+}
+
+func (s *Service) handleWarpSyncStream(stream libp2pnetwork.Stream) {
+	if stream == nil {
+		return
+	}
+
+	s.readStream(stream, decodeSyncMessage, s.handleWarpSyncMessage, MaxBlockResponseSize)
+}
+
+func decodeWarpSyncMessage(in []byte, _ peer.ID, _ bool) (messages.P2PMessage, error) {
+	msg := new(messages.WarpProofRequest)
+	err := msg.Decode(in)
+	return msg, err
+}
+
+func (s *Service) handleWarpSyncMessage(stream libp2pnetwork.Stream, msg messages.P2PMessage) error {
+	if msg == nil {
+		return nil
+	}
+
+	defer func() {
+		err := stream.Close()
+		if err != nil && err.Error() != ErrStreamReset.Error() {
+			logger.Warnf("failed to close stream: %s", err)
+		}
+	}()
+
+	if req, ok := msg.(*messages.WarpProofRequest); ok {
+		resp, err := s.handleWarpSyncRequest(*req)
+		if err != nil {
+			logger.Debugf("cannot create response for request: %s", err)
+			return nil
+		}
+
+		if _, err = stream.Write(resp); err != nil {
+			logger.Debugf("failed to send WarpSyncResponse message to peer %s: %s", stream.Conn().RemotePeer(), err)
+			return err
+		}
+	}
+
+	return nil
+}

dot/network/warp_sync_test.go

@@ -0,0 +1,103 @@
+// Copyright 2024 ChainSafe Systems (ON)
+// SPDX-License-Identifier: LGPL-3.0-only
+
+package network
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/ChainSafe/gossamer/dot/network/messages"
+	"github.com/ChainSafe/gossamer/lib/common"
+	"github.com/libp2p/go-libp2p/core/peer"
+	"github.com/stretchr/testify/require"
+	gomock "go.uber.org/mock/gomock"
+)
+
+func TestDecodeWarpSyncMessage(t *testing.T) {
+	t.Parallel()
+
+	// Basic WarpProofRequest
+	testWarpReqMessage := &messages.WarpProofRequest{
+		Begin: common.MustBlake2bHash([]byte("test")),
+	}
+
+	// Test encoding
+	reqEnc, err := testWarpReqMessage.Encode()
+	require.NoError(t, err)
+
+	// Test decoding
+	testPeer := peer.ID("me")
+	msg, err := decodeWarpSyncMessage(reqEnc, testPeer, true)
+	require.NoError(t, err)
+
+	req, ok := msg.(*messages.WarpProofRequest)
+	require.True(t, ok)
+	require.Equal(t, testWarpReqMessage, req)
+}
+
+// createServiceWithWarpSyncHelper creates a basic service with warp sync handler support
+func createServiceWithWarpSyncHelper(t *testing.T, warpSyncProvider WarpSyncProvider) *Service {
+	t.Helper()
+
+	config := &Config{
+		BasePath:         t.TempDir(),
+		Port:             availablePort(t),
+		NoBootstrap:      true,
+		NoMDNS:           true,
+		warpSyncProvider: warpSyncProvider,
+	}
+
+	srvc := createTestService(t, config)
+	srvc.noGossip = true
+	handler := newTestStreamHandler(decodeSyncMessage)
+	srvc.host.registerStreamHandler(srvc.host.protocolID, handler.handleStream)
+
+	return srvc
+}
+
+func TestHandleWarpSyncRequestOk(t *testing.T) {
+	t.Parallel()
+
+	// Creates warp sync provider mock to generate proofs with the expected result
+	expectedProof := []byte{0x01}
+
+	ctrl := gomock.NewController(t)
+	warpSyncProvider := NewMockWarpSyncProvider(ctrl)
+	warpSyncProvider.EXPECT().generate(common.EmptyHash).Return(expectedProof, nil).Times(1)
+
+	// Initiate service using the warp sync provider mock
+	srvc := createServiceWithWarpSyncHelper(t, warpSyncProvider)
+
+	// Handle request and check resulting proof
+	req := messages.WarpProofRequest{
+		Begin: common.EmptyHash,
+	}
+
+	resp, err := srvc.handleWarpSyncRequest(req)
+	require.NoError(t, err)
+	require.Equal(t, expectedProof, resp)
+}
+
+func TestHandleWarpSyncRequestError(t *testing.T) {
+	t.Parallel()
+
+	// Creates warp sync provider mock to generate proofs with the expected erro
+	expectedError := fmt.Errorf("error generating proof")
+	ctrl := gomock.NewController(t)
+
+	warpSyncProvider := NewMockWarpSyncProvider(ctrl)
+	warpSyncProvider.EXPECT().generate(common.EmptyHash).Return(nil, expectedError).Times(1)
+
+	// Initiate service using the warp sync provider mock
+	srvc := createServiceWithWarpSyncHelper(t, warpSyncProvider)
+
+	// Handle request and check resulting error
+	req := messages.WarpProofRequest{
+		Begin: common.EmptyHash,
+	}
+
+	resp, err := srvc.handleWarpSyncRequest(req)
+	require.Nil(t, resp)
+	require.ErrorIs(t, err, expectedError)
+}