mirror of
https://github.com/kuitoi/kuitoi-Server.git
synced 2025-08-17 16:25:36 +00:00
Packets handled (Recursive finding second packet)
This commit is contained in:
parent
acdb32d900
commit
642c91d59c
@ -17,9 +17,11 @@ class Client:
|
|||||||
def __init__(self, reader, writer, core):
|
def __init__(self, reader, writer, core):
|
||||||
self.__reader = reader
|
self.__reader = reader
|
||||||
self.__writer = writer
|
self.__writer = writer
|
||||||
self._down_rw = (None, None)
|
|
||||||
self.__Core = core
|
self.__Core = core
|
||||||
self.__alive = True
|
self.__alive = True
|
||||||
|
self.__packets_queue = []
|
||||||
|
self.__tasks = []
|
||||||
|
self._down_rw = (None, None)
|
||||||
self._loop = asyncio.get_event_loop()
|
self._loop = asyncio.get_event_loop()
|
||||||
self._log = utils.get_logger("client(None:0)")
|
self._log = utils.get_logger("client(None:0)")
|
||||||
self._addr = writer.get_extra_info("sockname")
|
self._addr = writer.get_extra_info("sockname")
|
||||||
@ -128,46 +130,66 @@ class Client:
|
|||||||
writer.write(header + data)
|
writer.write(header + data)
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
except ConnectionError:
|
except ConnectionError:
|
||||||
self.log.debug('tcp_send: Disconnected')
|
self.log.debug('_send: Disconnected')
|
||||||
self.__alive = False
|
self.__alive = False
|
||||||
|
await self._remove_me()
|
||||||
|
|
||||||
async def _recv(self):
|
async def __handle_packet(self, data, int_header):
|
||||||
try:
|
self.log.debug(f"int_header: {int_header}; data: {data};")
|
||||||
header = await self.__reader.read(4)
|
if len(data) != int_header:
|
||||||
|
self.log.debug(f"WARN Expected to read {int_header} bytes, instead got {len(data)}")
|
||||||
|
|
||||||
int_header = 0
|
recv2 = data[int_header:]
|
||||||
for i in range(len(header)):
|
header2 = recv2[:4]
|
||||||
int_header += header[i]
|
data2 = recv2[4:]
|
||||||
|
int_header2 = int.from_bytes(header2, byteorder='little', signed=True)
|
||||||
|
self.log.debug(f"header2: {header2}; int_header2: {int_header2}; data2: {data2}")
|
||||||
|
t = asyncio.create_task(self.__handle_packet(data2, int_header2))
|
||||||
|
self.__tasks.append(t)
|
||||||
|
data = data[:4 + int_header]
|
||||||
|
|
||||||
if int_header <= 0:
|
abg = b"ABG:"
|
||||||
await asyncio.sleep(0.1)
|
if len(data) > len(abg) and data.startswith(abg):
|
||||||
self.is_disconnected()
|
data = zlib.decompress(data[len(abg):])
|
||||||
if self.__alive:
|
self.log.debug(f"ABG Packet: {len(data)}")
|
||||||
self.log.debug(f"Header: {header}")
|
|
||||||
await self.kick("Invalid packet - header negative")
|
|
||||||
return b""
|
|
||||||
|
|
||||||
if int_header > 100 * MB:
|
self.__packets_queue.append(data)
|
||||||
await self.kick("Header size limit exceeded")
|
self.log.debug(f"Packets in queue: {len(self.__packets_queue)}")
|
||||||
self.log.warning(f"Client {self.nick}:{self.cid} sent header of >100MB - "
|
|
||||||
f"assuming malicious intent and disconnecting the client.")
|
|
||||||
return b""
|
|
||||||
|
|
||||||
data = await self.__reader.read(100 * MB)
|
async def _recv(self, one=False):
|
||||||
self.log.debug(f"header: `{header}`; int_header: `{int_header}`; data: `{data}`;")
|
while self.__alive:
|
||||||
|
try:
|
||||||
|
header = await self.__reader.read(4)
|
||||||
|
|
||||||
if len(data) != int_header:
|
int_header = int.from_bytes(header, byteorder='little', signed=True)
|
||||||
self.log.debug(f"WARN Expected to read {int_header} bytes, instead got {len(data)}")
|
|
||||||
|
|
||||||
abg = b"ABG:"
|
if int_header <= 0:
|
||||||
if len(data) > len(abg) and data.startswith(abg):
|
await asyncio.sleep(0.1)
|
||||||
data = zlib.decompress(data[len(abg):])
|
self.is_disconnected()
|
||||||
self.log.debug(f"ABG Packet: {len(data)}")
|
if self.__alive:
|
||||||
return data
|
self.log.debug(f"Header: {header}")
|
||||||
return data
|
await self.kick("Invalid packet - header negative")
|
||||||
except ConnectionError:
|
self.__packets_queue.append(None)
|
||||||
self.__alive = False
|
continue
|
||||||
return b""
|
|
||||||
|
if int_header > 100 * MB:
|
||||||
|
await self.kick("Header size limit exceeded")
|
||||||
|
self.log.warning(f"Client {self.nick}:{self.cid} sent header of >100MB - "
|
||||||
|
f"assuming malicious intent and disconnecting the client.")
|
||||||
|
self.__packets_queue.append(None)
|
||||||
|
continue
|
||||||
|
|
||||||
|
data = await self.__reader.read(100 * MB)
|
||||||
|
if one:
|
||||||
|
self.log.debug(f"int_header: `{int_header}`; data: `{data}`;")
|
||||||
|
return data
|
||||||
|
else:
|
||||||
|
t = asyncio.create_task(self.__handle_packet(data, int_header))
|
||||||
|
self.__tasks.append(t)
|
||||||
|
|
||||||
|
except ConnectionError:
|
||||||
|
self.__alive = False
|
||||||
|
self.__packets_queue.append(None)
|
||||||
|
|
||||||
async def _split_load(self, start, end, d_sock, filename):
|
async def _split_load(self, start, end, d_sock, filename):
|
||||||
# TODO: Speed limiter
|
# TODO: Speed limiter
|
||||||
@ -191,8 +213,7 @@ class Client:
|
|||||||
|
|
||||||
async def _sync_resources(self):
|
async def _sync_resources(self):
|
||||||
while self.__alive:
|
while self.__alive:
|
||||||
data = await self._recv()
|
data = await self._recv(True)
|
||||||
self.log.debug(f"data: {data!r}")
|
|
||||||
if data.startswith(b"f"):
|
if data.startswith(b"f"):
|
||||||
file = data[1:].decode("utf-8")
|
file = data[1:].decode("utf-8")
|
||||||
# TODO: i18n
|
# TODO: i18n
|
||||||
@ -249,14 +270,14 @@ class Client:
|
|||||||
else:
|
else:
|
||||||
await self._send(bytes(mod_list, "utf-8"))
|
await self._send(bytes(mod_list, "utf-8"))
|
||||||
elif data == b"Done":
|
elif data == b"Done":
|
||||||
await self._send(b"M/levels/" + bytes(config.Game['map'], 'utf-8') + b"/info.json")
|
|
||||||
for c in range(config.Game['max_cars']):
|
for c in range(config.Game['max_cars']):
|
||||||
self._cars.append(None)
|
self._cars.append(None)
|
||||||
|
await self._send(b"M/levels/" + bytes(config.Game['map'], 'utf-8') + b"/info.json")
|
||||||
break
|
break
|
||||||
return
|
return
|
||||||
|
|
||||||
def _get_cid_vid(self, data: str):
|
def _get_cid_vid(self, data: str):
|
||||||
s = data[data.find(":", 1)+1:]
|
s = data[data.find(":", 1) + 1:]
|
||||||
id_sep = s.find('-')
|
id_sep = s.find('-')
|
||||||
if id_sep == -1:
|
if id_sep == -1:
|
||||||
self.log.debug(f"Invalid packet: Could not parse pid/vid from packet, as there is no '-' separator: '{s}'")
|
self.log.debug(f"Invalid packet: Could not parse pid/vid from packet, as there is no '-' separator: '{s}'")
|
||||||
@ -352,8 +373,10 @@ class Client:
|
|||||||
# Codes: V W X Y
|
# Codes: V W X Y
|
||||||
if 89 >= data[0] >= 86:
|
if 89 >= data[0] >= 86:
|
||||||
await self._send(data, to_all=True, to_self=False)
|
await self._send(data, to_all=True, to_self=False)
|
||||||
|
try:
|
||||||
data = data.decode()
|
data = data.decode()
|
||||||
|
except UnicodeDecodeError:
|
||||||
|
self.log.debug(f"UnicodeDecodeError: {data}")
|
||||||
|
|
||||||
code = data[0]
|
code = data[0]
|
||||||
match code:
|
match code:
|
||||||
@ -370,6 +393,8 @@ class Client:
|
|||||||
if not client:
|
if not client:
|
||||||
continue
|
continue
|
||||||
for car in client.cars:
|
for car in client.cars:
|
||||||
|
if not car:
|
||||||
|
continue
|
||||||
await self._send(car['packet'])
|
await self._send(car['packet'])
|
||||||
|
|
||||||
case "C": # Chat handler
|
case "C": # Chat handler
|
||||||
@ -420,9 +445,19 @@ class Client:
|
|||||||
async def _looper(self):
|
async def _looper(self):
|
||||||
await self._send(f"P{self.cid}") # Send clientID
|
await self._send(f"P{self.cid}") # Send clientID
|
||||||
await self._sync_resources()
|
await self._sync_resources()
|
||||||
|
tasks = self.__tasks
|
||||||
|
recv = asyncio.create_task(self._recv())
|
||||||
|
tasks.append(recv)
|
||||||
while self.__alive:
|
while self.__alive:
|
||||||
data = await self._recv()
|
if len(self.__packets_queue) > 0:
|
||||||
self._loop.create_task(self._handle_codes(data))
|
for index, packet in enumerate(self.__packets_queue):
|
||||||
|
del self.__packets_queue[index]
|
||||||
|
self.log.debug(f"Packet: {packet}")
|
||||||
|
task = self._loop.create_task(self._handle_codes(packet))
|
||||||
|
tasks.append(task)
|
||||||
|
else:
|
||||||
|
await asyncio.sleep(0.1)
|
||||||
|
await asyncio.gather(*tasks)
|
||||||
|
|
||||||
async def _remove_me(self):
|
async def _remove_me(self):
|
||||||
await asyncio.sleep(0.3)
|
await asyncio.sleep(0.3)
|
||||||
|
@ -15,8 +15,10 @@ from core import Core, utils
|
|||||||
class Client:
|
class Client:
|
||||||
|
|
||||||
def __init__(self, reader: StreamReader, writer: StreamWriter, core: Core) -> "Client":
|
def __init__(self, reader: StreamReader, writer: StreamWriter, core: Core) -> "Client":
|
||||||
|
self.__tasks = []
|
||||||
self.__reader = reader
|
self.__reader = reader
|
||||||
self.__writer = writer
|
self.__writer = writer
|
||||||
|
self.__packets_queue = []
|
||||||
self._down_rw: Tuple[StreamReader, StreamWriter] | Tuple[None, None] = (None, None)
|
self._down_rw: Tuple[StreamReader, StreamWriter] | Tuple[None, None] = (None, None)
|
||||||
self._log = utils.get_logger("client(id: )")
|
self._log = utils.get_logger("client(id: )")
|
||||||
self._addr = writer.get_extra_info("sockname")
|
self._addr = writer.get_extra_info("sockname")
|
||||||
@ -50,7 +52,8 @@ class Client:
|
|||||||
async def kick(self, reason: str) -> None: ...
|
async def kick(self, reason: str) -> None: ...
|
||||||
async def _send(self, data: bytes | str, to_all: bool = False, to_self: bool = True, to_udp: bool = False, writer: StreamWriter = None) -> None: ...
|
async def _send(self, data: bytes | str, to_all: bool = False, to_self: bool = True, to_udp: bool = False, writer: StreamWriter = None) -> None: ...
|
||||||
async def _sync_resources(self) -> None: ...
|
async def _sync_resources(self) -> None: ...
|
||||||
async def _recv(self) -> bytes: ...
|
async def __handle_packet(self, data, int_header): ...
|
||||||
|
async def _recv(self, one=False) -> bytes | None: ...
|
||||||
async def _split_load(self, start: int, end: int, d_sock: bool, filename: str) -> None: ...
|
async def _split_load(self, start: int, end: int, d_sock: bool, filename: str) -> None: ...
|
||||||
async def _get_cid_vid(self, s: str) -> Tuple[int, int]: ...
|
async def _get_cid_vid(self, s: str) -> Tuple[int, int]: ...
|
||||||
async def _handle_vehicle_codes(self, data) -> None: ...
|
async def _handle_vehicle_codes(self, data) -> None: ...
|
||||||
|
@ -25,7 +25,7 @@ class TCPServer:
|
|||||||
client = self.Core.create_client(reader, writer)
|
client = self.Core.create_client(reader, writer)
|
||||||
# TODO: i18n
|
# TODO: i18n
|
||||||
self.log.info(f"Identifying new ClientConnection...")
|
self.log.info(f"Identifying new ClientConnection...")
|
||||||
data = await client._recv()
|
data = await client._recv(True)
|
||||||
self.log.debug(f"Version: {data}")
|
self.log.debug(f"Version: {data}")
|
||||||
if data.decode("utf-8") != f"VC{self.Core.client_major_version}":
|
if data.decode("utf-8") != f"VC{self.Core.client_major_version}":
|
||||||
# TODO: i18n
|
# TODO: i18n
|
||||||
@ -34,7 +34,7 @@ class TCPServer:
|
|||||||
else:
|
else:
|
||||||
await client._send(b"S") # Accepted client version
|
await client._send(b"S") # Accepted client version
|
||||||
|
|
||||||
data = await client._recv()
|
data = await client._recv(True)
|
||||||
self.log.debug(f"Key: {data}")
|
self.log.debug(f"Key: {data}")
|
||||||
if len(data) > 50:
|
if len(data) > 50:
|
||||||
# TODO: i18n
|
# TODO: i18n
|
||||||
@ -133,7 +133,7 @@ class TCPServer:
|
|||||||
break
|
break
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
# TODO: i18n
|
# TODO: i18n
|
||||||
self.log.error("Error while connecting..")
|
self.log.error("Error while handling connection...")
|
||||||
self.log.exception(e)
|
self.log.exception(e)
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
break
|
break
|
||||||
|
Loading…
x
Reference in New Issue
Block a user