From 2c79fedfd066c2f585e2c11884fbaf3c842d52d8 Mon Sep 17 00:00:00 2001
From: petricadaipegsp <155911522+petricadaipegsp@users.noreply.github.com>
Date: Mon, 25 Nov 2024 00:03:57 +0100
Subject: [PATCH] blossomsub: Avoid looping back unwanted messages (#384)

---
 go-libp2p-blossomsub/blossomsub.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/go-libp2p-blossomsub/blossomsub.go b/go-libp2p-blossomsub/blossomsub.go
index 2ad0770..60b3058 100644
--- a/go-libp2p-blossomsub/blossomsub.go
+++ b/go-libp2p-blossomsub/blossomsub.go
@@ -912,6 +912,13 @@ func (bs *BlossomSubRouter) handleIWant(p peer.ID, ctl *pb.ControlMessage) []*pb
 
 	msgs := make([]*pb.Message, 0, len(ihave))
 	for _, msg := range ihave {
+		if peer.ID(msg.GetFrom()) == p {
+			continue
+		}
+		mid := bs.p.idGen.RawID(msg)
+		if _, ok := bs.unwanted[p][string(mid)]; ok {
+			continue
+		}
 		msgs = append(msgs, msg)
 	}