varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-09 14:40:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix:

Browse Source 
prevent selecting peers from topic not in peer topics
This commit is contained in:
NIC619
2019-07-18 19:39:01 +08:00
parent b5c3420c16
commit 404dc67e83
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
libp2p/pubsub/gossipsub.py
View File

@ -191,15 +191,16 @@ class GossipSub(IPubsubRouter):
# There are less than D peers (let this number be x) # There are less than D peers (let this number be x)
# in the fanout for a topic (or the topic is not in the fanout). # in the fanout for a topic (or the topic is not in the fanout).
# Selects the remaining number of peers (D-x) from peers.gossipsub[topic]. # Selects the remaining number of peers (D-x) from peers.gossipsub[topic].
gossipsub_peers_in_topic = [peer for peer in self.pubsub.peer_topics[topic] if topic in self.pubsub.peer_topics:
if peer in self.peers_gossipsub] gossipsub_peers_in_topic = [peer for peer in self.pubsub.peer_topics[topic]
selected_peers = \ if peer in self.peers_gossipsub]
GossipSub.select_from_minus(self.degree - fanout_size, selected_peers = \
gossipsub_peers_in_topic, GossipSub.select_from_minus(self.degree - fanout_size,
fanout_peers) gossipsub_peers_in_topic,
fanout_peers)
# Combine fanout peers with selected peers # Combine fanout peers with selected peers
fanout_peers += selected_peers fanout_peers += selected_peers
# Add fanout peers to mesh and notifies them with a GRAFT(topic) control message. # Add fanout peers to mesh and notifies them with a GRAFT(topic) control message.
for peer in fanout_peers: for peer in fanout_peers: