Merge pull request ipfs#143 from libp2p/fix/issue-142

Fix panic in FindPeersConnectedToPeer

.gx/lastpubver

@@ -1 +1 @@
-4.0.2: QmT9TxakNKCHg3uBcLnNzBSBhhACvqH8tRzJvYZjUevrvE
+4.0.3: QmSAaZBCtay8ssHG7SzBQgmrMnxZUTR4e1cZJqqiQFveZ5

package.json

@@ -166,6 +166,5 @@
   "license": "MIT",
   "name": "go-libp2p-kad-dht",
   "releaseCmd": "git commit -a -m \"gx publish $VERSION\"",
-  "version": "4.0.2"
+  "version": "4.0.3"
 }
-

routing.go

@@ -504,6 +504,7 @@ func (dht *IpfsDHT) FindPeersConnectedToPeer(ctx context.Context, id peer.ID) (<
 
 	peerchan := make(chan *pstore.PeerInfo, asyncQueryBuffer)
 	peersSeen := make(map[peer.ID]struct{})
+	var peersSeenMx sync.Mutex
 
 	peers := dht.routingTable.NearestPeers(kb.ConvertPeerID(id), AlphaValue)
 	if len(peers) == 0 {
@@ -524,10 +525,13 @@ func (dht *IpfsDHT) FindPeersConnectedToPeer(ctx context.Context, id peer.ID) (<
 			pi := pb.PBPeerToPeerInfo(pbp)
 
 			// skip peers already seen
+			peersSeenMx.Lock()
 			if _, found := peersSeen[pi.ID]; found {
+				peersSeenMx.Unlock()
 				continue
 			}
 			peersSeen[pi.ID] = struct{}{}
+			peersSeenMx.Unlock()
 
 			// if peer is connected, send it to our client.
 			if pb.Connectedness(*pbp.Connection) == inet.Connected {