Onionr/static-data/default-plugins/torgossip/clientfuncs/fanout.py

64 lines
1.8 KiB
Python
Raw Permalink Normal View History

2021-02-20 23:45:24 +00:00
"""Onionr - Private P2P Communication.
Fanout blocks to Onionr TorGossip peer
"""
import os
import sys
from typing import TYPE_CHECKING
from random import SystemRandom
2021-02-21 02:29:00 +00:00
from time import sleep
2021-02-20 23:45:24 +00:00
if TYPE_CHECKING:
from socket import socket
2021-02-21 02:29:00 +00:00
import logger
2021-02-20 23:45:24 +00:00
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
from commands import GossipCommands
2021-02-21 02:29:00 +00:00
from .commandsender import command_sender
2021-02-20 23:45:24 +00:00
"""
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
def fanout_to_peers(
socket_pool: dict,
block_hash: bytes,
block_data: 'KastenPacked',
fanout_count: int = 4):
2021-02-21 02:29:00 +00:00
2021-02-20 23:45:24 +00:00
peers = list(socket_pool)
SystemRandom().shuffle(peers)
2021-02-21 02:29:00 +00:00
sent_counter = 0
fanout_count = min(fanout_count, len(peers))
fanout_count = max(1, fanout_count)
while sent_counter < fanout_count:
try:
peer = peers.pop()
except IndexError:
sleep(2)
command_sender(peer, GossipCommands.CHECK_HAS_BLOCK, block_hash)
if peer.recv(1) == b'\x01':
continue
command_sender(peer, GossipCommands.PUT_BLOCK, block_hash, block_data)
if peer.recv(1) != b'\x01':
logger.warn(f"Failed to fanout {block_hash} to {peer}", terminal=True)
continue
sent_counter += 1
2021-02-20 23:45:24 +00:00