swarm/network: protect addrCountC in Kademlia.AddrCountC function

#1198 (comment)
ethersphere · Feb 11, 2019 · e411147 · e411147
1 parent 11d9441
commit e411147
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/swarm/network/kademlia.go b/swarm/network/kademlia.go
@@ -353,6 +353,9 @@ func (k *Kademlia) sendNeighbourhoodDepthChange() {
 // Not receiving from the returned channel will block Register function
 // when address count value changes.
 func (k *Kademlia) AddrCountC() <-chan int {
+	k.lock.Lock()
+	defer k.lock.Unlock()
+
 	if k.addrCountC == nil {
 		k.addrCountC = make(chan int)
 	}