SantaSpeen/KuiToi-Server
1
0
Fork 0
mirror of https://github.com/kuitoi/kuitoi-Server.git synced 2025-08-17 16:25:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor kick and tcp_send in Client

Browse Source
This commit is contained in:
Maxim Khomutov 2023-07-08 02:18:04 +03:00
parent 3a40dc34cd
commit 482615af22
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/core/core.py
View File

@ -42,26 +42,28 @@ class Client:
self.alive = False
return True
def kick(self, reason):
async def kick(self, reason):
self.log.info(f"Client: \"IP: {self.addr!r}; ID: {self.cid}\" - kicked with reason: \"{reason}\"")
self.tcp_send(b"K" + bytes(reason, "utf-8"))
self.socket.close()
await self.tcp_send(b"K" + bytes(reason, "utf-8"))
# self.writer.close()
# await self.writer.wait_closed()
self.alive = False
def tcp_send(self, data):
async def tcp_send(self, data):
# TNetwork.cpp; Line: 383
# BEAMP TCP protocol sends a header of 4 bytes, followed by the data.
# [][][][][][]...[]
# ^------^^---...-^
# size data
# size data
data = data.replace(b" ", b"_")
self.log.debug(f"tcp_send({data})")
if len(data) == 10:
data += b"."
header = len(data).to_bytes(4, "little")
self.log.debug(f'len(data) {len(data)}; header: {header}; send {header + data}')
self.socket.send(header + data)
header = len(data).to_bytes(4, "little", signed=True)
self.log.debug(f'len(data) {len(data)}; send {header + data}')
self.writer.write(header + data)
await self.writer.drain()
class Core: