kev/Onionr
1
0
Fork 0
Code Issues 24 Pull Requests Projects Releases 2 Wiki Activity

Only 1 command per peer connection should be handled

Browse Source
This commit is contained in:
Kevin F 2022-02-27 13:36:39 -06:00
parent 1a7ce7d386
commit 4f3da58a60
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/gossip/server.py
View File

@ -39,7 +39,7 @@ def gossip_server(
try: try:
cmd = await asyncio.wait_for(reader.read(1), 60) cmd = await asyncio.wait_for(reader.read(1), 60)
except asyncio.exceptions.CancelledError: except asyncio.exceptions.CancelledError:
writer.close() break
cmd = int.from_bytes(cmd, 'big') cmd = int.from_bytes(cmd, 'big')
if cmd == b'' or cmd == 0: if cmd == b'' or cmd == 0:
@ -48,7 +48,7 @@ def gossip_server(
case GossipCommands.PING: case GossipCommands.PING:
writer.write(b'PONG') writer.write(b'PONG')
case GossipCommands.CLOSE: case GossipCommands.CLOSE:
writer.close() pass
case GossipCommands.ANNOUNCE: case GossipCommands.ANNOUNCE:
async def _read_announce(): async def _read_announce():
address = await reader.read(56) address = await reader.read(56)
@ -60,8 +60,10 @@ def gossip_server(
threaded=True) threaded=True)
writer.write(int(1).to_bytes(1, 'big')) writer.write(int(1).to_bytes(1, 'big'))
await asyncio.wait_for(_read_announce(), 10) await asyncio.wait_for(_read_announce(), 10)
break
await writer.drain() await writer.drain()
writer.close()
async def main(): async def main():