Minor fixes

This commit is contained in:
Maxim Khomutov 2023-07-08 18:46:05 +03:00
parent abd4fb60c1
commit f7defaed0c
2 changed files with 6 additions and 6 deletions

View File

@ -67,9 +67,9 @@ class Client:
await self.writer.drain() await self.writer.drain()
async def recv(self): async def recv(self):
if not self.is_disconnected(): # if not self.is_disconnected():
self.log.debug(f"Client with {self.nick}({self.cid}) disconnected") # self.log.debug(f"Client with {self.nick}({self.cid}) disconnected")
return b"" # return b""
header = await self.reader.read(4) # header: 4 bytes header = await self.reader.read(4) # header: 4 bytes
int_header = 0 int_header = 0

View File

@ -27,10 +27,10 @@ class TCPServer:
self.log.debug(f"recv1 data: {data}") self.log.debug(f"recv1 data: {data}")
if len(data) > 50: if len(data) > 50:
await client.kick("Too long data") await client.kick("Too long data")
return return False, None
if "VC2.0" not in data.decode("utf-8"): if "VC2.0" not in data.decode("utf-8"):
await client.kick("Outdated Version.") await client.kick("Outdated Version.")
return return False, None
else: else:
await client.tcp_send(b"A") # Accepted client version await client.tcp_send(b"A") # Accepted client version
@ -38,7 +38,7 @@ class TCPServer:
self.log.debug(f"recv2 data: {data}") self.log.debug(f"recv2 data: {data}")
if len(data) > 50: if len(data) > 50:
await client.kick("Invalid Key (too long)!") await client.kick("Invalid Key (too long)!")
return return False, None
client.key = data.decode("utf-8") client.key = data.decode("utf-8")
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
url = 'https://auth.beammp.com/pkToUser' url = 'https://auth.beammp.com/pkToUser'