mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-04 21:47:28 +08:00
115 lines
4.0 KiB
Python
115 lines
4.0 KiB
Python
# import binascii
|
|
import logging
|
|
import socketserver
|
|
import threading
|
|
from json import dumps, loads
|
|
|
|
from .aes import decrypt, encrypt
|
|
from .config import Config
|
|
from .store import Store, TCPRouter, clear_player, clear_room
|
|
from .udp_class import bi
|
|
from .udp_parser import CommandParser
|
|
|
|
logging.basicConfig(format='[%(asctime)s] %(levelname)s in %(module)s: %(message)s',
|
|
level=logging.INFO)
|
|
|
|
|
|
class UDP_handler(socketserver.BaseRequestHandler):
|
|
def handle(self):
|
|
client_msg, server = self.request
|
|
# print(client_msg)
|
|
try:
|
|
token = client_msg[:8]
|
|
iv = client_msg[8:20]
|
|
tag = client_msg[20:32]
|
|
ciphertext = client_msg[32:]
|
|
if bi(token) not in Store.link_play_data:
|
|
return None
|
|
user = Store.link_play_data[bi(token)]
|
|
|
|
plaintext = decrypt(user['key'], b'', iv, ciphertext, tag)
|
|
except Exception as e:
|
|
logging.error(e)
|
|
return None
|
|
|
|
# if Config.DEBUG:
|
|
# logging.info(
|
|
# f'UDP-From-{self.client_address[0]}-{binascii.b2a_hex(plaintext)}')
|
|
|
|
commands = CommandParser(
|
|
user['room'], user['player_index']).get_commands(plaintext)
|
|
|
|
if user['room'].players[user['player_index']].player_id == 0:
|
|
clear_player(bi(token))
|
|
if user['room'].player_num == 0:
|
|
clear_room(user['room'])
|
|
commands = [i for i in commands if i[2] == 0x12]
|
|
# 处理不能正确被踢的问题
|
|
|
|
for i in commands:
|
|
iv, ciphertext, tag = encrypt(user['key'], i, b'')
|
|
# if Config.DEBUG:
|
|
# logging.info(
|
|
# f'UDP-To-{self.client_address[0]}-{binascii.b2a_hex(i)}')
|
|
|
|
server.sendto(token + iv + tag[:12] +
|
|
ciphertext, self.client_address)
|
|
|
|
|
|
AUTH_LEN = len(Config.AUTHENTICATION)
|
|
TCP_AES_KEY = Config.TCP_SECRET_KEY.encode('utf-8').ljust(16, b'\x00')[:16]
|
|
|
|
|
|
class TCP_handler(socketserver.StreamRequestHandler):
|
|
|
|
def handle(self):
|
|
try:
|
|
if self.rfile.read(AUTH_LEN).decode('utf-8') != Config.AUTHENTICATION:
|
|
self.wfile.write(b'No authentication')
|
|
logging.warning(
|
|
f'TCP-{self.client_address[0]}-No authentication')
|
|
return None
|
|
|
|
cipher_len = int.from_bytes(self.rfile.read(8), byteorder='little')
|
|
if cipher_len > Config.TCP_MAX_LENGTH:
|
|
self.wfile.write(b'Body too long')
|
|
logging.warning(f'TCP-{self.client_address[0]}-Body too long')
|
|
return None
|
|
|
|
iv = self.rfile.read(12)
|
|
tag = self.rfile.read(12)
|
|
ciphertext = self.rfile.read(cipher_len)
|
|
|
|
self.data = decrypt(TCP_AES_KEY, b'', iv, ciphertext, tag)
|
|
message = self.data.decode('utf-8')
|
|
data = loads(message)
|
|
except Exception as e:
|
|
logging.error(e)
|
|
return None
|
|
|
|
if Config.DEBUG:
|
|
logging.info(f'TCP-From-{self.client_address[0]}-{message}')
|
|
|
|
r = TCPRouter(data).handle()
|
|
try:
|
|
r = dumps(r)
|
|
if Config.DEBUG:
|
|
logging.info(f'TCP-To-{self.client_address[0]}-{r}')
|
|
iv, ciphertext, tag = encrypt(TCP_AES_KEY, r.encode('utf-8'), b'')
|
|
r = len(ciphertext).to_bytes(8, byteorder='little') + \
|
|
iv + tag[:12] + ciphertext
|
|
except Exception as e:
|
|
logging.error(e)
|
|
return None
|
|
self.wfile.write(r)
|
|
|
|
|
|
def link_play(ip: str = Config.HOST, udp_port: int = Config.UDP_PORT, tcp_port: int = Config.TCP_PORT):
|
|
udp_server = socketserver.ThreadingUDPServer((ip, udp_port), UDP_handler)
|
|
tcp_server = socketserver.ThreadingTCPServer((ip, tcp_port), TCP_handler)
|
|
|
|
threads = [threading.Thread(target=udp_server.serve_forever), threading.Thread(
|
|
target=tcp_server.serve_forever)]
|
|
[t.start() for t in threads]
|
|
[t.join() for t in threads]
|