feat(consensus/grandpa): Add warp sync proof verification (#4240)

dot/network/warp_sync.go

@@ -8,18 +8,33 @@ import (
 	"fmt"
 
 	"github.com/ChainSafe/gossamer/dot/network/messages"
+	"github.com/ChainSafe/gossamer/dot/types"
+	"github.com/ChainSafe/gossamer/internal/primitives/consensus/grandpa"
+	primitives "github.com/ChainSafe/gossamer/internal/primitives/consensus/grandpa"
 	"github.com/ChainSafe/gossamer/lib/common"
 	libp2pnetwork "github.com/libp2p/go-libp2p/core/network"
 	"github.com/libp2p/go-libp2p/core/peer"
 )
 
 const MaxAllowedSameRequestPerPeer = 5
 
+type WarpSyncVerificationResult struct {
+	SetId         grandpa.SetID
+	AuthorityList primitives.AuthorityList
+	Header        types.Header
+	Completed     bool
+}
+
 // 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)
+	Verify(
+		encodedProof []byte,
+		setId grandpa.SetID,
+		authorities primitives.AuthorityList,
+	) (*WarpSyncVerificationResult, error)
 }
 
 func (s *Service) handleWarpSyncRequest(req messages.WarpProofRequest) ([]byte, error) {

dot/services.go

@@ -20,7 +20,6 @@ import (
 	"github.com/ChainSafe/gossamer/dot/sync"
 	"github.com/ChainSafe/gossamer/dot/system"
 	"github.com/ChainSafe/gossamer/dot/types"
-	consensus_grandpa "github.com/ChainSafe/gossamer/internal/client/consensus/grandpa"
 	"github.com/ChainSafe/gossamer/internal/database"
 	"github.com/ChainSafe/gossamer/internal/log"
 	"github.com/ChainSafe/gossamer/internal/metrics"
@@ -350,7 +349,7 @@ func (nodeBuilder) createNetworkService(config *cfg.Config, stateSrvc *state.Ser
 		return nil, fmt.Errorf("failed to parse network log level: %w", err)
 	}
 
-	warpSyncProvider := consensus_grandpa.NewWarpSyncProofProvider(
+	warpSyncProvider := grandpa.NewWarpSyncProofProvider(
 		stateSrvc.Block, stateSrvc.Grandpa,
 	)
 

dot/types/authority.go

@@ -167,3 +167,5 @@ func AuthoritiesRawToAuthorityAsAddress(authsRaw []AuthorityRaw, kt crypto.KeyTy
 	}
 	return auths, nil
 }
+
+type AuthorityList []Authority

internal/client/consensus/grandpa/justification.go

@@ -42,7 +42,7 @@ type decodeGrandpaJustification[
 	Hasher runtime.Hasher[Hash],
 ] GrandpaJustification[Hash, N]
 
-func decodeJustification[
+func DecodeJustification[
 	Hash runtime.Hash,
 	N runtime.Number,
 	Hasher runtime.Hasher[Hash],
@@ -100,7 +100,7 @@ func DecodeGrandpaJustificationVerifyFinalizes[
 	setID uint64,
 	voters grandpa.VoterSet[string],
 ) (GrandpaJustification[Hash, N], error) {
-	justification, err := decodeJustification[Hash, N, Hasher](encoded)
+	justification, err := DecodeJustification[Hash, N, Hasher](encoded)
 	if err != nil {
 		return GrandpaJustification[Hash, N]{}, fmt.Errorf("error decoding justification for header: %s", err)
 	}

internal/client/consensus/grandpa/justification_test.go

@@ -76,7 +76,7 @@ func TestJustificationEncoding(t *testing.T) {
 	encodedJustification, err := scale.Marshal(expected)
 	require.NoError(t, err)
 
-	justification, err := decodeJustification[hash.H256, uint64, runtime.BlakeTwo256](encodedJustification)
+	justification, err := DecodeJustification[hash.H256, uint64, runtime.BlakeTwo256](encodedJustification)
 	require.NoError(t, err)
 	require.Equal(t, expected, justification.Justification)
 }