ash/WebStudio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Don't try to SEL a nonexistent session

Browse source
This commit is contained in:
Marks Polakovs 2020-04-12 19:48:05 +02:00
parent a073d9a5d4
commit 555c7e90f4
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
server.py
View file

@ -371,15 +371,18 @@ async def telnet_server(
else:
writer.write("WONT\r\n".encode("utf-8"))
else:
session = active_sessions[sid]
if session is None:
writer.write("FAIL\r\n".encode("utf-8"))
if sid not in active_sessions:
writer.write("WONT\r\n".encode("utf-8"))
else:
if live_session is not None:
await live_session.end()
asyncio.ensure_future(session.activate())
live_session = session
writer.write("OKAY\r\n".encode("utf-8"))
session = active_sessions[sid]
if session is None:
writer.write("FAIL\r\n".encode("utf-8"))
else:
if live_session is not None:
await live_session.end()
asyncio.ensure_future(session.activate())
live_session = session
writer.write("OKAY\r\n".encode("utf-8"))
else:
writer.write("WHAT\r\n".encode("utf-8"))
await writer.drain()