fix(share/peer-manager): fix potential nil dereference in peer manager (

celestiaorg#4117)
renaynay · Feb 13, 2025 · 75dca6f · 75dca6f
1 parent 70d6351
commit 75dca6f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/share/shwap/p2p/shrex/peers/manager.go b/share/shwap/p2p/shrex/peers/manager.go
@@ -286,7 +286,12 @@ func (m *Manager) doneFunc(datahash share.DataHash, peerID peer.ID, source peerS
 				m.nodes.putOnCooldown(peerID)
 				return
 			}
-			m.getPool(datahash.String()).putOnCooldown(peerID)
+			p := m.getPool(datahash.String())
+			if p == nil {
+				// pool was removed
+				return
+			}
+			p.putOnCooldown(peerID)
 		case ResultBlacklistPeer:
 			m.blacklistPeers(reasonMisbehave, peerID)
 		}