fix(node): PubSub should create msgid according to the msg hash

celestiaorg · Nov 23, 2021 · 85f445e · 85f445e
1 parent 88f97ae
commit 85f445e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/go.mod b/go.mod
@@ -34,6 +34,7 @@ require (
 	github.com/libp2p/go-libp2p-pubsub v0.5.4
 	github.com/libp2p/go-libp2p-record v0.1.3
 	github.com/libp2p/go-libp2p-routing-helpers v0.2.3
+	github.com/minio/blake2b-simd v0.0.0-20160723061019-3f5f724cb5b1
 	github.com/mitchellh/go-homedir v1.1.0
 	github.com/multiformats/go-base32 v0.0.4
 	github.com/multiformats/go-multiaddr v0.4.0

diff --git a/node/p2p/pubsub.go b/node/p2p/pubsub.go
@@ -5,6 +5,8 @@ import (
 
 	"github.com/libp2p/go-libp2p-core/host"
 	pubsub "github.com/libp2p/go-libp2p-pubsub"
+	pubsub_pb "github.com/libp2p/go-libp2p-pubsub/pb"
+	"github.com/minio/blake2b-simd"
 	"go.uber.org/fx"
 
 	"github.com/celestiaorg/celestia-node/node/fxutil"
@@ -29,6 +31,7 @@ func PubSub(cfg Config) func(pubSubParams) (*pubsub.PubSub, error) {
 		opts := []pubsub.Option{
 			pubsub.WithPeerExchange(cfg.PeerExchange || cfg.Bootstrapper),
 			pubsub.WithDirectPeers(fpeers),
+			pubsub.WithMessageIdFn(hashMsgID),
 		}
 
 		return pubsub.NewGossipSub(
@@ -39,6 +42,11 @@ func PubSub(cfg Config) func(pubSubParams) (*pubsub.PubSub, error) {
 	}
 }
 
+func hashMsgID(m *pubsub_pb.Message) string {
+	hash := blake2b.Sum256(m.Data)
+	return string(hash[:])
+}
+
 type pubSubParams struct {
 	fx.In