mirror of
https://github.com/kuitoi/kuitoi-Server.git
synced 2025-08-17 08:15:42 +00:00
103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
# Developed by KuiToi Dev
|
|
# File core.udp_server.py
|
|
# Written by: SantaSpeen
|
|
# Core version: 0.4.0
|
|
# Licence: FPA
|
|
# (c) kuitoi.su 2023
|
|
import asyncio
|
|
|
|
from core import utils
|
|
|
|
|
|
class UDPServer(asyncio.DatagramTransport):
|
|
transport = None
|
|
|
|
def __init__(self, core, host=None, port=None):
|
|
super().__init__()
|
|
self.log = utils.get_logger("UDPServer")
|
|
self.loop = asyncio.get_event_loop()
|
|
self.Core = core
|
|
self.host = host
|
|
self.port = port
|
|
self.run = False
|
|
# self.transport = transport
|
|
|
|
def connection_made(self, transport):
|
|
self.log.debug("set connection transport")
|
|
# self.transport = self.transport()
|
|
|
|
async def handle_datagram(self, data, addr):
|
|
try:
|
|
cid = data[0] - 1
|
|
code = data[2:3].decode()
|
|
|
|
client = self.Core.get_client(cid=cid)
|
|
if client:
|
|
if client._udp_sock != (self.transport, addr):
|
|
client._udp_sock = (self.transport, addr)
|
|
self.log.debug(f"Set UDP Sock for CID: {cid}")
|
|
match code:
|
|
case "p":
|
|
self.log.debug(f"[{cid}] Send ping")
|
|
# TODO: Call event onSentPing
|
|
self.transport.sendto(b"p", addr) # Send ping
|
|
case "Z":
|
|
# TODO: Call event onChangePosition
|
|
if client:
|
|
await client._send(data[2:], to_all=True, to_self=False, to_udp=True)
|
|
case _:
|
|
self.log.debug(f"[{cid}] Unknown code: {code}")
|
|
else:
|
|
self.log.debug(f"Client not found.")
|
|
|
|
except Exception as e:
|
|
self.log.error(f"Error handle_datagram: {e}")
|
|
|
|
def datagram_received(self, *args, **kwargs):
|
|
self.loop.create_task(self.handle_datagram(*args, **kwargs))
|
|
|
|
def connection_lost(self, exc):
|
|
if exc is not None and exc != KeyboardInterrupt:
|
|
self.log.debug(f'Connection raised: {exc}')
|
|
self.log.debug(f'Disconnected.')
|
|
|
|
def error_received(self, exc):
|
|
self.log.debug(f'error_received: {exc}')
|
|
self.log.exception(exc)
|
|
self.connection_lost(exc)
|
|
self.transport.close()
|
|
|
|
async def _start(self):
|
|
self.log.debug("Starting UDP server.")
|
|
try:
|
|
while self.Core.run:
|
|
await asyncio.sleep(0.2)
|
|
|
|
d = UDPServer
|
|
self.transport, p = await self.loop.create_datagram_endpoint(
|
|
lambda: d(self.Core),
|
|
local_addr=(self.host, self.port)
|
|
)
|
|
d.transport = self.transport
|
|
|
|
if not self.run:
|
|
self.log.debug(f"UDP server started on {self.transport.get_extra_info('sockname')}")
|
|
|
|
self.run = True
|
|
while not self.transport.is_closing():
|
|
await asyncio.sleep(0.2)
|
|
self.log.info("UDP сервер сдох 1")
|
|
except OSError as e:
|
|
self.log.error("Cannot bind port or other error")
|
|
self.log.exception(e)
|
|
except Exception as e:
|
|
self.log.error(f"Error: {e}")
|
|
self.log.exception(e)
|
|
finally:
|
|
self.run = False
|
|
self.Core.run = False
|
|
|
|
def _stop(self):
|
|
self.log.debug("Stopping UDP server")
|
|
self.transport.close()
|