forked from Hay1tsme/artemis
73 lines
3.0 KiB
Python
73 lines
3.0 KiB
Python
|
from twisted.internet.interfaces import IAddress
|
||
|
|
from twisted.internet.protocol import DatagramProtocol
|
||
|
|
from twisted.internet.protocol import Protocol
|
||
|
|
from autobahn.twisted.websocket import WebSocketServerProtocol, WebSocketServerFactory
|
||
|
|
from datetime import datetime
|
||
|
|
import logging
|
||
|
|
import json
|
||
|
|
|
||
|
|
from core.config import CoreConfig
|
||
|
|
from .config import PokkenConfig
|
||
|
|
from .base import PokkenBase
|
||
|
|
|
||
|
|
class PokkenStunProtocol(DatagramProtocol):
|
||
|
|
def __init__(self, cfg: CoreConfig, game_cfg: PokkenConfig, type: str) -> None:
|
||
|
|
super().__init__()
|
||
|
|
self.core_config = cfg
|
||
|
|
self.game_config = game_cfg
|
||
|
|
self.logger = logging.getLogger("pokken")
|
||
|
|
self.server_type = type
|
||
|
|
|
||
|
|
def datagramReceived(self, data, addr):
|
||
|
|
self.logger.debug(
|
||
|
|
f"{self.server_type} from from {addr[0]}:{addr[1]} -> {self.transport.getHost().port} - {data.hex()}"
|
||
|
|
)
|
||
|
|
self.transport.write(data, addr)
|
||
|
|
|
||
|
|
# 474554202f20485454502f312e310d0a436f6e6e656374696f6e3a20557067726164650d0a486f73743a207469746c65732e6861793174732e6d653a393030330d0a5365632d576562536f636b65742d4b65793a204f4a6b6d522f376b646d6953326573483548783776413d3d0d0a5365632d576562536f636b65742d56657273696f6e3a2031330d0a557067726164653a20776562736f636b65740d0a557365722d4167656e743a20576562536f636b65742b2b2f302e332e300d0a0d0a
|
||
|
|
class PokkenAdmissionProtocol(WebSocketServerProtocol):
|
||
|
|
def __init__(self, cfg: CoreConfig, game_cfg: PokkenConfig):
|
||
|
|
super().__init__()
|
||
|
|
self.core_config = cfg
|
||
|
|
self.game_config = game_cfg
|
||
|
|
self.logger = logging.getLogger("pokken")
|
||
|
|
|
||
|
|
self.base = PokkenBase(cfg, game_cfg)
|
||
|
|
|
||
|
|
def onMessage(self, payload, isBinary: bool) -> None:
|
||
|
|
msg = json.loads(payload)
|
||
|
|
self.logger.debug(f"WebSocket from from {self.transport.getPeer().host}:{self.transport.getPeer().port} -> {self.transport.getHost().port} - {msg}")
|
||
|
|
|
||
|
|
handler = getattr(self.base, f"handle_admission_{msg['api'].lower()}")
|
||
|
|
resp = handler(msg, self.transport.getPeer().host)
|
||
|
|
|
||
|
|
if "type" not in resp:
|
||
|
|
resp['type'] = "res"
|
||
|
|
if "data" not in resp:
|
||
|
|
resp['data'] = {}
|
||
|
|
if "api" not in resp:
|
||
|
|
resp['api'] = msg["api"]
|
||
|
|
if "result" not in resp:
|
||
|
|
resp['result'] = 'true'
|
||
|
|
|
||
|
|
self.logger.debug(f"Websocket response: {resp}")
|
||
|
|
self.sendMessage(json.dumps(resp).encode(), isBinary)
|
||
|
|
|
||
|
|
# 0001002c2112a442334a0506a62efa71477dcd698022002872655455524e2053796e6320436c69656e7420302e33202d20524643353338392f7475726e2d3132
|
||
|
|
class PokkenAdmissionFactory(WebSocketServerFactory):
|
||
|
|
protocol = PokkenAdmissionProtocol
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
cfg: CoreConfig,
|
||
|
|
game_cfg: PokkenConfig
|
||
|
|
) -> None:
|
||
|
|
self.core_config = cfg
|
||
|
|
self.game_config = game_cfg
|
||
|
|
super().__init__(f"ws://{self.game_config.server.hostname}:{self.game_config.server.port_admission}")
|
||
|
|
|
||
|
|
def buildProtocol(self, addr: IAddress) -> Protocol:
|
||
|
|
p = self.protocol(self.core_config, self.game_config)
|
||
|
|
p.factory = self
|
||
|
|
return p
|