56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from asyncio import sleep
|
|
from threading import Thread
|
|
from queue import Empty
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from logger import log as logging
|
|
from ...constants import BLOCK_ID_SIZE, BLOCK_MAX_SIZE, BLOCK_SIZE_LEN
|
|
|
|
if TYPE_CHECKING:
|
|
from queue import Queue
|
|
import socket
|
|
from ...dandelion import DandelionPhase
|
|
|
|
from onionrblocks import Block
|
|
|
|
|
|
async def do_stem_stream(
|
|
peer_socket: 'socket.socket',
|
|
block_queue: "Queue[Block]",
|
|
d_phase: 'DandelionPhase'):
|
|
remaining_time = d_phase.remaining_time()
|
|
my_phase_id = d_phase.phase_id
|
|
|
|
|
|
while remaining_time > 1 and my_phase_id == d_phase.phase_id:
|
|
# Primary client component that communicate's with gossip.server.acceptstem
|
|
remaining_time = d_phase.remaining_time()
|
|
while remaining_time:
|
|
try:
|
|
# queues can't block because we're in async
|
|
bl = block_queue.get(block=False)
|
|
assert hasattr(bl, 'raw')
|
|
except Empty:
|
|
remaining_time = d_phase.remaining_time()
|
|
await sleep(1)
|
|
else:
|
|
break
|
|
if not remaining_time:
|
|
break
|
|
logging.info("Sending block over dandelion++")
|
|
|
|
block_size = str(len(bl.raw)).zfill(BLOCK_SIZE_LEN)
|
|
def _send_it():
|
|
try:
|
|
with peer_socket:
|
|
try:
|
|
peer_socket.sendall(bl.id.zfill(BLOCK_ID_SIZE).encode('utf-8'))
|
|
except AttributeError:
|
|
peer_socket.sendall(bl.id.zfill(BLOCK_ID_SIZE))
|
|
peer_socket.sendall(block_size.encode('utf-8'))
|
|
peer_socket.sendall(bl.raw)
|
|
except OSError:
|
|
pass
|
|
Thread(target=_send_it, daemon=True, name="stemout block").start()
|