mirror of
https://github.com/kuitoi/kuitoi-Server.git
synced 2025-08-17 08:15:42 +00:00
Compare commits
7 Commits
dcafef918a
...
dd2c461581
Author | SHA1 | Date | |
---|---|---|---|
dd2c461581 | |||
a8c153691c | |||
52893513d0 | |||
1f595db700 | |||
565750e784 | |||
13321fb9b5 | |||
cdf226ac5c |
@ -11,17 +11,21 @@ BeamingDrive Multiplayer (BeamMP) server compatible with BeamMP clients.
|
||||
- [x] Private access (Without key, Direct connect)
|
||||
- [x] Public access (With key, listing in Launcher)
|
||||
- [X] Player authentication
|
||||
- [ ] KuiToi System
|
||||
- [ ] Servers counter
|
||||
- [ ] Players counter
|
||||
- [ ] Etc.
|
||||
- [ ] TCP Server part:
|
||||
- [x] Handle code
|
||||
- [x] Understanding BeamMP header
|
||||
- [ ] Upload mods
|
||||
- [x] Connecting to the world
|
||||
- [x] Chat
|
||||
- [ ] Player counter _(Code: Ss)_
|
||||
- [x] Players online counter
|
||||
- [ ] Car state synchronizations _(Codes: We, Vi)_
|
||||
- [ ] "ABG:" (compressed data)
|
||||
- [x] Decompress data
|
||||
- [ ] Vehicle data
|
||||
- [ ] Vehicle data _(Code: Os)_
|
||||
- [ ] UDP Server part:
|
||||
- [ ] Players synchronizations _(Code: Zp)_
|
||||
- [ ] Ping _(Code: p)_
|
||||
|
@ -2,7 +2,7 @@
|
||||
# File core.__init__.py
|
||||
# Written by: SantaSpeen
|
||||
# Version 1.3
|
||||
# Core version: 0.2.0
|
||||
# Core version: 0.2.1
|
||||
# Licence: FPA
|
||||
# (c) kuitoi.su 2023
|
||||
# Special thanks to: AI Sage(https://poe.com/Sage), AI falcon-40b-v7(https://OpenBuddy.ai)
|
||||
@ -10,8 +10,8 @@
|
||||
__title__ = 'KuiToi-Server'
|
||||
__description__ = 'BeamingDrive Multiplayer server compatible with BeamMP clients.'
|
||||
__url__ = 'https://github.com/kuitoi/kuitoi-Server'
|
||||
__version__ = '0.2.0'
|
||||
__build__ = 776
|
||||
__version__ = '0.2.1'
|
||||
__build__ = 874
|
||||
__author__ = 'SantaSpeen'
|
||||
__author_email__ = 'admin@kuitoi.su'
|
||||
__license__ = "FPA"
|
||||
|
171
src/core/core.py
171
src/core/core.py
@ -1,11 +1,12 @@
|
||||
# Developed by KuiToi Dev
|
||||
# File core.core.py
|
||||
# Written by: SantaSpeen
|
||||
# Version 0.2.0
|
||||
# Core version: 0.2.1
|
||||
# Licence: FPA
|
||||
# (c) kuitoi.su 2023
|
||||
import asyncio
|
||||
import os
|
||||
import random
|
||||
import zlib
|
||||
from threading import Thread
|
||||
|
||||
@ -36,8 +37,8 @@ class Client:
|
||||
self.alive = True
|
||||
|
||||
def _update_logger(self):
|
||||
self.log.debug(f"Update logger")
|
||||
self.log = utils.get_logger(f"client({self.nick}:{self.cid})")
|
||||
self.log.debug(f"Update logger")
|
||||
|
||||
def is_disconnected(self):
|
||||
if not self.alive:
|
||||
@ -55,11 +56,10 @@ class Client:
|
||||
async def kick(self, reason):
|
||||
self.log.info(f"Client: \"IP: {self.addr!r}; ID: {self.cid}\" - kicked with reason: \"{reason}\"")
|
||||
await self.tcp_send(b"K" + bytes(reason, "utf-8"))
|
||||
# self.writer.close()
|
||||
# await self.writer.wait_closed()
|
||||
self.alive = False
|
||||
await self.remove_me()
|
||||
|
||||
async def tcp_send(self, data):
|
||||
async def tcp_send(self, data, to_all=False, writer=None):
|
||||
|
||||
# TNetwork.cpp; Line: 383
|
||||
# BeamMP TCP protocol sends a header of 4 bytes, followed by the data.
|
||||
@ -67,18 +67,33 @@ class Client:
|
||||
# ^------^^---...-^
|
||||
# size data
|
||||
|
||||
if writer is None:
|
||||
writer = self.writer
|
||||
|
||||
if to_all:
|
||||
for client in self.Core.clients:
|
||||
if not client:
|
||||
continue
|
||||
await client.tcp_send(data)
|
||||
return
|
||||
|
||||
self.log.debug(f"tcp_send({data})")
|
||||
if len(data) == 10:
|
||||
data += b"."
|
||||
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()
|
||||
try:
|
||||
writer.write(header + data)
|
||||
await writer.drain()
|
||||
except ConnectionError:
|
||||
self.log.debug('Disconnected')
|
||||
self.alive = False
|
||||
|
||||
async def recv(self):
|
||||
async def recv(self, kostil=False):
|
||||
# if not self.is_disconnected():
|
||||
# self.log.debug(f"Client with {self.nick}({self.cid}) disconnected")
|
||||
# return b""
|
||||
try:
|
||||
header = await self.reader.read(4) # header: 4 bytes
|
||||
|
||||
int_header = 0
|
||||
@ -86,6 +101,14 @@ class Client:
|
||||
int_header += header[i]
|
||||
|
||||
if int_header <= 0:
|
||||
await asyncio.sleep(0.1)
|
||||
if not self.alive:
|
||||
self.log.debug(f"Disconnected")
|
||||
self.writer.close()
|
||||
return b''
|
||||
if kostil:
|
||||
return
|
||||
self.log.debug(f"Header: {header}")
|
||||
await self.kick("Invalid packet - header negative")
|
||||
return b""
|
||||
|
||||
@ -95,7 +118,7 @@ class Client:
|
||||
f"assuming malicious intent and disconnecting the client.")
|
||||
return b""
|
||||
|
||||
data = await self.reader.read(101 * MB)
|
||||
data = await self.reader.read(100 * MB)
|
||||
self.log.debug(f"header: `{header}`; int_header: `{int_header}`; data: `{data}`;")
|
||||
|
||||
if len(data) != int_header:
|
||||
@ -107,23 +130,35 @@ class Client:
|
||||
self.log.debug(f"ABG: {data}")
|
||||
return data
|
||||
return data
|
||||
except ConnectionError:
|
||||
self.alive = False
|
||||
await self.remove_me()
|
||||
return b""
|
||||
|
||||
async def sync_resources(self):
|
||||
while True:
|
||||
data = await self.recv()
|
||||
if not data:
|
||||
await asyncio.sleep(.1)
|
||||
continue
|
||||
self.log.debug(f"Received: {data}")
|
||||
if data.startswith(b"f"):
|
||||
# TODO: SendFile
|
||||
pass
|
||||
file = data[1:].decode("utf-8")
|
||||
self.log.debug(f"Sending File: {file}")
|
||||
await self.kick(f"TODO: SendFile({file})")
|
||||
elif data.startswith(b"SR"):
|
||||
# TODO: Create mods list
|
||||
self.log.debug("Sending Mod Info")
|
||||
mods = []
|
||||
mod_list = b''
|
||||
# * code *
|
||||
if len(mods) == 0:
|
||||
mod_list = '%s;%s;'
|
||||
for mod in self.Core.mods_list:
|
||||
if type(mod) == int:
|
||||
continue
|
||||
mod_list = (mod_list % (mod['path'], mod['size'])).replace(";", ";%s;")
|
||||
mod_list = mod_list.replace("%s;", "")
|
||||
if len(mod_list) == 0:
|
||||
await self.tcp_send(b"-")
|
||||
else:
|
||||
await self.tcp_send(mod_list)
|
||||
await self.tcp_send(bytes(mod_list, "utf-8"))
|
||||
data = await self.recv()
|
||||
if data == b"Done":
|
||||
await self.tcp_send(b"M/levels/" + bytes(config.Game['map'], 'utf-8') + b"/info.json")
|
||||
@ -131,7 +166,6 @@ class Client:
|
||||
|
||||
async def looper(self):
|
||||
# self.is_disconnected()
|
||||
self.log.debug(f"Alive: {self.alive}")
|
||||
await self.tcp_send(b"P" + bytes(f"{self.cid}", "utf-8"))
|
||||
await self.sync_resources()
|
||||
while self.alive:
|
||||
@ -139,19 +173,31 @@ class Client:
|
||||
if data == b"":
|
||||
if not self.alive:
|
||||
break
|
||||
elif self.is_disconnected():
|
||||
break
|
||||
else:
|
||||
await asyncio.sleep(.2)
|
||||
continue
|
||||
code = data.decode()[0]
|
||||
self.log.debug(f"Code: {code}, data: {data}")
|
||||
self.log.debug(f"Received code: {code}, data: {data}")
|
||||
match code:
|
||||
case "H":
|
||||
# Client connected
|
||||
await self.tcp_send(b"Sn" + bytes(self.nick, "utf-8"))
|
||||
await self.tcp_send(b"Sn" + bytes(self.nick, "utf-8"), to_all=True)
|
||||
case "C":
|
||||
# Chat
|
||||
await self.tcp_send(data)
|
||||
await self.tcp_send(data, to_all=True)
|
||||
|
||||
async def remove_me(self):
|
||||
self.log.debug(f"Removing client {self.nick}({self.cid})")
|
||||
await asyncio.sleep(0.3)
|
||||
if not self.writer.is_closing():
|
||||
self.writer.close()
|
||||
if (self.cid > 0 or self.nick is not None) and self.Core.clients_by_id.get(self.cid):
|
||||
_, down_w = self.down_rw
|
||||
if down_w and not down_w.is_closing():
|
||||
down_w.close()
|
||||
self.Core.clients[self.cid] = None
|
||||
self.Core.clients_by_id.pop(self.cid)
|
||||
self.Core.clients_by_nick.pop(self.nick)
|
||||
|
||||
|
||||
class Core:
|
||||
@ -177,40 +223,60 @@ class Core:
|
||||
self.client_major_version = "2.0"
|
||||
self.BeamMP_version = "3.2.0"
|
||||
|
||||
def get_client(self, sock=None, cid=None, nick=None):
|
||||
def get_client(self, cid=None, nick=None):
|
||||
if cid:
|
||||
return self.clients_by_id.get(cid)
|
||||
if nick:
|
||||
return self.clients_by_nick.get(nick)
|
||||
if sock:
|
||||
return self.clients_by_nick.get(sock.getsockname())
|
||||
|
||||
def insert_client(self, client):
|
||||
self.log.debug(f"Inserting client: {client.cid}")
|
||||
self.clients_by_nick.update({client.nick: client})
|
||||
self.clients_by_id.update({client.cid: client})
|
||||
self.clients[client.cid] = client
|
||||
|
||||
def create_client(self, *args, **kwargs):
|
||||
client = Client(*args, **kwargs)
|
||||
cid = 1
|
||||
for client in self.clients:
|
||||
if client.cid == cid:
|
||||
async def insert_client(self, client):
|
||||
await asyncio.sleep(random.randint(3, 9) * 0.01)
|
||||
cid = 0
|
||||
for _client in self.clients:
|
||||
if not _client:
|
||||
break
|
||||
if _client.cid == cid:
|
||||
cid += 1
|
||||
else:
|
||||
break
|
||||
await asyncio.sleep(random.randint(3, 9) * 0.01)
|
||||
if not self.clients[cid]:
|
||||
client.cid = cid
|
||||
self.clients_by_nick.update({client.nick: client})
|
||||
self.log.debug(f"Inserting client: id{client.cid}")
|
||||
self.clients_by_id.update({client.cid: client})
|
||||
self.clients[client.cid] = client
|
||||
client._update_logger()
|
||||
self.log.debug(f"Create client; client.cid: {client.cid};")
|
||||
return
|
||||
await self.insert_client(client)
|
||||
|
||||
def create_client(self, *args, **kwargs):
|
||||
self.log.debug(f"Create client")
|
||||
client = Client(core=self, *args, **kwargs)
|
||||
return client
|
||||
|
||||
async def check_alive(self):
|
||||
await asyncio.sleep(5)
|
||||
self.log.debug(f"Checking if clients is alive")
|
||||
def get_clients_list(self, need_cid=False):
|
||||
out = ""
|
||||
for client in self.clients:
|
||||
d = client.is_disconnected()
|
||||
if d:
|
||||
self.log.debug(f"Client ID: {client.cid} died...")
|
||||
if not client:
|
||||
continue
|
||||
out += f"{client.nick}"
|
||||
if need_cid:
|
||||
out += f":{client.cid}"
|
||||
out += ","
|
||||
if out:
|
||||
out = out[:-1]
|
||||
return out
|
||||
|
||||
async def check_alive(self):
|
||||
maxp = config.Game['players']
|
||||
while self.run:
|
||||
await asyncio.sleep(1)
|
||||
ca = f"Ss{len(self.clients_by_id)}/{maxp}:{self.get_clients_list()}"
|
||||
for client in self.clients:
|
||||
if not client:
|
||||
continue
|
||||
await client.tcp_send(bytes(ca, "utf-8"))
|
||||
|
||||
@staticmethod
|
||||
def start_web():
|
||||
@ -223,10 +289,10 @@ class Core:
|
||||
webapp.uvserver = uvserver
|
||||
uvserver.run()
|
||||
|
||||
@staticmethod
|
||||
async def stop_me():
|
||||
async def stop_me(self):
|
||||
while webapp.data_run[0]:
|
||||
await asyncio.sleep(1)
|
||||
self.run = False
|
||||
raise KeyboardInterrupt
|
||||
|
||||
# noinspection SpellCheckingInspection,PyPep8Naming
|
||||
@ -246,6 +312,7 @@ class Core:
|
||||
modstotalsize = self.mods_list[0]
|
||||
modstotal = len(self.mods_list) - 1
|
||||
while self.run:
|
||||
try:
|
||||
data = {"uuid": config.Auth["key"], "players": len(self.clients), "maxplayers": config.Game["players"],
|
||||
"port": config.Server["server_port"], "map": f"/levels/{config.Game['map']}/info.json",
|
||||
"private": config.Auth['private'], "version": self.BeamMP_version,
|
||||
@ -304,11 +371,17 @@ class Core:
|
||||
return ok
|
||||
|
||||
await asyncio.sleep(5)
|
||||
except Exception as e:
|
||||
self.log.error(f"Error in heartbeat: {e}")
|
||||
|
||||
async def main(self):
|
||||
self.run = True
|
||||
self.tcp = self.tcp(self, self.server_ip, self.server_port)
|
||||
self.udp = self.udp(self, self.server_ip, self.server_port)
|
||||
console.add_command(
|
||||
"list",
|
||||
lambda x: f"Players list: {self.get_clients_list(True)}"
|
||||
)
|
||||
try:
|
||||
# WebApi Start
|
||||
if config.WebAPI["enabled"]:
|
||||
@ -336,9 +409,11 @@ class Core:
|
||||
self.log.info(f"Loaded {lmods} mods: {round(self.mods_list[0] / MB, 2)}mb")
|
||||
|
||||
await self.heartbeat(True)
|
||||
for i in range(int(config.Game["players"] * 1.3)):
|
||||
self.clients.append(None)
|
||||
tasks = []
|
||||
# self.check_alive()
|
||||
nrtasks = [self.tcp.start, self.udp.start, console.start, self.stop_me, self.heartbeat, ]
|
||||
# self.udp.start,
|
||||
nrtasks = [self.tcp.start, console.start, self.stop_me, self.heartbeat, self.check_alive]
|
||||
for task in nrtasks:
|
||||
tasks.append(asyncio.create_task(task()))
|
||||
t = asyncio.wait(tasks, return_when=asyncio.FIRST_EXCEPTION)
|
||||
@ -354,7 +429,7 @@ class Core:
|
||||
pass
|
||||
finally:
|
||||
self.tcp.stop()
|
||||
self.udp.stop()
|
||||
# self.udp.stop()
|
||||
self.run = False
|
||||
|
||||
def start(self):
|
||||
|
@ -24,7 +24,7 @@ class Client:
|
||||
self.addr = writer.get_extra_info("sockname")
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.Core = core
|
||||
self.cid = 0
|
||||
self.cid: int = 0
|
||||
self.key: str = None
|
||||
self.nick: str = None
|
||||
self.roles: str = None
|
||||
@ -32,12 +32,12 @@ class Client:
|
||||
self.alive = True
|
||||
def is_disconnected(self) -> bool: ...
|
||||
async def kick(self, reason: str) -> None: ...
|
||||
async def tcp_send(self, data: bytes) -> None: ...
|
||||
async def tcp_send(self, data: bytes, to_all:bool = False, writer: StreamWriter = None) -> None: ...
|
||||
async def sync_resources(self) -> None: ...
|
||||
async def recv(self) -> bytes: ...
|
||||
async def looper(self) -> None: ...
|
||||
def _update_logger(self) -> None: ...
|
||||
|
||||
async def remove_me(self) -> None: ...
|
||||
|
||||
class Core:
|
||||
def __init__(self):
|
||||
@ -45,7 +45,7 @@ class Core:
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.run = False
|
||||
self.direct = False
|
||||
self.clients: List[Client]= []
|
||||
self.clients: List[Client | None]= []
|
||||
self.clients_by_id: Dict[{int: Client}]= {}
|
||||
self.clients_by_nick: Dict[{str: Client}] = {}
|
||||
self.clients_counter: int = 0
|
||||
@ -59,13 +59,13 @@ class Core:
|
||||
self.web_stop: Callable = lambda: None
|
||||
self.client_major_version = "2.0"
|
||||
self.BeamMP_version = "3.2.0"
|
||||
def insert_client(self, client: Client) -> None: ...
|
||||
async def insert_client(self, client: Client) -> None: ...
|
||||
def create_client(self, *args, **kwargs) -> Client: ...
|
||||
def get_clients_list(self, need_cid=False) -> str: ...
|
||||
async def check_alive(self) -> None: ...
|
||||
@staticmethod
|
||||
def start_web() -> None: ...
|
||||
@staticmethod
|
||||
def stop_me() -> None: ...
|
||||
def stop_me(self) -> None: ...
|
||||
async def heartbeat(self, test=False) -> None: ...
|
||||
async def main(self) -> None: ...
|
||||
def start(self) -> None: ...
|
||||
|
@ -1,11 +1,13 @@
|
||||
# Developed by KuiToi Dev
|
||||
# File core.tcp_server.py
|
||||
# Written by: SantaSpeen
|
||||
# Version 0.2.0
|
||||
# Core version: 0.2.1
|
||||
# Licence: FPA
|
||||
# (c) kuitoi.su 2023
|
||||
import asyncio
|
||||
import traceback
|
||||
from asyncio import AbstractEventLoop
|
||||
from threading import Thread
|
||||
|
||||
import aiohttp
|
||||
|
||||
@ -15,10 +17,11 @@ from core import utils
|
||||
class TCPServer:
|
||||
def __init__(self, core, host, port):
|
||||
self.log = utils.get_logger("TCPServer")
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.Core = core
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.run = False
|
||||
|
||||
async def auth_client(self, reader, writer):
|
||||
client = self.Core.create_client(reader, writer)
|
||||
@ -37,49 +40,58 @@ class TCPServer:
|
||||
await client.kick("Invalid Key (too long)!")
|
||||
return False, None
|
||||
client.key = data.decode("utf-8")
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
url = 'https://auth.beammp.com/pkToUser'
|
||||
async with session.post(url, data={'key': client.key}) as response:
|
||||
res = await response.json()
|
||||
self.log.debug(f"res: {res}")
|
||||
try:
|
||||
if res.get("error"):
|
||||
await client.kick('Invalid key! Please restart your game.')
|
||||
return
|
||||
return False, None
|
||||
client.nick = res["username"]
|
||||
client.roles = res["roles"]
|
||||
client.guest = res["guest"]
|
||||
client._update_logger()
|
||||
except Exception as e:
|
||||
self.log.error(f"Auth error: {e}")
|
||||
await client.kick('Invalid authentication data! Try to connect in 5 minutes.')
|
||||
await client.kick('Invalid authentication data! Try to reconnect in 5 minutes.')
|
||||
return False, None
|
||||
|
||||
for _client in self.Core.clients:
|
||||
if not _client:
|
||||
continue
|
||||
if _client.nick == client.nick and _client.guest == client.guest:
|
||||
await client.kick('Stale Client (replaced by new client)')
|
||||
return False, None
|
||||
|
||||
ev.call_event("on_auth", client)
|
||||
|
||||
if len(self.Core.clients) > config.Game["players"]:
|
||||
if len(self.Core.clients_by_id) > config.Game["players"]:
|
||||
await client.kick("Server full!")
|
||||
return False, None
|
||||
else:
|
||||
self.log.info("Identification success")
|
||||
self.Core.insert_client(client)
|
||||
await self.Core.insert_client(client)
|
||||
|
||||
return True, client
|
||||
|
||||
async def set_down_rw(self, reader, writer):
|
||||
try:
|
||||
cid = (await reader.read(1)).decode() # FIXME: wtf? 1 byte?
|
||||
self.log.debug(f"Client: \"ID: {cid}\" - HandleDownload!")
|
||||
if not cid.isdigit():
|
||||
return False
|
||||
cid = (await reader.read(1))[0]
|
||||
ok = False
|
||||
for _client in self.Core.clients:
|
||||
if not _client:
|
||||
continue
|
||||
if _client.cid == cid:
|
||||
_client.down_rw = (reader, writer)
|
||||
return True
|
||||
ok = True
|
||||
self.log.debug(f"Client: {_client.nick}:{cid} - HandleDownload!")
|
||||
if not ok:
|
||||
writer.close()
|
||||
self.log.debug(f"Unknown client - HandleDownload")
|
||||
finally:
|
||||
return False
|
||||
return
|
||||
|
||||
async def handle_code(self, code, reader, writer):
|
||||
match code:
|
||||
@ -87,17 +99,17 @@ class TCPServer:
|
||||
result, client = await self.auth_client(reader, writer)
|
||||
if result:
|
||||
await client.looper()
|
||||
return True
|
||||
return False
|
||||
return result, client
|
||||
case "D":
|
||||
return await self.set_down_rw(reader, writer)
|
||||
await self.set_down_rw(reader, writer)
|
||||
case "P":
|
||||
writer.write(b"P")
|
||||
await writer.drain()
|
||||
return True
|
||||
writer.close()
|
||||
case _:
|
||||
self.log.error(f"Unknown code: {code}")
|
||||
return False
|
||||
writer.close()
|
||||
return False, None
|
||||
|
||||
async def handle_client(self, reader, writer):
|
||||
while True:
|
||||
@ -107,28 +119,37 @@ class TCPServer:
|
||||
break
|
||||
code = data.decode()
|
||||
self.log.debug(f"Received {code!r} from {writer.get_extra_info('sockname')!r}")
|
||||
result = await self.handle_code(code, reader, writer)
|
||||
if not result:
|
||||
# task = asyncio.create_task(self.handle_code(code, reader, writer))
|
||||
# await asyncio.wait([task], return_when=asyncio.FIRST_EXCEPTION)
|
||||
_, cl = await self.handle_code(code, reader, writer)
|
||||
if cl:
|
||||
await cl.remove_me()
|
||||
break
|
||||
except Exception as e:
|
||||
self.log.error("Error while connecting..")
|
||||
self.log.error(f"Error: {e}")
|
||||
self.log.exception(e)
|
||||
traceback.print_exc()
|
||||
break
|
||||
|
||||
async def start(self):
|
||||
self.log.debug("Starting TCP server.")
|
||||
self.run = True
|
||||
try:
|
||||
server = await asyncio.start_server(self.handle_client, self.host, self.port,
|
||||
backlog=int(config.Game["players"] * 1.3))
|
||||
except OSError as e:
|
||||
self.log.error(f"Error: {e}")
|
||||
self.Core.run = False
|
||||
raise e
|
||||
self.log.debug(f"TCP server started on {server.sockets[0].getsockname()!r}")
|
||||
while True:
|
||||
async with server:
|
||||
await server.serve_forever()
|
||||
except OSError as e:
|
||||
self.log.error("Cannot bind port")
|
||||
raise e
|
||||
except BaseException as e:
|
||||
self.log.error(f"Error: {e}")
|
||||
raise e
|
||||
finally:
|
||||
self.run = False
|
||||
self.Core.run = False
|
||||
|
||||
def stop(self):
|
||||
self.log.debug("Stopping TCP server")
|
||||
|
@ -15,13 +15,14 @@ from core.core import Client
|
||||
class TCPServer:
|
||||
def __init__(self, core: Core, host, port):
|
||||
self.log = utils.get_logger("TCPServer")
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.Core = core
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.run = False
|
||||
async def auth_client(self, reader: StreamReader, writer: StreamWriter) -> Tuple[bool, Client]: ...
|
||||
async def set_down_rw(self, reader: StreamReader, writer: StreamWriter) -> bool: ...
|
||||
async def handle_code(self, code: str, reader: StreamReader, writer: StreamWriter) -> bool: ...
|
||||
async def handle_code(self, code: str, reader: StreamReader, writer: StreamWriter) -> Tuple[bool, Client]: ...
|
||||
async def handle_client(self, reader: StreamReader, writer: StreamWriter) -> None: ...
|
||||
async def start(self) -> None: ...
|
||||
async def stop(self) -> None: ...
|
||||
|
@ -14,43 +14,47 @@ class UDPServer:
|
||||
|
||||
def __init__(self, core, host, port):
|
||||
self.log = utils.get_logger("UDPServer")
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.Core = core
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.run = False
|
||||
|
||||
async def handle_client(self, srv_sock):
|
||||
async def handle_client(self, reader, writer):
|
||||
while True:
|
||||
try:
|
||||
data, addr = await self.loop.sock_recv(srv_sock, 1024)
|
||||
data = await reader.read(1)
|
||||
if not data:
|
||||
break
|
||||
code = data.decode()
|
||||
self.log.debug(f"Received {code!r} from {addr!r}")
|
||||
# if not await self.handle_code(code, sock):
|
||||
# break
|
||||
self.log.debug(f"Received {code!r} from {writer.get_extra_info('sockname')!r}")
|
||||
# await self.handle_code(code, reader, writer)
|
||||
# task = asyncio.create_task(self.handle_code(code, reader, writer))
|
||||
# await asyncio.wait([task], return_when=asyncio.FIRST_EXCEPTION)
|
||||
if not writer.is_closing():
|
||||
writer.close()
|
||||
self.log.debug("Disconnected.")
|
||||
break
|
||||
except Exception as e:
|
||||
self.log.error("Error while connecting..")
|
||||
self.log.error(f"Error: {e}")
|
||||
traceback.print_exc()
|
||||
break
|
||||
srv_sock.close()
|
||||
self.log.error("Error while connecting..")
|
||||
|
||||
async def start(self):
|
||||
self.log.debug("Starting UDP server.")
|
||||
self.run = True
|
||||
try:
|
||||
pass
|
||||
# self.log.debug("Starting UDP server.")
|
||||
# await self.stop()
|
||||
# srv_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
# srv_sock.bind((self.host, self.port))
|
||||
# self.log.debug(f"Serving on {srv_sock.getsockname()}")
|
||||
# try:
|
||||
# await self.handle_client(srv_sock)
|
||||
# except Exception as e:
|
||||
# self.log.error(f"Error: {e}")
|
||||
# traceback.print_exc()
|
||||
# finally:
|
||||
# await self.stop()
|
||||
except OSError as e:
|
||||
self.log.error("Cannot bind port or other error")
|
||||
raise e
|
||||
except BaseException as e:
|
||||
self.log.error(f"Error: {e}")
|
||||
raise e
|
||||
finally:
|
||||
self.run = False
|
||||
self.Core.run = False
|
||||
|
||||
def stop(self):
|
||||
pass
|
||||
# self.log.debug("Stopping UDP server")
|
||||
self.log.debug("Stopping UDP server")
|
||||
|
@ -13,10 +13,11 @@ class UDPServer:
|
||||
|
||||
def __init__(self, core, host, port):
|
||||
self.log = utils.get_logger("UDPServer")
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.Core = core
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.run = False
|
||||
async def handle_client(self, srv_sock) -> None: ...
|
||||
async def start(self) -> None: ...
|
||||
|
||||
|
@ -26,6 +26,7 @@ class Console:
|
||||
prompt_out="",
|
||||
not_found="Command \"%s\" not found in alias.",
|
||||
debug=False) -> None:
|
||||
self.__logger = get_logger("console")
|
||||
self.__is_run = False
|
||||
self.__prompt_in = prompt_in
|
||||
self.__prompt_out = prompt_out
|
||||
@ -44,10 +45,11 @@ class Console:
|
||||
self.completer = NestedCompleter.from_nested_dict(self.__alias)
|
||||
|
||||
def __debug(self, *x):
|
||||
if self.__is_debug:
|
||||
x = list(x)
|
||||
x.insert(0, "\r CONSOLE DEBUG:")
|
||||
self.__print(*x)
|
||||
self.__logger.debug(f"{x}")
|
||||
# if self.__is_debug:
|
||||
# x = list(x)
|
||||
# x.insert(0, "\r CONSOLE DEBUG:")
|
||||
# self.__print(*x)
|
||||
|
||||
def __getitem__(self, item):
|
||||
print(item)
|
||||
@ -132,7 +134,8 @@ class Console:
|
||||
print_formatted_text(s)
|
||||
|
||||
def log(self, s: AnyStr) -> None:
|
||||
self.write(s)
|
||||
self.__logger.info(f"{s}")
|
||||
# self.write(s)
|
||||
|
||||
def __lshift__(self, s: AnyStr) -> None:
|
||||
self.write(s)
|
||||
@ -196,17 +199,20 @@ class Console:
|
||||
cmd_s = cmd_in.split(" ")
|
||||
cmd = cmd_s[0]
|
||||
if cmd == "":
|
||||
pass
|
||||
continue
|
||||
else:
|
||||
command_object = self.__func.get(cmd)
|
||||
if command_object:
|
||||
self.log(str(command_object['f'](cmd_s[1:])))
|
||||
out = command_object['f'](cmd_s[1:])
|
||||
if out:
|
||||
self.log(out)
|
||||
else:
|
||||
self.log(self.__not_found % cmd)
|
||||
except KeyboardInterrupt:
|
||||
raise KeyboardInterrupt
|
||||
except Exception as e:
|
||||
print(f"Error in console.py: {e}")
|
||||
self.__logger.exception(e)
|
||||
|
||||
async def start(self):
|
||||
self.__is_run = True
|
||||
|
Loading…
x
Reference in New Issue
Block a user